diff --git a/docs/targets/bedrock_knowledge_bases.md b/docs/targets/bedrock_knowledge_bases.md index 6cb3836..93e4365 100644 --- 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 index 0983b57..5c95770 100644 --- 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, }