add deepseek provider

Author: djcopley

README.md

@@ -9,7 +9,7 @@
 </p>
 
 ShellOracle is an innovative terminal utility designed for intelligent shell command generation, bringing a new level of
-efficiency to your command-line interactions. ShellOracle currently supports Ollama, OpenAI, LocalAI, and Grok!
+efficiency to your command-line interactions. ShellOracle currently supports Ollama, OpenAI, Deepseek, LocalAI, and Grok!
 
 ![ShellOracle](https://i.imgur.com/lqTW1lO.gif)
 
@@ -98,6 +98,11 @@ Refer to the [Ollama docs](https://ollama.ai) for installation, available models
 To use ShellOracle with OpenAI's models, create an [API key](https://platform.openai.com/account/api-keys). Edit
 your `~/.shelloracle/config.toml` to change your provider and enter your API key.
 
+### Deepseek
+
+To use ShellOracle with Deepseek's models, create an [API key](https://platform.deepseek.com/api_keys). Edit
+your `~/.shelloracle/config.toml` to change your provider and enter your API key.
+
 ### LocalAI
 
 Refer to the [LocalAI docs](https://localai.io/) for installation, available models, and usage.

src/shelloracle/providers/__init__.py

@@ -74,12 +74,13 @@ def __get__(self, instance: Provider, owner: type[Provider]) -> T:
 
 
 def _providers() -> dict[str, type[Provider]]:
+    from shelloracle.providers.deepseek import Deepseek
     from shelloracle.providers.localai import LocalAI
     from shelloracle.providers.ollama import Ollama
     from shelloracle.providers.openai import OpenAI
     from shelloracle.providers.xai import XAI
 
-    return {Ollama.name: Ollama, OpenAI.name: OpenAI, LocalAI.name: LocalAI, XAI.name: XAI}
+    return {Ollama.name: Ollama, OpenAI.name: OpenAI, LocalAI.name: LocalAI, XAI.name: XAI, Deepseek.name: Deepseek}
 
 
 def get_provider(name: str) -> type[Provider]:

src/shelloracle/providers/deepseek.py

@@ -0,0 +1,32 @@
+from collections.abc import AsyncIterator
+
+from openai import APIError, AsyncOpenAI
+
+from shelloracle.providers import Provider, ProviderError, Setting, system_prompt
+
+
+class Deepseek(Provider):
+    name = "Deepseek"
+
+    api_key = Setting(default="")
+    model = Setting(default="deepseek-chat")
+
+    def __init__(self):
+        if not self.api_key:
+            msg = "No API key provided"
+            raise ProviderError(msg)
+        self.client = AsyncOpenAI(base_url="https://api.deepseek.com/v1", api_key=self.api_key)
+
+    async def generate(self, prompt: str) -> AsyncIterator[str]:
+        try:
+            stream = await self.client.chat.completions.create(
+                model=self.model,
+                messages=[{"role": "system", "content": system_prompt}, {"role": "user", "content": prompt}],
+                stream=True,
+            )
+            async for chunk in stream:
+                if chunk.choices[0].delta.content is not None:
+                    yield chunk.choices[0].delta.content
+        except APIError as e:
+            msg = f"Something went wrong while querying Deepseek: {e}"
+            raise ProviderError(msg) from e

tests/providers/test_deepseek.py

@@ -0,0 +1,41 @@
+import pytest
+
+from shelloracle.providers.deepseek import Deepseek
+
+
+class TestOpenAI:
+    @pytest.fixture
+    def deepseek_config(self, set_config):
+        config = {
+            "shelloracle": {"provider": "Deepseek"},
+            "provider": {
+                "Deepseek": {
+                    "api_key": "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx",
+                    "model": "grok-beta",
+                }
+            },
+        }
+        set_config(config)
+
+    @pytest.fixture
+    def deepseek_instance(self, deepseek_config):
+        return Deepseek()
+
+    def test_name(self):
+        assert Deepseek.name == "Deepseek"
+
+    def test_api_key(self, deepseek_instance):
+        assert (
+            deepseek_instance.api_key
+            == "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
+        )
+
+    def test_model(self, deepseek_instance):
+        assert deepseek_instance.model == "grok-beta"
+
+    @pytest.mark.asyncio
+    async def test_generate(self, mock_asyncopenai, deepseek_instance):
+        result = ""
+        async for response in deepseek_instance.generate(""):
+            result += response
+        assert result == "head -c 100 /dev/urandom | hexdump -C"