Skip to content

Commit 4f27293

Browse files
author
Kunjan Shah
committed
Add get_config in groq_client.py file
1 parent 3bc5b74 commit 4f27293

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

‎giskard/llm/client/groq_client.py‎

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,14 @@ def __init__(
3939
logger.info(f"Initializing GroqClient with model: {model}")
4040
self.model = model
4141
self._client = client or Groq()
42-
logger.info("GroqClient initialized successfully")
42+
logger.info("GroqClient initialized successfully")
43+
44+
def get_config(self) -> dict:
45+
"""Return the configuration of the LLM client."""
46+
return {
47+
"client_type": self.__class__.__name__,
48+
"model": self.model
49+
}
4350

4451
def complete(
4552
self,
@@ -89,4 +96,4 @@ def complete(
8996

9097
msg = completion.choices[0].message
9198

92-
return ChatMessage(role=msg.role, content=msg.content)
99+
return ChatMessage(role=msg.role, content=msg.content)

0 commit comments

Comments
 (0)