Support for GROQ API #65

valentinfrlch · Sep 15, 2024 · 79eab54 · 79eab54
1 parent a862e6f
commit 79eab54
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/custom_components/llmvision/const.py b/custom_components/llmvision/const.py
@@ -36,6 +36,7 @@
 ERROR_OPENAI_NOT_CONFIGURED = "OpenAI provider is not configured"
 ERROR_ANTHROPIC_NOT_CONFIGURED = "Anthropic provider is not configured"
 ERROR_GOOGLE_NOT_CONFIGURED = "Google provider is not configured"
+ERROR_GROQ_NOT_CONFIGURED = "Groq provider is not configured"
 ERROR_LOCALAI_NOT_CONFIGURED = "LocalAI provider is not configured"
 ERROR_OLLAMA_NOT_CONFIGURED = "Ollama provider is not configured"
 ERROR_CUSTOM_OPENAI_NOT_CONFIGURED = "Custom OpenAI provider is not configured"

diff --git a/custom_components/llmvision/request_handlers.py b/custom_components/llmvision/request_handlers.py
@@ -22,6 +22,7 @@
     ERROR_OPENAI_NOT_CONFIGURED,
     ERROR_ANTHROPIC_NOT_CONFIGURED,
     ERROR_GOOGLE_NOT_CONFIGURED,
+    ERROR_GROQ_NOT_CONFIGURED,
     ERROR_LOCALAI_NOT_CONFIGURED,
     ERROR_OLLAMA_NOT_CONFIGURED,
     ERROR_NO_IMAGE_INPUT
@@ -396,6 +397,10 @@ def _validate_call(self, provider, api_key, base64_images, ip_address=None, port
         elif provider == 'Google':
             if not api_key:
                 raise ServiceValidationError(ERROR_GOOGLE_NOT_CONFIGURED)
+        # Checks for Groq
+        elif provider == 'Groq':
+            if not api_key:
+                raise ServiceValidationError(ERROR_GROQ_NOT_CONFIGURED)
         # Checks for LocalAI
         elif provider == 'LocalAI':
             if not ip_address or not port: