From 75bcf2ad0afd1e08f6551c3d5d4a27257f72fc0b Mon Sep 17 00:00:00 2001
From: Sheepsta300 <128811766+Sheepsta300@users.noreply.github.com>
Date: Wed, 11 Dec 2024 12:35:36 +1300
Subject: [PATCH] Lint file

---
 .../tools/azure_content_safety.ipynb             | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/docs/docs/integrations/tools/azure_content_safety.ipynb b/docs/docs/integrations/tools/azure_content_safety.ipynb
index 5a678da1c1a68..73420ca5129de 100644
--- a/docs/docs/integrations/tools/azure_content_safety.ipynb
+++ b/docs/docs/integrations/tools/azure_content_safety.ipynb
@@ -63,9 +63,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import os\n",
     "import getpass\n",
-    "from langchain import hub"
+    "import os"
    ]
   },
   {
@@ -90,10 +89,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from langchain_openai import AzureChatOpenAI\n",
+    "\n",
     "from libs.community.langchain_community.tools.azure_ai_services.content_safety import (\n",
     "    AzureContentSafetyTextTool,\n",
-    ")\n",
-    "from langchain_openai import AzureChatOpenAI"
+    ")"
    ]
   },
   {
@@ -266,7 +266,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "agent.invoke({\"messages\": [(\"user\",  f\"Can you check the following text for harmful content : {input}\")]})"
+    "agent.invoke(\n",
+    "    {\n",
+    "        \"messages\": [\n",
+    "            (\"user\", f\"Can you check the following text for harmful content : {input}\")\n",
+    "        ]\n",
+    "    }\n",
+    ")"
    ]
   },
   {