Update PCAgent/api.py

PCAgent/api.py

@@ -35,6 +35,23 @@ def inference_chat(chat, model, api_url, token):
     for role, content in chat:
         messages.append({"role": role, "content": content})
 
+    payload = json.dumps({
+        "model": "gpt-4o",
+        "messages": messages
+    })
+
+    headers = {
+        'Accept': 'application/json',
+        'Authorization': f'Bearer {token}',
+        'User-Agent': 'Apifox/1.0.0 (https://apifox.com)',
+        'Content-Type': 'application/json'
+    }
+    
+    response = requests.request("POST", api_url, headers=headers, data=payload)
+    data = response.json()
+    print(data)
+    return data
+
     # client = OpenAI(
     #     # 若没有配置环境变量，请用百炼API Key将下行替换为：api_key="sk-xxx",
     #     api_key=token, 
@@ -59,23 +76,23 @@ def inference_chat(chat, model, api_url, token):
     #     else:
     #         break
 
-    dashscope.base_http_api_url = api_url
-    num_try = 5
-    for _ in range(num_try):
-        try:
-            response = MultiModalConversation.call(api_key=token, model=model, messages=messages)
-            response = response['output']['choices'][0]['message']['content'][0]["text"]
-        except Exception as e:
-            print(f"Network Error: {e}")
-            try:
-                print(response)
-            except:
-                print("Request Failed")
-            time.sleep(2)
-        else:
-            break
+    # dashscope.base_http_api_url = api_url
+    # num_try = 5
+    # for _ in range(num_try):
+    #     try:
+    #         response = MultiModalConversation.call(api_key=token, model=model, messages=messages)
+    #         response = response['output']['choices'][0]['message']['content'][0]["text"]
+    #     except Exception as e:
+    #         print(f"Network Error: {e}")
+    #         try:
+    #             print(response)
+    #         except:
+    #             print("Request Failed")
+    #         time.sleep(2)
+    #     else:
+    #         break
     
-    return response
+    # return response
     # return json.loads(completion.model_dump_json())['choices'][0]['message']['content']
 
     # headers = {