ai21: AI21 Jamba docs (#21978)

- Updated docs to have an example to use Jamba instead of J2

---------

Co-authored-by: Asaf Gardin <asafg@ai21.com>
Co-authored-by: Erick Friis <erick@langchain.dev>

docs/docs/integrations/chat/ai21.ipynb

@@ -95,7 +95,7 @@
     "from langchain_ai21 import ChatAI21\n",
     "from langchain_core.prompts import ChatPromptTemplate\n",
     "\n",
-    "chat = ChatAI21(model=\"j2-ultra\")\n",
+    "chat = ChatAI21(model=\"jamba-instruct\")\n",
     "\n",
     "prompt = ChatPromptTemplate.from_messages(\n",
     "    [\n",
@@ -107,14 +107,6 @@
     "chain = prompt | chat\n",
     "chain.invoke({\"english_text\": \"Hello, how are you?\"})"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c159a79f",
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {

libs/partners/ai21/README.md

@@ -27,11 +27,13 @@ Then initialize
 from langchain_core.messages import HumanMessage
 from langchain_ai21.chat_models import ChatAI21
 
-chat = ChatAI21(model="j2-ultra")
+chat = ChatAI21(model="jamab-instruct")
 messages = [HumanMessage(content="Hello from AI21")]
 chat.invoke(messages)
 ```
 
+For a list of the supported models, see [this page](https://docs.ai21.com/reference/python-sdk#chat)
+
 ## LLMs
 You can use AI21's generative AI models as Langchain LLMs: