langchain/libs/partners/ai21/langchain_ai21/chat/chat_factory.py

from langchain_ai21.chat.chat_adapter import (
    ChatAdapter,
    J2ChatAdapter,
    JambaChatCompletionsAdapter,
)


def create_chat_adapter(model: str) -> ChatAdapter:
    if "j2" in model:
        return J2ChatAdapter()

    if "jamba" in model:
        return JambaChatCompletionsAdapter()

    raise ValueError(f"Model {model} not supported.")
partners: AI21 Labs Jamba Support (#20815) Description: Added support for AI21 new model - Jamba Twitter handle: https://github.com/AI21Labs --------- Co-authored-by: Asaf Gardin <asafg@ai21.com> Co-authored-by: Erick Friis <erick@langchain.dev> 5 months ago			`from langchain_ai21.chat.chat_adapter import (`
			`ChatAdapter,`
			`J2ChatAdapter,`
			`JambaChatCompletionsAdapter,`
			`)`


			`def create_chat_adapter(model: str) -> ChatAdapter:`
			`if "j2" in model:`
			`return J2ChatAdapter()`

			`if "jamba" in model:`
			`return JambaChatCompletionsAdapter()`

			`raise ValueError(f"Model {model} not supported.")`