feat(cohere_command_light): support for command light

aws-samples · Nov 22, 2023 · cb75134 · cb75134
1 parent 6229b4b
commit cb75134
Showing 1 changed file with 3 additions and 1 deletion.
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
@@ -55,4 +55,6 @@ def get_prompt(self):
 
 
 # Register the adapter
-registry.register(r"^bedrock.cohere.command-text*", BedrockCohereCommandAdapter)
+registry.register(
+    r"^bedrock\.cohere\.command-(text|light-text).*", BedrockCohereCommandAdapter
+)