Minor fixes (#3101)

Azure · Apr 4, 2024 · f94feec · f94feec
1 parent d4df206
commit f94feec
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 53 deletions.
diff --git a/sdk/python/foundation-models/cohere/cohere-cmdR.ipynb b/sdk/python/foundation-models/cohere/cohere-cmdR.ipynb
@@ -52,7 +52,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "metadata": {
     "name": "imports"
    },
@@ -93,7 +93,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {
     "name": "chat_invoke"
    },
@@ -113,57 +113,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "metadata": {
     "name": "chat_response"
    },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "Claude Monet.\n"
-     ]
-    }
-   ],
-   "source": [
-    "print(chat_response.text)"
-   ]
-  },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "Tools"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 5,
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
    "outputs": [],
    "source": [
-    "messages = [\n",
-    "    ChatMessage(\n",
-    "        role=\"system\",\n",
-    "        content=\"You are a helpful assistant that help users to find information about traveling, how to get to places and the different transportations options. You care about the environment and you always have that in mind when answering inqueries.\",\n",
-    "    ),\n",
-    "    ChatMessage(\n",
-    "        role=\"user\",\n",
-    "        content=\"When is the next flight from Miami to Seattle?\",\n",
-    "    ),\n",
-    "]\n",
-    "\n",
-    "chat_response = client.chat(\n",
-    "    model=\"azureai\", messages=messages, tools=tools, tool_choice=ToolChoice.auto\n",
-    ")"
+    "print(chat_response.text)"
    ]
   },
   {

diff --git a/sdk/python/foundation-models/cohere/langchain.ipynb b/sdk/python/foundation-models/cohere/langchain.ipynb
@@ -117,7 +117,7 @@
    "source": [
     "chat_model = ChatCohere(\n",
     "    endpoint=\"https://<endpoint>.<region>.inference.ai.azure.com/v1\",\n",
-    "    mistral_api_key=\"<key>\",\n",
+    "    cohere_api_key=\"<key>\",\n",
     ")"
    ]
   },
@@ -210,7 +210,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.8"
+   "version": "3.10.13"
   }
  },
  "nbformat": 4,

diff --git a/sdk/python/foundation-models/cohere/litellm.ipynb b/sdk/python/foundation-models/cohere/litellm.ipynb
@@ -102,7 +102,7 @@
    "source": [
     "response = client.chat.completions.create(\n",
     "    messages=[{\"content\": \"Who is the most renowned French painter?\", \"role\": \"user\"}],\n",
-    "    model=\"mistral\",\n",
+    "    model=\"cohere\",\n",
     "    custom_llm_provider=\"custom_openai\",\n",
     ")"
    ]

diff --git a/sdk/python/foundation-models/cohere/openaisdk.ipynb b/sdk/python/foundation-models/cohere/openaisdk.ipynb
@@ -6,7 +6,7 @@
    "source": [
     "# Use OpenAI SDK with Cohere Command R or Cohere Command R+ in Azure AI and Azure ML\n",
     "\n",
-    "Use `openai` SDK to consume Mistral deployments in Azure AI and Azure ML. The Cohere family of models in Azure AI and Azure ML offers an API compatible with the OpenAI Chat Completion API. It allows customers and users to transition seamlessly from OpenAI models to Cohere LLMs. \n",
+    "Use `openai` SDK to consume Cohere deployments in Azure AI and Azure ML. The Cohere family of models in Azure AI and Azure ML offers an API compatible with the OpenAI Chat Completion API. It allows customers and users to transition seamlessly from OpenAI models to Cohere LLMs. \n",
     "\n",
     "The API can be directly used with OpenAI's client libraries or third-party tools, like LangChain or LlamaIndex.\n",
     "\n",
@@ -168,7 +168,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.8"
+   "version": "3.10.13"
   }
  },
  "nbformat": 4,