From 5115a12a7c611305825063b662fc8222535ded31 Mon Sep 17 00:00:00 2001
From: Vits <vittorio.mayellaro.dev@gmail.com>
Date: Wed, 31 Jul 2024 11:23:45 +0200
Subject: [PATCH] Merge with main
---
semantic_router/index/pinecone.py | 7 +------
semantic_router/layer.py | 3 ---
2 files changed, 1 insertion(+), 9 deletions(-)
diff --git a/semantic_router/index/pinecone.py b/semantic_router/index/pinecone.py
index e88fa148..1908d905 100644
--- a/semantic_router/index/pinecone.py
+++ b/semantic_router/index/pinecone.py
@@ -11,7 +11,6 @@ from pydantic.v1 import BaseModel, Field
from semantic_router.index.base import BaseIndex
from semantic_router.utils.logger import logger
-from semantic_router.route import Route
def clean_route_name(route_name: str) -> str:
@@ -208,7 +207,6 @@ class PineconeIndex(BaseIndex):
self.index = self._init_index(force_create=True)
remote_routes = self.get_routes()
-
remote_dict: dict = {route: set() for route, _ in remote_routes}
for route, utterance in remote_routes:
remote_dict[route].add(utterance)
@@ -217,9 +215,6 @@ class PineconeIndex(BaseIndex):
for route, utterance in zip(local_route_names, local_utterances):
local_dict[route].add(utterance)
- logger.info(f"Local routes: {local_dict}")
- logger.info(f"Remote routes: {remote_dict}")
-
all_routes = set(remote_dict.keys()).union(local_dict.keys())
routes_to_add = []
@@ -324,7 +319,7 @@ class PineconeIndex(BaseIndex):
for i in range(0, len(vectors_to_upsert), batch_size):
batch = vectors_to_upsert[i : i + batch_size]
self._batch_upsert(batch)
-
+
def _remove_and_sync(self, routes_to_delete: dict):
for route, utterances in routes_to_delete.items():
remote_routes = self._get_routes_with_ids(route_name=route)
diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index 8888460d..c94d5731 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -497,9 +497,6 @@ class RouteLayer:
dimensions=len(self.encoder(["dummy"])[0])
)
- logger.info(f"ROUTES TO ADD: {(routes_to_add)}")
- logger.info(f"ROUTES TO DELETE: {(routes_to_delete)}")
-
layer_routes = [
Route(name=route, utterances=layer_routes_dict[route])
for route in layer_routes_dict.keys()
--
GitLab