From 985b7d1057ab006f5c67710ec20b5e9a14608b04 Mon Sep 17 00:00:00 2001
From: Bogdan Buduroiu <bogdan@buduroiu.com>
Date: Tue, 18 Feb 2025 12:25:43 +0800
Subject: [PATCH] fix: base router encoding type
---
semantic_router/routers/base.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/semantic_router/routers/base.py b/semantic_router/routers/base.py
index c63d6187..89cf18e1 100644
--- a/semantic_router/routers/base.py
+++ b/semantic_router/routers/base.py
@@ -562,7 +562,7 @@ class BaseRouter(BaseModel):
if vector is None:
if text is None:
raise ValueError("Either text or vector must be provided")
- vector = self._encode(text=[text])
+ vector = self._encode(text=[text], input_type="queries")
# convert to numpy array if not already
vector = xq_reshape(vector)
# get scores and routes
@@ -635,7 +635,7 @@ class BaseRouter(BaseModel):
if vector is None:
if text is None:
raise ValueError("Either text or vector must be provided")
- vector = await self._async_encode(text=[text])
+ vector = await self._async_encode(text=[text], input_type="queries")
# convert to numpy array if not already
vector = xq_reshape(vector)
# get scores and routes
--
GitLab