Skip to content

Commit f840ecb

Browse files
massi-angbigadsoleiman
authored andcommitted
feat(mistral): adapter registration regex for mistral
1 parent 3b9fa8c commit f840ecb

File tree

1 file changed

+4
-1
lines changed
  • lib/model-interfaces/langchain/functions/request-handler/adapters/bedrock

1 file changed

+4
-1
lines changed

lib/model-interfaces/langchain/functions/request-handler/adapters/bedrock/llama2_chat.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,4 +61,7 @@ def get_condense_question_prompt(self):
6161

6262

6363
# Register the adapter
64-
registry.register(r"(?i)^bedrock.meta.llama2-.*-chat.*", BedrockMetaLLama2ChatAdapter)
64+
registry.register(
65+
r"(^bedrock.meta.llama2-.*-chat.*|^bedrock.mistral.mi.*)",
66+
BedrockMetaLLama2ChatAdapter,
67+
)

0 commit comments

Comments
 (0)