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)