Merge pull request #567 from aurelio-labs/james/threshold-set

jamescalam · web-flow · commit 199603cfe54d · 2025-03-21T19:13:33.000+04:00
fix: threshold set
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.3\""
+        "!pip install -qU \"semantic-router>=0.1.4\""
       ]
     },
     {
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.3\""
+        "!pip install -qU \"semantic-router>=0.1.4\""
       ]
     },
     {
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.3\""
+    "!pip install -qU \"semantic-router>=0.1.4\""
    ]
   },
   {
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.3\" \\\n",
+        "    \"semantic-router>=0.1.4\" \\\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.3\""
+        "!pip install -qU \"semantic-router>=0.1.4\""
       ]
     },
     {
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.3\" \\\n",
+    "    \"semantic-router>=0.1.4\" \\\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.3\""
+        "!pip install -qU \"semantic-router>=0.1.4\""
       ]
     },
     {
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.3\""
+    "!pip install -qU \"semantic-router>=0.1.4\""
    ]
   },
   {
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.3\""
+    "!pip install -qU \"semantic-router[pinecone]>=0.1.4\""
    ]
   },
   {
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.3\""
+    "!pip install -qU \"semantic-router[pinecone]>=0.1.4\""
    ]
   },
   {
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.3\""
+    "!pip install -qU \"semantic-router[pinecone]>=0.1.4\""
    ]
   },
   {
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.3\""
+    "!pip install -qU \"semantic-router[pinecone]>=0.1.4\""
    ]
   },
   {
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.3
+!pip install -qU semantic-router>=0.1.4
 ```
 
 ### 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.3"
+!pip install -qU "semantic-router>=0.1.4"
 ```
 
 ## Define SemanticRouter
diff --git a/docs/user-guide/guides/migration-to-v1.md b/docs/user-guide/guides/migration-to-v1.md
@@ -31,7 +31,7 @@ The v0.1 release of semantic router introduces several breaking changes to impro
   The `retrieve_multiple_routes` method has been removed. If you need similar functionality:
   
   - In versions 0.1.0-0.1.2: You can use the deprecated `_semantic_classify_multiple_routes` method
-  - In version 0.1.3+: Use the `__call__` or `acall` methods with appropriate `limit` parameter
+  - In version 0.1.3+ (0.1.4+ is recommended): Use the `__call__` or `acall` methods with appropriate `limit` parameter.
   
   ```python
   # Before (v0.0.x)
@@ -44,7 +44,7 @@ The v0.1 release of semantic router introduces several breaking changes to impro
   query_results = semantic_router._query(query_text)
   multiple_routes = semantic_router._semantic_classify_multiple_routes(query_results)
   
-  # After (v0.1.3+)
+  # After (v0.1.4+)
   semantic_router = SemanticRouter(encoder=encoder, routes=routes, auto_sync="local")
   # Return all routes that pass their score thresholds
   all_routes = semantic_router(query_text, limit=None)
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "semantic-router"
-version = "0.1.3"
+version = "0.1.4"
 description = "Super fast semantic router for AI decision making"
 authors = [{ name = "Aurelio AI", email = "hello@aurelio.ai" }]
 requires-python = ">=3.9,<3.14"
diff --git a/semantic_router/routers/base.py b/semantic_router/routers/base.py
@@ -611,7 +611,9 @@ def _pass_routes(
                 # if no route is found we cannot use it
                 continue
             if current_threshold := (
-                route.score_threshold if route.score_threshold else self.score_threshold
+                route.score_threshold
+                if route.score_threshold is not None
+                else self.score_threshold
             ):
                 # check if our route score exceeds the set threshold
                 passed = total_score >= current_threshold
@@ -694,7 +696,9 @@ async def _async_pass_routes(
                 # if no route is found we cannot use it
                 continue
             if current_threshold := (
-                route.score_threshold if route.score_threshold else self.score_threshold
+                route.score_threshold
+                if route.score_threshold is not None
+                else self.score_threshold
             ):
                 # check if our route score exceeds the set threshold
                 passed = total_score >= current_threshold
@@ -1600,6 +1604,8 @@ def set_threshold(self, threshold: float, route_name: str | None = None):
         if route_name is None:
             for route in self.routes:
                 route.score_threshold = threshold
+            # set the router threshold too
+            self.score_threshold = threshold
         else:
             route_get: Route | None = self.get(route_name)
             if route_get is not None:
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.3\""`
	`56`	`+ "!pip install -qU \"semantic-router>=0.1.4\""`
`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.3\""`
	`49`	`+ "!pip install -qU \"semantic-router>=0.1.4\""`
`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.3\""`
	`98`	`+ "!pip install -qU \"semantic-router>=0.1.4\""`
`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.3\" \\\n",`
	`55`	`+ " \"semantic-router>=0.1.4\" \\\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.3\""`
	`40`	`+ "!pip install -qU \"semantic-router>=0.1.4\""`
`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.3\" \\\n",`
	`81`	`+ " \"semantic-router>=0.1.4\" \\\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.3\""`
	`16`	`+ "!pip install -qU \"semantic-router>=0.1.4\""`
`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.3\""`
	`53`	`+ "!pip install -qU \"semantic-router[pinecone]>=0.1.4\""`
`54`	`54`	`]`
`55`	`55`	`},`
`56`	`56`	`{`
Original file line number	Diff line number	Diff line change
`@@ -6,7 +6,7 @@`
`6`	`6`	`"metadata": {},`
`7`	`7`	`"outputs": [],`
`8`	`8`	`"source": [`
`9`		`- "!pip install -qU \"semantic-router[pinecone]>=0.1.3\""`
	`9`	`+ "!pip install -qU \"semantic-router[pinecone]>=0.1.4\""`
`10`	`10`	`]`
`11`	`11`	`},`
`12`	`12`	`{`