feat: add support for DeepSeek provider

README.md

@@ -168,6 +168,15 @@ provider = cohere
 api_key = <your API key>
 ```
 
+If you use [DeepSeek](https://www.deepseek.com):
+
+```ini
+[deepseek]
+provider = deepseek
+api_key = <your API key>
+model = deepseek-chat
+```
+
 ## 🙉 How to use
 
 ### Transform comments into commands and vice versa

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "fish_ai"
-version = "1.0.3"
+version = "1.1.0"
 authors = [{ name = "Bastian Fredriksson", email = "realiserad@gmail.com" }]
 description = "Provides core functionality for fish-ai, an AI plugin for the fish shell."
 readme = "README.md"

src/fish_ai/engine.py

@@ -168,6 +168,12 @@ def get_openai_client():
             api_key=get_config('api_key'),
             organization=get_config('organization'),
         )
+    elif (get_config('provider') == 'deepseek'):
+        # DeepSeek is compatible with OpenAI Python SDK
+        return OpenAI(
+            api_key=get_config('api_key'),
+            base_url='https://api.deepseek.com'
+        )
     else:
         raise Exception('Unknown provider "{}".'
                         .format(get_config('provider')))