fix little inconsistency with get_config

giskard/llm/client/bedrock.py

@@ -61,6 +61,12 @@ def complete(
 
         return self._parse_completion(completion, caller_id)
 
+    def get_config(self) -> dict:
+        """Return the configuration of the LLM client."""
+        return {
+            "client_type": self.__class__.__name__,
+            "model": self.model
+        }
 
 @deprecated("ClaudeBedrockClient is deprecated: https://docs.giskard.ai/en/latest/open_source/setting_up/index.html")
 class ClaudeBedrockClient(BaseBedrockClient):

giskard/llm/client/gemini.py

@@ -57,6 +57,13 @@ def __init__(self, model: str = "gemini-pro", _client=None):
         self.model = model
         self._client = _client or genai.GenerativeModel(self.model)
 
+    def get_config(self) -> dict:
+        """Return the configuration of the LLM client."""
+        return {
+            "client_type": self.__class__.__name__,
+            "model": self.model
+        }
+
     def complete(
         self,
         messages: Sequence[ChatMessage],

giskard/llm/client/litellm.py

@@ -155,6 +155,7 @@ def complete(
     def get_config(self) -> dict:
         """Return the configuration of the LLM client."""
         return {
+            "client_type": self.__class__.__name__,
             "model": self.model,
             "disable_structured_output": self.disable_structured_output,
             "completion_params": self.completion_params,

giskard/llm/client/mistral.py

@@ -24,6 +24,13 @@ def __init__(self, model: str = "mistral-large-latest", client: Mistral = None):
         self.model = model
         self._client = client or Mistral(api_key=os.getenv("MISTRAL_API_KEY", ""))
 
+    def get_config(self) -> dict:
+        """Return the configuration of the LLM client."""
+        return {
+            "client_type": self.__class__.__name__,
+            "model": self.model
+        }
+
     def complete(
         self,
         messages: Sequence[ChatMessage],

giskard/llm/client/openai.py

@@ -37,6 +37,14 @@ def __init__(
         self._client = client or openai.OpenAI()
         self.json_mode = json_mode if json_mode is not None else _supports_json_format(model)
 
+    def get_config(self) -> dict:
+        """Return the configuration of the LLM client."""
+        return {
+            "client_type": self.__class__.__name__,
+            "model": self.model,
+            "json_mode": self.json_mode,
+        }
+
     def complete(
         self,
         messages: Sequence[ChatMessage],

giskard/rag/testset.py

@@ -158,12 +158,7 @@ def push_to_hub(
         template = template_path.read_text()
 
         # Make and push the dataset card
-        llm_client = get_default_client()
-        if not hasattr(llm_client, "get_config") or not callable(getattr(llm_client, "get_config")):
-            logging.warning(f"The 'get_config' abstract method has not been implemented for {type(llm_client)}. Configuration is empty.")
-            config = {}
-        else:
-            config = {"metadata": llm_client.get_config()}
+        config = {"metadata": get_default_client().get_config()}
         content = template.format(repo_id=repo_id, num_items=len(self._dataframe), config=json.dumps(config, indent=4))
         return DatasetCard(content=content).push_to_hub(repo_id=repo_id, token=token, repo_type="dataset")