From 86a38aec85123f5e7126118fe3e7b627a8af53b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D9=8A?= <ajay.calsoft@gmail.com>
Date: Mon, 27 Jan 2025 17:58:29 +0530
Subject: [PATCH] Create deepseek_text_gen.py

---
 .../text_generation/deepseek_text_gen.py         | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 lib/gpt_providers/text_generation/deepseek_text_gen.py

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="<DeepSeek 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)