main
HuangHai 4 weeks ago
parent a70dae980a
commit 07504ad3f2

@ -28,5 +28,5 @@ DEEPSEEK_URL = 'https://api.deepseek.com'
# 阿里云中用来调用 deepseek v3 的密钥【驿来特】
MODEL_API_KEY = "sk-f6da0c787eff4b0389e4ad03a35a911f"
MODEL_NAME = "qwen-plus"
#MODEL_NAME = "deepseek-v3"
#MODEL_NAME = "qwen-plus"
MODEL_NAME = "deepseek-v3"

@ -1,39 +1,10 @@
import requests
from Config.Config import MODEL_API_KEY, MODEL_NAME
class QWenClient:
def __init__(self):
self.api_key = MODEL_API_KEY
self.model_name = MODEL_NAME
self.base_url = "https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation"
def generate(self, prompt):
headers = {
"Authorization": f"Bearer {self.api_key}",
"Content-Type": "application/json"
}
payload = {
"model": self.model_name,
"input": {
"messages": [
{"role": "user", "content": prompt}
]
}
}
response = requests.post(self.base_url, json=payload, headers=headers)
response.raise_for_status()
return response.json()
from Util.ALiYunUtil import ALiYunUtil
if __name__ == '__main__':
client = QWenClient()
ali_util = ALiYunUtil()
while True:
prompt = input("请输入问题(输入q退出): ")
if prompt.lower() == 'q':
break
try:
result = client.generate(prompt)
print("回答:", result['output']['text'])
except Exception as e:
print(f"发生错误: {str(e)}")
answer = ali_util.chat(prompt)
print("回答:", answer)

@ -0,0 +1,28 @@
from openai import OpenAI
from Config.Config import MODEL_API_KEY, MODEL_NAME
class ALiYunUtil:
def __init__(self):
self.client = OpenAI(
api_key=MODEL_API_KEY,
base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"
)
self.model_name = MODEL_NAME
def chat(self, prompt, model=None):
"""
与阿里云大模型对话
:param prompt: 用户输入的问题
:param model: 可选指定使用的模型默认使用Config中的MODEL_NAME
:return: 模型返回的答案
"""
try:
completion = self.client.chat.completions.create(
model=model or self.model_name,
messages=[
{'role': 'user', 'content': prompt}
]
)
return completion.choices[0].message.content
except Exception as e:
return f"发生错误: {str(e)}"
Loading…
Cancel
Save