Projects/antiwear-gateway
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Dify配置改为从环境变量读取，自动拼接/chat-messages

Browse Source
This commit is contained in:
ZF Sun
2026-04-22 17:18:02 +08:00
parent dab4b807d9
commit 8883f2aaa6
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
digital-human.py
View File

@@ -36,8 +36,8 @@ mock_time_created = 20230512084843
mock_id1 = "131424542124" mock_id1 = "131424542124"
mock_id2 = "1231414141541252525" mock_id2 = "1231414141541252525"
mock_user = "abc-123" mock_user = "abc-123"
dify_url = "http://<your dify ip>/v1/chat-messages" dify_url = os.getenv("DIFY_URL", "")
dify_api_key = 'Bearer <your dify api key>' dify_api_key = "Bearer " + os.getenv("DIFY_API_KEY", "")
class Message(BaseModel): class Message(BaseModel):
@@ -54,8 +54,8 @@ class Resquest_Body(BaseModel):
# 对接dify，非流式传输 # 对接dify，非流式传输
def request_dify_api(question, request, conversation_id): def request_dify_api(question, request, conversation_id):
# 目标URL # 目标URL（自动拼接 /chat-messages）
url = dify_url url = dify_url.rstrip("/") + "/chat-messages"
headers_json = { headers_json = {
'Content-Type': 'application/json', 'Content-Type': 'application/json',
'Authorization': dify_api_key 'Authorization': dify_api_key
@@ -72,12 +72,11 @@ def request_dify_api(question, request, conversation_id):
response = requests.post(url, data=json.dumps(data), headers=headers_json) response = requests.post(url, data=json.dumps(data), headers=headers_json)
resp_data = response.json() resp_data = response.json()
return resp_data["answer"] return resp_data["answer"]
return resp_data["answer"]
# 对接dify，流式传输 # 对接dify，流式传输
async def generate_streaming_messages(question, request, session_id, conversation_id): async def generate_streaming_messages(question, request, session_id, conversation_id):
url = dify_url url = dify_url.rstrip("/") + "/chat-messages"
headers_json = { headers_json = {
'Content-Type': 'application/json', 'Content-Type': 'application/json',
'Authorization': dify_api_key 'Authorization': dify_api_key