重整项目结构

promptAgent copy.py

@@ -1,124 +0,0 @@
-import json
-import requests
-from ExcelHelper import ExcelHelper
-
-def handle_response(data_dict):
-    message = data_dict.get("message")
-    if len(message) > 0:
-        content = message.get("content")
-        if len(content) > 0:
-            response_type = content.get("type")
-            if response_type == "text":
-                text = content.get("text", "No text provided")
-                return f"{text}"
-
-            elif response_type == "image":
-                images = content.get("image", [])
-                image_urls = ", ".join(image.get("image_url") for image in images)
-                return f"{image_urls}"
-
-            elif response_type == "code":
-                return f"{content.get('code')}"
-
-            elif response_type == "execution_output":
-                return f"{content.get('content')}"
-
-            elif response_type == "system_error":
-                return f"{content.get('content')}"
-
-            elif response_type == "tool_calls":
-                return f"{data_dict['tool_calls']}"
-
-            elif response_type == "browser_result":
-                content = json.loads(content.get("content", "{}"))
-                return f"Browser Result - Title: {content.get('title')} URL: {content.get('url')}"
-
-def send_message(assistant_id, access_token, prompt, conversation_id=None, file_list=None, meta_data=None):
-    url = "https://chatglm.cn/chatglm/assistant-api/v1/stream"
-    headers = {
-        "Authorization": f"Bearer {access_token}",
-        "Content-Type": "application/json"
-    }
-    data = {
-        "assistant_id": assistant_id,
-        "prompt": prompt,
-    }
-
-
-    if conversation_id:
-        data["conversation_id"] = conversation_id
-    if file_list:
-        data["file_list"] = file_list
-    if meta_data:
-        data["meta_data"] = meta_data
-
-    with requests.post(url, json=data, headers=headers) as response:
-        if response.status_code == 200:
-            for line in response.iter_lines():
-                if line:
-                    decoded_line = line.decode('utf-8')
-                    if decoded_line.startswith('data:'):
-                        data_dict = json.loads(decoded_line[5:])
-                        output = handle_response(data_dict)
-
-
-        else:
-            return "Request failed", response.status_code
-        print(output)
-    return output
-
-def get_access_token(api_key, api_secret):
-    url = "https://chatglm.cn/chatglm/assistant-api/v1/get_token"
-    data = {
-        "api_key": api_key,
-        "api_secret": api_secret
-    }
-
-    response = requests.post(url, json=data)
-    token_info = response.json()
-    return token_info['result']['access_token']
-
-def process_nodes(process):
-    if (process == None):
-        return
-    item = process[0]
-
-    while (item != None):
-        prompt = item['prompt']
-        print(f'process node {item}')
-        result = send_message(assistant_id, access_token, prompt)
-        item['content'] = result
-        # if String result contains "继续处理"
-        if "正向处理" in result:
-            item = next((element for element in process if element['id'] == item["affirm_next"]), None)
-        else:
-            item = next((element for element in process if element['id'] == item["negative_next"]), None)
-    
-    for item in process:
-        if item.get('content') is None:
-            item['content'] = ""
-
-    return process
-
-
-# Here you need to replace the API Key and API Secret with your，I provide a test key and secret here
-api_key = '25bda2c39c0f8ca0' 
-api_secret = 'e0008b9b9727cb8ceea5a132dbe62495' 
-token = get_access_token(api_key, api_secret)
-
-assistant_id = "66a9feedfc354408c4de57dc" 
-access_token = token
-
-excelData = []
-
-# load customer aqusition process
-for index in range(0, 4):
-    file_path = f'customer_aqusition_{index}.json'
-    customer_aqusition_process = json.loads(open(file_path, 'r',encoding='utf-8').read())
-    customer_aqusition_process = process_nodes(customer_aqusition_process)
-    excelData.append(customer_aqusition_process)
-
-
-# 创建 ExcelHelper 实例并生成 Excel 文件
-excel_helper = ExcelHelper(excelData)
-excel_helper.create_excel('output.xlsx')

test/test_agent_proxy.py

@@ -0,0 +1,8 @@
+from AgentProxy import AgentProxy
+
+api_key = 'c6bbe7f48063a2c1' 
+api_secret = '5f8e7d3a97465cc099bf19bd1b70c266' 
+assistant_id = "66bb09a84673b57506fe7bbd" 
+agent = AgentProxy(assistant_id, api_key, api_secret)
+
+print(agent.send_message("你好"))