Merge pull request #81 from aurelio-labs/james/dependency-groups

feat: created fastembed and hybrid dependency groups

.github/workflows/test.yml

@@ -35,7 +35,7 @@ jobs:
           cache: poetry
       - name: Install dependencies
         run: |
-          poetry install
+          poetry install --all-extras
       - name: Install nltk
         run: |
           pip install nltk

docs/00-introduction.ipynb

@@ -157,7 +157,7 @@
    "source": [
     "from semantic_router.layer import RouteLayer\n",
     "\n",
-    "dl = RouteLayer(encoder=encoder, routes=routes)"
+    "rl = RouteLayer(encoder=encoder, routes=routes)"
    ]
   },
   {
@@ -184,7 +184,7 @@
     }
    ],
    "source": [
-    "dl(\"don't you love politics?\")"
+    "rl(\"don't you love politics?\")"
    ]
   },
   {
@@ -204,7 +204,7 @@
     }
    ],
    "source": [
-    "dl(\"how's the weather today?\")"
+    "rl(\"how's the weather today?\")"
    ]
   },
   {
@@ -231,7 +231,7 @@
     }
    ],
    "source": [
-    "dl(\"I'm interested in learning about llama 2\")"
+    "rl(\"I'm interested in learning about llama 2\")"
    ]
   },
   {

docs/01-save-load-from-file.ipynb

@@ -132,7 +132,7 @@
     "    \"Enter Cohere API Key: \"\n",
     ")\n",
     "\n",
-    "layer = RouteLayer(routes=routes)"
+    "rl = RouteLayer(routes=routes)"
    ]
   },
   {
@@ -156,7 +156,7 @@
     }
    ],
    "source": [
-    "layer.to_json(\"layer.json\")"
+    "rl.to_json(\"layer.json\")"
    ]
   },
   {
@@ -190,9 +190,9 @@
     "import json\n",
     "\n",
     "with open(\"layer.json\", \"r\") as f:\n",
-    "    router_json = json.load(f)\n",
+    "    layer_json = json.load(f)\n",
     "\n",
-    "print(router_json)"
+    "print(layer_json)"
    ]
   },
   {
@@ -217,7 +217,7 @@
     }
    ],
    "source": [
-    "layer = RouteLayer.from_json(\"layer.json\")"
+    "rl = RouteLayer.from_json(\"layer.json\")"
    ]
   },
   {
@@ -244,9 +244,9 @@
    ],
    "source": [
     "print(\n",
-    "    f\"\"\"{layer.encoder.type=}\n",
-    "{layer.encoder.name=}\n",
-    "{layer.routes=}\"\"\"\n",
+    "    f\"\"\"{rl.encoder.type=}\n",
+    "{rl.encoder.name=}\n",
+    "{rl.routes=}\"\"\"\n",
     ")"
    ]
   },

docs/02-dynamic-routes.ipynb

@@ -125,7 +125,7 @@
     "    \"Enter Cohere API Key: \"\n",
     ")\n",
     "\n",
-    "layer = RouteLayer(routes=routes)"
+    "rl = RouteLayer(routes=routes)"
    ]
   },
   {
@@ -152,7 +152,7 @@
     }
    ],
    "source": [
-    "layer(\"how's the weather today?\")"
+    "rl(\"how's the weather today?\")"
    ]
   },
   {
@@ -291,7 +291,7 @@
     }
    ],
    "source": [
-    "layer.add(time_route)"
+    "rl.add(time_route)"
    ]
   },
   {
@@ -330,7 +330,7 @@
     "    \"Enter OpenRouter API Key: \"\n",
     ")\n",
     "\n",
-    "layer(\"what is the time in new york city?\")"
+    "rl(\"what is the time in new york city?\")"
    ]
   },
   {

docs/03-basic-langchain-agent.ipynb

@@ -223,7 +223,7 @@
     "from semantic_router import RouteLayer\n",
     "from semantic_router.encoders import OpenAIEncoder\n",
     "\n",
-    "layer = RouteLayer(encoder=OpenAIEncoder(), routes=routes)"
+    "rl = RouteLayer(encoder=OpenAIEncoder(), routes=routes)"
    ]
   },
   {
@@ -258,7 +258,7 @@
     }
    ],
    "source": [
-    "layer(\"should I buy ON whey or MP?\")"
+    "rl(\"should I buy ON whey or MP?\")"
    ]
   },
   {
@@ -284,7 +284,7 @@
     }
    ],
    "source": [
-    "layer(\"how's the weather today?\")"
+    "rl(\"how's the weather today?\")"
    ]
   },
   {
@@ -310,7 +310,7 @@
     }
    ],
    "source": [
-    "layer(\"how do I get big arms?\")"
+    "rl(\"how do I get big arms?\")"
    ]
   },
   {
@@ -382,7 +382,7 @@
    "outputs": [],
    "source": [
     "def semantic_layer(query: str):\n",
-    "    route = layer(query)\n",
+    "    route = rl(query)\n",
     "    if route.name == \"get_time\":\n",
     "        query += f\" (SYSTEM NOTE: {get_time()})\"\n",
     "    elif route.name == \"supplement_brand\":\n",