Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Add get_config in groq_client.py file
  • Loading branch information
Kunjan Shah committed Jun 13, 2025
commit 4f272931404426dabac7d41d95eeb3eab71451ce
11 changes: 9 additions & 2 deletions giskard/llm/client/groq_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,14 @@ def __init__(
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
}

def complete(
self,
Expand Down Expand Up @@ -89,4 +96,4 @@ def complete(

msg = completion.choices[0].message

return ChatMessage(role=msg.role, content=msg.content)
return ChatMessage(role=msg.role, content=msg.content)
Loading