From bb332064faf0ed292f5ad06dbe6a6c9b3cf0543e Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Mon, 29 Apr 2024 16:36:03 +0400
Subject: [PATCH] Fixing pytest.

---
 ...c-routes-via-openai-function-calling.ipynb | 77 +++++++++++++++----
 tests/unit/test_route.py                      |  1 +
 2 files changed, 61 insertions(+), 17 deletions(-)

diff --git a/docs/10-dynamic-routes-via-openai-function-calling.ipynb b/docs/10-dynamic-routes-via-openai-function-calling.ipynb
index 3a33fc6e..3db3b3fe 100644
--- a/docs/10-dynamic-routes-via-openai-function-calling.ipynb
+++ b/docs/10-dynamic-routes-via-openai-function-calling.ipynb
@@ -45,20 +45,11 @@
         "id": "dLElfRhgur0v"
       },
       "outputs": [
-        {
-          "name": "stderr",
-          "output_type": "stream",
-          "text": [
-            "\n",
-            "[notice] A new release of pip is available: 23.1.2 -> 24.0\n",
-            "[notice] To update, run: python.exe -m pip install --upgrade pip\n"
-          ]
-        },
         {
           "name": "stdout",
           "output_type": "stream",
           "text": [
-            "Requirement already satisfied: tzdata in c:\\users\\siraj\\documents\\personal\\work\\aurelio\\virtual environments\\semantic_router_2\\lib\\site-packages (2024.1)\n"
+            "Requirement already satisfied: tzdata in c:\\users\\siraj\\documents\\personal\\work\\aurelio\\virtual environments\\semantic_router_3\\lib\\site-packages (2024.1)\n"
           ]
         },
         {
@@ -72,7 +63,7 @@
         }
       ],
       "source": [
-        "!pip install -qU semantic-router==0.0.34\n",
+        "# !pip install -qU semantic-router==0.0.37\n",
         "!pip install tzdata"
       ]
     },
@@ -105,7 +96,7 @@
           "name": "stderr",
           "output_type": "stream",
           "text": [
-            "c:\\Users\\Siraj\\Documents\\Personal\\Work\\Aurelio\\Virtual Environments\\semantic_router_2\\Lib\\site-packages\\tqdm\\auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
+            "c:\\Users\\Siraj\\Documents\\Personal\\Work\\Aurelio\\Virtual Environments\\semantic_router_3\\Lib\\site-packages\\tqdm\\auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
             "  from .autonotebook import tqdm as notebook_tqdm\n"
           ]
         }
@@ -161,7 +152,27 @@
           "name": "stderr",
           "output_type": "stream",
           "text": [
-            "\u001b[32m2024-04-29 14:35:09 INFO semantic_router.utils.logger local\u001b[0m\n"
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger local\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 1 length: 34\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 1 trunc length: 34\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 2 length: 51\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 2 trunc length: 51\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 3 length: 66\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 3 trunc length: 66\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 4 length: 38\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 4 trunc length: 38\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 5 length: 27\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 5 trunc length: 27\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 6 length: 24\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 6 trunc length: 24\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 7 length: 21\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 7 trunc length: 21\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 8 length: 20\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 8 trunc length: 20\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 9 length: 25\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 9 trunc length: 25\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 10 length: 22\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:45 INFO semantic_router.utils.logger Document 10 trunc length: 22\u001b[0m\n"
           ]
         }
       ],
@@ -206,6 +217,14 @@
         "outputId": "f3a1dc0b-d760-4efb-b634-d3547011dcb7"
       },
       "outputs": [
+        {
+          "name": "stderr",
+          "output_type": "stream",
+          "text": [
+            "\u001b[32m2024-04-29 16:25:46 INFO semantic_router.utils.logger Document 1 length: 24\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:46 INFO semantic_router.utils.logger Document 1 trunc length: 24\u001b[0m\n"
+          ]
+        },
         {
           "data": {
             "text/plain": [
@@ -280,7 +299,7 @@
         {
           "data": {
             "text/plain": [
-              "'06:35'"
+              "'08:25'"
             ]
           },
           "execution_count": 6,
@@ -415,7 +434,13 @@
           "name": "stderr",
           "output_type": "stream",
           "text": [
-            "\u001b[32m2024-04-29 14:35:10 INFO semantic_router.utils.logger Adding `get_time` route\u001b[0m\n"
+            "\u001b[32m2024-04-29 16:25:46 INFO semantic_router.utils.logger Adding `get_time` route\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:46 INFO semantic_router.utils.logger Document 1 length: 34\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:46 INFO semantic_router.utils.logger Document 1 trunc length: 34\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:46 INFO semantic_router.utils.logger Document 2 length: 27\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:46 INFO semantic_router.utils.logger Document 2 trunc length: 27\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:46 INFO semantic_router.utils.logger Document 3 length: 32\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:46 INFO semantic_router.utils.logger Document 3 trunc length: 32\u001b[0m\n"
           ]
         }
       ],
@@ -448,7 +473,9 @@
           "name": "stderr",
           "output_type": "stream",
           "text": [
-            "\u001b[33m2024-04-29 14:35:11 WARNING semantic_router.utils.logger No LLM provided for dynamic route, will use OpenAI LLM default. Ensure API key is set in OPENAI_API_KEY environment variable.\u001b[0m\n"
+            "\u001b[32m2024-04-29 16:25:47 INFO semantic_router.utils.logger Document 1 length: 34\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:47 INFO semantic_router.utils.logger Document 1 trunc length: 34\u001b[0m\n",
+            "\u001b[33m2024-04-29 16:25:47 WARNING semantic_router.utils.logger No LLM provided for dynamic route, will use OpenAI LLM default. Ensure API key is set in OPENAI_API_KEY environment variable.\u001b[0m\n"
           ]
         },
         {
@@ -475,7 +502,7 @@
         {
           "data": {
             "text/plain": [
-              "'06:35'"
+              "'08:25'"
             ]
           },
           "execution_count": 12,
@@ -508,6 +535,14 @@
       "execution_count": 13,
       "metadata": {},
       "outputs": [
+        {
+          "name": "stderr",
+          "output_type": "stream",
+          "text": [
+            "\u001b[32m2024-04-29 16:25:48 INFO semantic_router.utils.logger Document 1 length: 24\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:48 INFO semantic_router.utils.logger Document 1 trunc length: 24\u001b[0m\n"
+          ]
+        },
         {
           "data": {
             "text/plain": [
@@ -536,6 +571,14 @@
       "execution_count": 14,
       "metadata": {},
       "outputs": [
+        {
+          "name": "stderr",
+          "output_type": "stream",
+          "text": [
+            "\u001b[32m2024-04-29 16:25:48 INFO semantic_router.utils.logger Document 1 length: 25\u001b[0m\n",
+            "\u001b[32m2024-04-29 16:25:48 INFO semantic_router.utils.logger Document 1 trunc length: 25\u001b[0m\n"
+          ]
+        },
         {
           "data": {
             "text/plain": [
diff --git a/tests/unit/test_route.py b/tests/unit/test_route.py
index 3cedeadb..178470ea 100644
--- a/tests/unit/test_route.py
+++ b/tests/unit/test_route.py
@@ -125,6 +125,7 @@ class TestRoute:
             "utterances": ["utterance"],
             "description": None,
             "function_schema": None,
+            "openai_function_schema": None,
             "llm": None,
             "score_threshold": None,
         }
-- 
GitLab