提交

app/utils/get_vlm_token.py

@@ -0,0 +1,29 @@
+import requests
+
+
+class Demo:
+    def __init__(self):
+        self.api_key = "bX7pK9mR2nL4qF6s"  # 用户的 API Key
+        self.api_secret = "eJ8wT3hY5cA2vN9uZ1xD7fG4"  # 用户的 API Secret
+        self.host = "http://10.0.220.110"  # 接口主地址
+        self.headers = {"Content-Type": "application/json", "Authorization": ""}
+
+    # 获取认证token
+    def getAuthToken(self):
+        # print("开始登录")
+        url = f"{self.host}/api/openapi/authenticate"
+        data = {"apiKey": self.api_key, "apiSecret": self.api_secret}
+        try:
+            response = requests.post(url, json=data)
+            response.raise_for_status()
+            result = response.json()
+
+            if result["success"]:
+                self.headers["Authorization"] = f"Bearer {result['data']['token']}"
+                print("登录vlm成功")
+                return result["data"]["token"]
+            else:
+                raise Exception(f"登录失败: {result['message']}")
+        except requests.RequestException as e:
+            print(f"登录请求失败: {e}")
+            raise