28 lines
944 B
Python
28 lines
944 B
Python
import config
|
|
from openai import OpenAI
|
|
|
|
class DashScopeChatClient:
|
|
def __init__(self, api_key=None, base_url=None, model="qwen-plus"):
|
|
self.api_key = api_key or config.DASHSCOPE_API_KEY
|
|
self.base_url = base_url or "https://dashscope.aliyuncs.com/compatible-mode/v1"
|
|
self.model = model
|
|
self.client = OpenAI(api_key=self.api_key, base_url=self.base_url)
|
|
|
|
def ask(self, question, system_prompt="You are a helpful assistant."):
|
|
messages = [
|
|
{"role": "system", "content": system_prompt},
|
|
{"role": "user", "content": question}
|
|
]
|
|
completion = self.client.chat.completions.create(
|
|
model=self.model,
|
|
messages=messages
|
|
)
|
|
return completion.choices[0].message.content
|
|
|
|
# 示例用法
|
|
if __name__ == "__main__":
|
|
chat = DashScopeChatClient()
|
|
answer = chat.ask("智人之上 300字书评 简洁精炼")
|
|
print(answer)
|
|
|