发送端代码示例
import requests
def chat_LM():
url = "http://localhost:xxxx/chat/agent_chat" # 替换为实际的API服务器地址
data = {
"query": "我想要计算1+1",
"history": [],
"stream": False,
"model_name": "chatglm3-6b",
"temperature": 0.1,
"max_tokens": 0,
"prompt_name": "default",
}
headers = {
'accept': 'application/json',
'Content-Type': 'application/json',
}
response=requests.post(url, headers=headers, json=data, stream=True)
print(response)
if response.status_code == 200:
response_data = response.json()
# print(response_data)
generated_response = response_data["final_answer"]
print("Generated Response:", generated_response)
- 发送格式查看
tests/api文件夹下的文件为调用api的格式文件
查看requests.post的格式就可以了
- 结果查返回的格式为json
在设定的api端口http://localhost:xxxx/docs中可以点击查看。
返回的json的内容修改在api.py 文件里的mount_knowledge_routes函数里每个api功能对应的函数里可以修改json,yield json即为返回的respons结果。
api.post函数即为router输出
(python中yield的作用:和return的功能一样,可以返回函数的输出,但是用法不一样,具体去看yield的用法)