# -*- coding: utf-8 -*-

from openai import OpenAI

'''
deepseek-chat
'''


class DeepSeek(object):
    def call_deepseek(self, text, prompt_words):
        try:
            message = text + '\n\n' + prompt_words
            print('call deepseek')
            client = OpenAI(
                api_key="sk-20afb9967a124b63a6f67dcb69f17d74",
                base_url="https://api.deepseek.com",
            )

            completion = client.chat.completions.create(
                model="deepseek-chat",
                messages=[
                    {"role": "system", "content": "你是一个数据整理助手"},
                    {"role": "user", "content": f"{message}"},
                ],
                stream=False,
                temperature=0.3,
            )

            result = completion.choices[0].message.content

            return result
        except Exception as e:
            print(e)