Skip to content

Commit

Permalink
feat(mistral): separate adapter
Browse files Browse the repository at this point in the history
  • Loading branch information
massi-ang authored and bigadsoleiman committed Mar 7, 2024
1 parent f840ecb commit 0545d45
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -3,3 +3,4 @@
from .ai21_j2 import *
from .cohere import *
from .llama2_chat import *
from .mistral import *
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,6 @@ def get_condense_question_prompt(self):

# Register the adapter
registry.register(
r"(^bedrock.meta.llama2-.*-chat.*|^bedrock.mistral.mi.*)",
r"^bedrock.meta.llama2-.*-chat.*",
BedrockMetaLLama2ChatAdapter,
)
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
from .llama2_chat import BedrockMetaLLama2ChatAdapter
from ..registry import registry


class BedrockMistralAdapter(BedrockMetaLLama2ChatAdapter): ...


# Register the adapter
registry.register(
r"^bedrock.mistral.mi.*",
BedrockMistralAdapter,
)

0 comments on commit 0545d45

Please sign in to comment.