Merge pull request #565 from aurelio-labs/josh/fix-threshold

jamescalam · web-flow · commit 6aacd0bd9e20 · 2025-03-20T18:52:02.000+04:00
fix: threshold issue
diff --git a/docs/00-introduction.ipynb b/docs/00-introduction.ipynb
@@ -53,7 +53,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install -qU \"semantic-router==0.1.1\""
+        "!pip install -qU \"semantic-router==0.1.2\""
       ]
     },
     {
diff --git a/docs/01-save-load-from-file.ipynb b/docs/01-save-load-from-file.ipynb
@@ -46,7 +46,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install -qU \"semantic-router==0.1.1\""
+        "!pip install -qU \"semantic-router==0.1.2\""
       ]
     },
     {
diff --git a/docs/02-dynamic-routes.ipynb b/docs/02-dynamic-routes.ipynb
@@ -95,7 +95,7 @@
    ],
    "source": [
     "!pip install tzdata\n",
-    "!pip install -qU \"semantic-router==0.1.1\""
+    "!pip install -qU \"semantic-router==0.1.2\""
    ]
   },
   {
diff --git a/docs/03-basic-langchain-agent.ipynb b/docs/03-basic-langchain-agent.ipynb
@@ -52,7 +52,7 @@
       "outputs": [],
       "source": [
         "!pip install -qU \\\n",
-        "    \"semantic-router==0.1.1\" \\\n",
+        "    \"semantic-router==0.1.2\" \\\n",
         "    langchain==0.0.352 \\\n",
         "    openai>=1.6.1"
       ]
diff --git a/docs/06-threshold-optimization.ipynb b/docs/06-threshold-optimization.ipynb
@@ -37,7 +37,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install -qU \"semantic-router==0.1.1\""
+        "!pip install -qU \"semantic-router==0.1.2\""
       ]
     },
     {
diff --git a/docs/08-async-dynamic-routes.ipynb b/docs/08-async-dynamic-routes.ipynb
@@ -78,7 +78,7 @@
    "outputs": [],
    "source": [
     "!pip install -qU \\\n",
-    "    \"semantic-router==0.1.1\" \\\n",
+    "    \"semantic-router==0.1.2\" \\\n",
     "    tzdata"
    ]
   },
diff --git a/docs/09-route-filter.ipynb b/docs/09-route-filter.ipynb
@@ -53,7 +53,7 @@
       },
       "outputs": [],
       "source": [
-        "!pip install -qU \"semantic-router==0.1.1\""
+        "!pip install -qU \"semantic-router==0.1.2\""
       ]
     },
     {
diff --git a/docs/examples/hybrid-router.ipynb b/docs/examples/hybrid-router.ipynb
@@ -171,10 +171,10 @@
     "from semantic_router.routers import HybridRouter\n",
     "\n",
     "router = HybridRouter(\n",
-    "    encoder=dense_encoder, \n",
-    "    sparse_encoder=sparse_encoder, \n",
+    "    encoder=dense_encoder,\n",
+    "    sparse_encoder=sparse_encoder,\n",
     "    routes=routes,\n",
-    "    alpha=0.5  # Balance between dense (0) and sparse (1) embeddings\n",
+    "    alpha=0.5,  # Balance between dense (0) and sparse (1) embeddings\n",
     ")"
    ]
   },
diff --git a/docs/indexes/local.ipynb b/docs/indexes/local.ipynb
@@ -13,7 +13,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install -qU \"semantic-router==0.1.1\""
+    "!pip install -qU \"semantic-router==0.1.2\""
    ]
   },
   {
diff --git a/docs/indexes/pinecone-local.ipynb b/docs/indexes/pinecone-local.ipynb
@@ -50,7 +50,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install -qU \"semantic-router[pinecone]==0.1.1\""
+    "!pip install -qU \"semantic-router[pinecone]==0.1.2\""
    ]
   },
   {
diff --git a/docs/indexes/pinecone-sync-routes.ipynb b/docs/indexes/pinecone-sync-routes.ipynb
@@ -6,7 +6,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install -qU \"semantic-router[pinecone]==0.1.1\""
+    "!pip install -qU \"semantic-router[pinecone]==0.1.2\""
    ]
   },
   {
diff --git a/docs/indexes/pinecone.ipynb b/docs/indexes/pinecone.ipynb
@@ -6,7 +6,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install -qU \"semantic-router[pinecone]==0.1.1\""
+    "!pip install -qU \"semantic-router[pinecone]==0.1.2\""
    ]
   },
   {
diff --git a/docs/indexes/pinecone_async.ipynb b/docs/indexes/pinecone_async.ipynb
@@ -6,7 +6,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install -qU \"semantic-router[pinecone]==0.1.1\""
+    "!pip install -qU \"semantic-router[pinecone]==0.1.2\""
    ]
   },
   {
diff --git a/docs/user-guide/features/dynamic-routes.md b/docs/user-guide/features/dynamic-routes.md
@@ -91,7 +91,7 @@ Then add the new route to a router.
 
 ```python
 !pip install tzdata
-!pip install -qU semantic-router==0.1.1
+!pip install -qU semantic-router==0.1.2
 ```
 
 ### Initializing Routes and SemanticRouter
diff --git a/docs/user-guide/features/threshold-optimization.md b/docs/user-guide/features/threshold-optimization.md
@@ -7,7 +7,7 @@ Given that this one `score_threshold` parameter can define the choice of a route
 [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/aurelio-labs/semantic-router/blob/main/docs/06-threshold-optimization.ipynb) [![Open nbviewer](https://raw.githubusercontent.com/pinecone-io/examples/master/assets/nbviewer-shield.svg)](https://nbviewer.org/github/aurelio-labs/semantic-router/blob/main/docs/06-threshold-optimization)
 
 ```python
-!pip install -qU "semantic-router==0.1.1"
+!pip install -qU "semantic-router==0.1.2"
 ```
 
 ## Define SemanticRouter
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "semantic-router"
-version = "0.1.1"
+version = "0.1.2"
 description = "Super fast semantic router for AI decision making"
 authors = [{ name = "Aurelio AI", email = "hello@aurelio.ai" }]
 requires-python = ">=3.9,<3.14"
@@ -48,6 +48,19 @@ postgres = []
 fastembed = ["fastembed>=0.3.0,<0.4 ; python_version < '3.13'"]
 docs = ["pydoc-markdown>=4.8.2 ; python_version < '3.12'"]
 cohere = ["cohere>=5.9.4,<6.00"]
+dev = [
+    "ipykernel>=6.25.0,<7",
+    "ruff>=0.1.5,<0.2",
+    "pytest~=8.2",
+    "pytest-mock>=3.12.0,<4",
+    "pytest-cov>=4.1.0,<5",
+    "pytest-xdist>=3.5.0,<4",
+    "pytest-asyncio>=0.24.0,<0.25",
+    "mypy>=1.7.1,<2",
+    "types-pyyaml>=6.0.12.12,<7",
+    "types-requests>=2.31.0,<3",
+    "black[jupyter]>=23.1.0",
+]
 all = [
     "semantic-router[local]",
     "semantic-router[pinecone]",
@@ -71,21 +84,6 @@ torchvision = [
     { index = "pytorch-cpu", marker = "python_version < '3.13' or sys_platform != 'darwin'" },
 ]
 
-[dependency-groups]
-dev = [
-    "ipykernel>=6.25.0,<7",
-    "ruff>=0.1.5,<0.2",
-    "pytest~=8.2",
-    "pytest-mock>=3.12.0,<4",
-    "pytest-cov>=4.1.0,<5",
-    "pytest-xdist>=3.5.0,<4",
-    "pytest-asyncio>=0.24.0,<0.25",
-    "mypy>=1.7.1,<2",
-    "types-pyyaml>=6.0.12.12,<7",
-    "types-requests>=2.31.0,<3",
-    "black[jupyter]>=23.1.0",
-]
-
 [tool.hatch.build.targets.sdist]
 include = ["semantic_router"]
 
diff --git a/semantic_router/__init__.py b/semantic_router/__init__.py
@@ -3,4 +3,4 @@
 
 __all__ = ["SemanticRouter", "HybridRouter", "Route", "RouterConfig"]
 
-__version__ = "0.1.1"
+__version__ = "0.1.2"
diff --git a/semantic_router/routers/hybrid.py b/semantic_router/routers/hybrid.py
@@ -349,7 +349,9 @@ def __call__(
         top_class, top_class_scores = self._semantic_classify(
             query_results=query_results
         )
-        passed = self._pass_threshold(top_class_scores, self.score_threshold)
+        route = self.check_for_matching_routes(top_class)
+        passed = self._check_threshold(top_class_scores, route)
+
         if passed:
             return RouteChoice(name=top_class, similarity_score=max(top_class_scores))
         else:
@@ -450,6 +452,7 @@ def fit(
                 if not isinstance(self.sparse_encoder, AsymmetricSparseMixin)
                 else self.sparse_encoder.encode_queries(X[i : i + batch_size])
             )
+
             Xq_d.extend(emb_d)
             Xq_s.extend(emb_s)
         # initial eval (we will iterate from here)
diff --git a/uv.lock b/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -53,7 +53,7 @@`
`53`	`53`	`},`
`54`	`54`	`"outputs": [],`
`55`	`55`	`"source": [`
`56`		`- "!pip install -qU \"semantic-router==0.1.1\""`
	`56`	`+ "!pip install -qU \"semantic-router==0.1.2\""`
`57`	`57`	`]`
`58`	`58`	`},`
`59`	`59`	`{`
Original file line number	Diff line number	Diff line change
`@@ -46,7 +46,7 @@`
`46`	`46`	`},`
`47`	`47`	`"outputs": [],`
`48`	`48`	`"source": [`
`49`		`- "!pip install -qU \"semantic-router==0.1.1\""`
	`49`	`+ "!pip install -qU \"semantic-router==0.1.2\""`
`50`	`50`	`]`
`51`	`51`	`},`
`52`	`52`	`{`
Original file line number	Diff line number	Diff line change
`@@ -95,7 +95,7 @@`
`95`	`95`	`],`
`96`	`96`	`"source": [`
`97`	`97`	`"!pip install tzdata\n",`
`98`		`- "!pip install -qU \"semantic-router==0.1.1\""`
	`98`	`+ "!pip install -qU \"semantic-router==0.1.2\""`
`99`	`99`	`]`
`100`	`100`	`},`
`101`	`101`	`{`
Original file line number	Diff line number	Diff line change
`@@ -52,7 +52,7 @@`
`52`	`52`	`"outputs": [],`
`53`	`53`	`"source": [`
`54`	`54`	`"!pip install -qU \\\n",`
`55`		`- " \"semantic-router==0.1.1\" \\\n",`
	`55`	`+ " \"semantic-router==0.1.2\" \\\n",`
`56`	`56`	`" langchain==0.0.352 \\\n",`
`57`	`57`	`" openai>=1.6.1"`
`58`	`58`	`]`
Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@`
`37`	`37`	`},`
`38`	`38`	`"outputs": [],`
`39`	`39`	`"source": [`
`40`		`- "!pip install -qU \"semantic-router==0.1.1\""`
	`40`	`+ "!pip install -qU \"semantic-router==0.1.2\""`
`41`	`41`	`]`
`42`	`42`	`},`
`43`	`43`	`{`
Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,7 @@`
`78`	`78`	`"outputs": [],`
`79`	`79`	`"source": [`
`80`	`80`	`"!pip install -qU \\\n",`
`81`		`- " \"semantic-router==0.1.1\" \\\n",`
	`81`	`+ " \"semantic-router==0.1.2\" \\\n",`
`82`	`82`	`" tzdata"`
`83`	`83`	`]`
`84`	`84`	`},`
Original file line number	Diff line number	Diff line change
`@@ -13,7 +13,7 @@`
`13`	`13`	`"metadata": {},`
`14`	`14`	`"outputs": [],`
`15`	`15`	`"source": [`
`16`		`- "!pip install -qU \"semantic-router==0.1.1\""`
	`16`	`+ "!pip install -qU \"semantic-router==0.1.2\""`
`17`	`17`	`]`
`18`	`18`	`},`
`19`	`19`	`{`
Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@`
`50`	`50`	`"metadata": {},`
`51`	`51`	`"outputs": [],`
`52`	`52`	`"source": [`
`53`		`- "!pip install -qU \"semantic-router[pinecone]==0.1.1\""`
	`53`	`+ "!pip install -qU \"semantic-router[pinecone]==0.1.2\""`
`54`	`54`	`]`
`55`	`55`	`},`
`56`	`56`	`{`