31 lines
945 B
Python
31 lines
945 B
Python
|
|
import requests
|
|
import urllib3
|
|
|
|
########################################
|
|
# 该文件实现了与大模型的简单通信
|
|
########################################
|
|
|
|
# 忽略https的安全性警告
|
|
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
|
|
|
def ask_llm(question):
|
|
url = "https://45.125.46.134:25344/v1/completions"
|
|
headers = {"Content-Type": "application/json"}
|
|
data = {
|
|
"prompt": question
|
|
}
|
|
|
|
response = requests.post(url, headers=headers, json=data, verify=False)
|
|
|
|
if response.status_code == 200:
|
|
result = response.json()
|
|
return result['choices'][0]['text']
|
|
else:
|
|
return "大模型请求失败:", response.status_code
|
|
|
|
|
|
if __name__ == '__main__':
|
|
question = "假设你是一个咖啡厅的机器人服务员，有一个顾客的请求是'请给我一杯咖啡'，请生成对应的行为树来控制机器人完成该动作"
|
|
|
|
print(ask_llm(question)) |