diff --git a/lib/gpt_providers/text_generation/deepseek_text_gen.py b/lib/gpt_providers/text_generation/deepseek_text_gen.py new file mode 100644 index 00000000..f453a1fc --- /dev/null +++ b/lib/gpt_providers/text_generation/deepseek_text_gen.py @@ -0,0 +1,16 @@ +# Please install OpenAI SDK first: `pip3 install openai` + +from openai import OpenAI + +client = OpenAI(api_key="", base_url="https://api.deepseek.com") + +response = client.chat.completions.create( + model="deepseek-chat", + messages=[ + {"role": "system", "content": "You are a helpful assistant"}, + {"role": "user", "content": "Hello"}, + ], + stream=False +) + +print(response.choices[0].message.content)