refactor: change config type to bedrock-knowledge-base for consistency

awslabs · May 21, 2024 · 76abb35 · 76abb35
1 parent b77092f
commit 76abb35
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/targets/bedrock_knowledge_bases.md b/docs/targets/bedrock_knowledge_bases.md
@@ -12,7 +12,7 @@ The principal must have the following permissions:
 
 ```yaml title="agenteval.yml"
 target:
-  type: bedrock-knowledgebase
+  type: bedrock-knowledge-base
   model_id: my-model-id
   knowledge_base_id: my-kb-id
 ```

diff --git a/src/agenteval/targets/target_factory.py b/src/agenteval/targets/target_factory.py
@@ -14,7 +14,7 @@
     "bedrock-agent": BedrockAgentTarget,
     "q-business": QBusinessTarget,
     "sagemaker-endpoint": SageMakerEndpointTarget,
-    "bedrock-knowledgebase": BedrockKnowledgeBaseTarget,
+    "bedrock-knowledge-base": BedrockKnowledgeBaseTarget,
 }