75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
|
import requests
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_response(messages, api_key, base_url):
|
|||
|
|
"""原有非流式调用（保留兼容）"""
|
|||
|
|
headers = {
|
|||
|
|
"Content-Type": "application/json",
|
|||
|
|
"Authorization": f"Bearer {api_key}"
|
|||
|
|
}
|
|||
|
|
data = {
|
|||
|
|
"model": "deepseek-chat",
|
|||
|
|
"messages": messages,
|
|||
|
|
"temperature": 0.7,
|
|||
|
|
"max_tokens": 2048
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
response = requests.post(base_url, headers=headers, json=data, timeout=30)
|
|||
|
|
response.raise_for_status()
|
|||
|
|
result = response.json()
|
|||
|
|
return result["choices"][0]["message"]["content"]
|
|||
|
|
except requests.exceptions.HTTPError as e:
|
|||
|
|
return f"API请求失败（HTTP错误）：{str(e)}，请检查API密钥是否正确"
|
|||
|
|
except requests.exceptions.ConnectionError:
|
|||
|
|
return "网络连接失败，请检查网络或API地址是否正确"
|
|||
|
|
except Exception as e:
|
|||
|
|
return f"请求异常：{str(e)}"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_response_stream(messages, api_key, base_url):
|
|||
|
|
"""流式调用DeepSeek API"""
|
|||
|
|
headers = {
|
|||
|
|
"Content-Type": "application/json",
|
|||
|
|
"Authorization": f"Bearer {api_key}"
|
|||
|
|
}
|
|||
|
|
data = {
|
|||
|
|
"model": "deepseek-chat",
|
|||
|
|
"messages": messages,
|
|||
|
|
"temperature": 0.7,
|
|||
|
|
"max_tokens": 2048,
|
|||
|
|
"stream": True # 开启流式输出
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
response = requests.post(
|
|||
|
|
base_url,
|
|||
|
|
headers=headers,
|
|||
|
|
json=data,
|
|||
|
|
timeout=30,
|
|||
|
|
stream=True
|
|||
|
|
)
|
|||
|
|
response.raise_for_status()
|
|||
|
|
|
|||
|
|
for line in response.iter_lines():
|
|||
|
|
if line:
|
|||
|
|
line = line.decode("utf-8").strip()
|
|||
|
|
if line.startswith("data: "):
|
|||
|
|
line = line[6:]
|
|||
|
|
if line == "[DONE]":
|
|||
|
|
break
|
|||
|
|
try:
|
|||
|
|
chunk = json.loads(line)
|
|||
|
|
if "choices" in chunk and len(chunk["choices"]) > 0:
|
|||
|
|
content = chunk["choices"][0]["delta"].get("content", "")
|
|||
|
|
if content:
|
|||
|
|
yield content
|
|||
|
|
except json.JSONDecodeError:
|
|||
|
|
continue
|
|||
|
|
except requests.exceptions.HTTPError as e:
|
|||
|
|
yield f"API请求失败（HTTP错误）：{str(e)}，请检查API密钥是否正确"
|
|||
|
|
except requests.exceptions.ConnectionError:
|
|||
|
|
yield "网络连接失败，请检查网络或API地址是否正确"
|
|||
|
|
except Exception as e:
|
|||
|
|
yield f"请求异常：{str(e)}"
|