from openai import OpenAI


class MoonShotHandle():
    def __init__(self, api_key=None, api_base=None):
        self.OPENAI_API_KEY = 'sk-tz1VaKqksTzk0F8HxlU4YVGwj7oa1g0c0puGNUZrdn9MDtzm'
        self.model = "moonshot-v1-8k"

    def chat(self, question):
        return self.chat_with_chatgpt(question)

    def chat_with_chatgpt(self, prompt):
        client = OpenAI(
            api_key=self.OPENAI_API_KEY,
            base_url="https://api.moonshot.cn/v1",
        )
        chat_completion = client.chat.completions.create(
            messages=[
                {
                    "role": "user",
                    "content": prompt,
                }
            ],
            model=self.model,
        )
        response = chat_completion.choices[0].message.content
        return response


# res = MoonShotHandle().chat("请问你是谁？")
# print(res)