feat(guardrails): removing support for Command-R due to lack of Langc…

…hain support
aws-samples · Jun 24, 2024 · 035ecc0 · 035ecc0
1 parent ff3ba70
commit 035ecc0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/model-interfaces/langchain/functions/request-handler/adapters/bedrock/cohere.py b/lib/model-interfaces/langchain/functions/request-handler/adapters/bedrock/cohere.py
@@ -4,7 +4,7 @@
 from langchain.prompts.prompt import PromptTemplate
 
 from ..base import ModelAdapter
-from .base import get_guardrails
+from .base import get_guardrails, Bedrock
 from genai_core.registry import registry
 
 
@@ -62,5 +62,5 @@ def get_prompt(self):
 
 # Register the adapter
 registry.register(
-    r"^bedrock\.cohere\.command-(text|light-text|r).*", BedrockCohereCommandAdapter
+    r"^bedrock\.cohere\.command-(text|light-text).*", BedrockCohereCommandAdapter
 )