通过Dify可以连接很多模型国内、国外的都可以进行选择
可以到Dify里创建一个空白应用,然后点击进入就可以看到API了
python
api_url = "http://192.168.15.131/v1/chat-messages"
api_key = "app-UtzTpVNwpTLUcGvRNnnK9QNY"
headers = {
"Authorization": f"Bearer {api_key}",
"Content-Type": "application/json"
}
speak = request.GET.get('speak')
data = {
"inputs": {},
"query": speak,
"response_mode": "streaming",
"conversation_id": "",
"user": "abc-123"
}
response = requests.post(api_url, headers=headers, data=json.dumps(data))
if response.status_code == 200:
for line in response.text.splitlines():
if line.startswith("data: "):
try:
json_data = json.loads(line[6:])
if json_data.get('event') == 'workflow_finished':
# 获取每次答案
answer = json_data['data']['outputs']['answer']
# 获取每次回答的total_tokens数量
total_tokens = json_data['data']['total_tokens']
today = datetime.now().date()
# 保存到数据库
save = Ai_info.objects.create(all_names=b_names, b_times=today, speak=speak, answer=answer, total_tokens=total_tokens, company=company_name)
return render(request, 'dify_DifyAI.html',
{'speak': speak, 'answer': answer, 'b_head': b_head})
except json.JSONDecodeError as e:
print(f"JSON语法错误: {e}")
# 处理JSON语法错误的情况
except UnicodeDecodeError as e:
print(f"编码错误: {e}")
else:
return render(request, 'error.html', {'error_message': f"Error: {response.status_code}, {response.text}"})