Mistral Small addition2

notebooks/official/generative_ai/mistralai_intro.ipynb

@@ -74,21 +74,18 @@
         "\n",
         "### Available Mistral AI models\n",
         "\n",
+        "*   ### Mistral Small 3.1 (25.03)\n",
+        "Mistral Small 3.1 (25.03) is the enhanced version of Mistral Small 3, featuring multimodal capabilities and an extended context length of up to 128k.\n",
+        "\n",
         "*   ### Codestral (25.01)\n",
         "A cutting-edge model specifically designed for code generation, including fill-in-the-middle and code completion.\n",
         "\n",
         "*   ### Mistral Large (24.11)\n",
         "Mistral Large (24.11) is the latest version of the Mistral Large model now with improved reasoning and function calling capabilities.\n",
         "\n",
-        "*   ### Mistral Large (24.07)\n",
-        "Complex tasks that require large reasoning capabilities or are highly specialized (synthetic text Generation, code generation, RAG, or agents). [Blog Post](https://mistral.ai/news/mistral-large-2407/)\n",
-        "\n",
         "*   ### Mistral Nemo\n",
         "Reasoning, world knowledge, and coding performance are state-of-the-art in its size category.\n",
         "\n",
-        "*   ### Codestral (24.05)\n",
-        "Coding specific tasks to enhance developers productivity with code completion and fill-in-the-middle capabilities.\n",
-        "\n",
         "\n",
         "## Objective\n",
         "\n",
@@ -163,8 +160,11 @@
       },
       "outputs": [],
       "source": [
-        "MODEL = \"codestral-2501\"  # @param [\"codestral-2501\", \"mistral-large-2411\", \"mistral-large\", \"mistral-nemo\", \"codestral\"]\n",
-        "if MODEL == \"mistral-large-2411\":\n",
+        "MODEL = \"mistral-small-2503\"  # @param [\"mistral-small-2503\", \"codestral-2501\", \"mistral-large-2411\", \"mistral-large\", \"mistral-nemo\", \"codestral\"]\n",
+        "if MODEL == \"mistral-small-2503\":\n",
+        "    available_regions = [\"europe-west4\", \"us-central1\"]\n",
+        "    available_versions = [\"latest\"]\n",
+        "elif MODEL == \"mistral-large-2411\":\n",
         "    available_regions = [\"europe-west4\", \"us-central1\"]\n",
         "    available_versions = [\"latest\"]\n",
         "elif MODEL == \"mistral-large\":\n",
@@ -693,8 +693,12 @@
       },
       "outputs": [],
       "source": [
-        "MODEL = \"mistral-large\"  # @param [\"mistral-large\", \"mistral-nemo\", \"codestral\"]\n",
-        "if MODEL == \"mistral-large\":\n",
+        "MODEL = \"mistral-small-2503\"  # @param [\"mistral-small-2503\", \"mistral-large\", \"mistral-nemo\", \"codestral\"]\n",
+        "\n",
+        "if MODEL == \"mistral-small\":\n",
+        "    available_regions = [\"europe-west4\", \"us-central1\"]\n",
+        "    available_versions = [\"latest\", \"2503\"]\n",
+        "elif MODEL == \"mistral-large\":\n",
         "    available_regions = [\"europe-west4\", \"us-central1\"]\n",
         "    available_versions = [\"latest\", \"2411\", \"2407\"]\n",
         "elif MODEL == \"mistral-nemo\":\n",
@@ -858,7 +862,7 @@
         "\n",
         "try:\n",
         "    resp = client.chat.complete(\n",
-        "        model=f\"{MODEL}-{MODEL_VERSION}\",\n",
+        "        model=f\"{MODEL}\",\n",
         "        messages=[\n",
         "            {\n",
         "                \"role\": \"user\",\n",
@@ -899,7 +903,7 @@
         "\n",
         "try:\n",
         "    stream = client.chat.stream(\n",
-        "        model=f\"{MODEL}-{MODEL_VERSION}\",\n",
+        "        model=f\"{MODEL}\",\n",
         "        max_tokens=1024,\n",
         "        messages=[\n",
         "            {\n",