Fix: code formatting via pre-commit

Author: Kunjan Shah

giskard/llm/client/__init__.py

@@ -36,10 +36,8 @@ def get_default_llm_api() -> str:
     global _default_llm_api
     if _default_llm_api is None:
         _default_llm_api = os.getenv(
-            "GSK_LLM_API", 
-            "azure" if "AZURE_OPENAI_API_KEY" in os.environ 
-            else "groq" if "GROQ_API_KEY" in os.environ
-            else "openai"
+            "GSK_LLM_API",
+            "azure" if "AZURE_OPENAI_API_KEY" in os.environ else "groq" if "GROQ_API_KEY" in os.environ else "openai",
         ).lower()
 
         if _default_llm_api not in {"azure", "openai", "groq"}:
@@ -92,7 +90,6 @@ def get_default_client() -> LLMClient:
     global _default_llm_api
     global _default_llm_model
     global _disable_structured_output
- 
 
     try:
         from .litellm import LiteLLMClient

giskard/llm/client/groq_client.py

@@ -1,16 +1,16 @@
 from typing import Optional, Sequence
 
-from dataclasses import asdict
 import logging
+from dataclasses import asdict
 
 from ..config import LLMConfigurationError
 from ..errors import LLMImportError
 from . import LLMClient
 from .base import ChatMessage
 
 try:
-    from groq import Groq
     import groq
+    from groq import Groq
 except ImportError as err:
     raise LLMImportError(flavor="llm") from err
 
@@ -29,24 +29,22 @@
 
 logger = logging.getLogger(__name__)
 
+
 class GroqClient(LLMClient):
     def __init__(
-        self, 
+        self,
         model: str = "llama-3.3-70b-versatile",  # Default model for Groq
         client: Groq = None,
-        #json_mode: Optional[bool] = None
+        # json_mode: Optional[bool] = None
     ):
         logger.info(f"Initializing GroqClient with model: {model}")
         self.model = model
         self._client = client or Groq()
-        logger.info("GroqClient initialized successfully")    
-    
+        logger.info("GroqClient initialized successfully")
+
     def get_config(self) -> dict:
         """Return the configuration of the LLM client."""
-        return {
-            "client_type": self.__class__.__name__,
-            "model": self.model
-        }
+        return {"client_type": self.__class__.__name__, "model": self.model}
 
     def complete(
         self,
@@ -59,12 +57,12 @@ def complete(
     ) -> ChatMessage:
         logger.info(f"GroqClient.complete called with model: {self.model}")
         logger.info(f"Messages: {messages}")
-        
+
         extra_params = dict()
 
-        extra_params["seed"] = seed 
+        extra_params["seed"] = seed
 
-        if format in {"json", "json_object"}:   
+        if format in {"json", "json_object"}:
             extra_params["response_format"] = {"type": "json_object"}
 
         try:
@@ -75,16 +73,16 @@ def complete(
                 max_tokens=max_tokens,
                 **extra_params,
             )
-        
-        except groq.AuthenticationError as err: 
+
+        except groq.AuthenticationError as err:
             raise LLMConfigurationError(AUTH_ERROR_MESSAGE) from err
-        
+
         except groq.BadRequestError as err:
             if format in {"json", "json_object"}:
                 raise LLMConfigurationError(
                     f"Model '{self.model}' does not support JSON output or the request format is incorrect.\n\n{JSON_MODE_GUIDANCE}"
                 ) from err
-            raise  
+            raise
 
         self.logger.log_call(
             prompt_tokens=completion.usage.prompt_tokens,

tests/llm/test_llm_client.py

@@ -1,4 +1,5 @@
 import json
+from importlib.util import find_spec
 from unittest.mock import MagicMock, Mock, patch
 
 import pydantic
@@ -11,9 +12,8 @@
 from giskard.llm.client import ChatMessage
 from giskard.llm.client.bedrock import ClaudeBedrockClient
 from giskard.llm.client.gemini import GeminiClient
-from giskard.llm.client.openai import OpenAIClient
 from giskard.llm.client.groq_client import GroqClient
-from importlib.util import find_spec
+from giskard.llm.client.openai import OpenAIClient
 
 PYDANTIC_V2 = pydantic.__version__.startswith("2.")
 
@@ -241,46 +241,33 @@ def test_gemini_client():
     assert isinstance(res, ChatMessage)
     assert res.content == "This is a test!"
 
+
 # Check if groq is installed
 has_groq = find_spec("groq") is not None
 
-@pytest.mark.skipif(not PYDANTIC_V2 or not has_groq, 
-                   reason="Groq client test requires Pydantic v2 and groq package")
+
+@pytest.mark.skipif(not PYDANTIC_V2 or not has_groq, reason="Groq client test requires Pydantic v2 and groq package")
 def test_groq_client():
     # Mock the Groq response
     demo_response = Mock()
     demo_response.usage = Mock(prompt_tokens=13, completion_tokens=7)
-    demo_response.choices = [
-        Mock(
-            message=Mock(
-                role="assistant",
-                content="This is a test!"
-            )
-        )
-    ]
+    demo_response.choices = [Mock(message=Mock(role="assistant", content="This is a test!"))]
 
     # Mock the Groq client
     mock_client = Mock()
     mock_client.chat.completions.create.return_value = demo_response
- 
+
     client = GroqClient(model="llama-3.3-70b-versatile", client=mock_client)
 
     # Call the complete method
-    res = client.complete(
-        [ChatMessage(role="user", content="Hello")],
-        temperature=0.7,
-        format="json",
-        max_tokens=100
-    )
+    res = client.complete([ChatMessage(role="user", content="Hello")], temperature=0.7, format="json", max_tokens=100)
 
     # Assert json_object format was passed
     assert mock_client.chat.completions.create.call_args[1]["response_format"] == {"type": "json_object"}
 
     # Assert that create was called with correct arguments
     mock_client.chat.completions.create.assert_called_once()
-    assert mock_client.chat.completions.create.call_args[1]["messages"] == [
-        {"role": "user", "content": "Hello"}
-    ]
+    assert mock_client.chat.completions.create.call_args[1]["messages"] == [{"role": "user", "content": "Hello"}]
     assert mock_client.chat.completions.create.call_args[1]["temperature"] == 0.7
     assert mock_client.chat.completions.create.call_args[1]["max_tokens"] == 100
     assert mock_client.chat.completions.create.call_args[1]["model"] == "llama-3.3-70b-versatile"