36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
|
import json
|
||
|
|
import requests
|
||
|
|
import time
|
||
|
|
|
||
|
|
def question(cont):
|
||
|
|
url=f"http://localhost:11434/api/chat"
|
||
|
|
req = json.dumps({
|
||
|
|
"model": "phi:latest",
|
||
|
|
"messages": [{
|
||
|
|
"role": "user",
|
||
|
|
"content": cont
|
||
|
|
}],
|
||
|
|
"stream": False
|
||
|
|
})
|
||
|
|
headers = {'content-type': 'application/json'}
|
||
|
|
session = requests.Session()
|
||
|
|
starttime = time.time()
|
||
|
|
|
||
|
|
try:
|
||
|
|
response = session.post(url, data=req, headers=headers)
|
||
|
|
response.raise_for_status() # 检查响应状态码是否为200
|
||
|
|
result = json.loads(response.text)
|
||
|
|
response_text = result["message"]["content"]
|
||
|
|
|
||
|
|
except requests.exceptions.RequestException as e:
|
||
|
|
print(f"请求失败: {e}")
|
||
|
|
response_text = "抱歉，我现在太忙了，休息一会，请稍后再试。"
|
||
|
|
|
||
|
|
print("接口调用耗时 :" + str(time.time() - starttime))
|
||
|
|
return response_text.strip()
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
for i in range(3):
|
||
|
|
query = "爱情是什么"
|
||
|
|
response = question(query)
|
||
|
|
print("\n The result is ", response)
|