Skip to content

Commit 0545d45

Browse files
massi-angbigadsoleiman
authored andcommitted
feat(mistral): separate adapter
1 parent f840ecb commit 0545d45

File tree

3 files changed

+14
-1
lines changed

3 files changed

+14
-1
lines changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,3 +3,4 @@
33
from .ai21_j2 import *
44
from .cohere import *
55
from .llama2_chat import *
6+
from .mistral import *

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,6 @@ def get_condense_question_prompt(self):
6262

6363
# Register the adapter
6464
registry.register(
65-
r"(^bedrock.meta.llama2-.*-chat.*|^bedrock.mistral.mi.*)",
65+
r"^bedrock.meta.llama2-.*-chat.*",
6666
BedrockMetaLLama2ChatAdapter,
6767
)
Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
from .llama2_chat import BedrockMetaLLama2ChatAdapter
2+
from ..registry import registry
3+
4+
5+
class BedrockMistralAdapter(BedrockMetaLLama2ChatAdapter): ...
6+
7+
8+
# Register the adapter
9+
registry.register(
10+
r"^bedrock.mistral.mi.*",
11+
BedrockMistralAdapter,
12+
)

0 commit comments

Comments
 (0)