Added Llama Bedrock client

Giskard-AI · May 29, 2024 · a90aaa9 · a90aaa9
1 parent ac9d386
commit a90aaa9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/giskard/llm/client/bedrock.py b/giskard/llm/client/bedrock.py
@@ -100,7 +100,7 @@ def complete(
  seed: Optional[int] = None,
  format=None,
  ) -> ChatMessage:
- # only supporting claude 3 to start
+ # only supporting llama
  if "llama" not in self.model:
  raise LLMConfigurationError(f"Only Llama models are supported as of now, got {self.model}")