From 77c9c0f06ada300f5067bef3d51361927090513d Mon Sep 17 00:00:00 2001 From: Rajat Ahuja Date: Tue, 20 May 2025 11:26:23 -0400 Subject: [PATCH] rm unused embedding store param --- src/deriver/tom/long_term.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/deriver/tom/long_term.py b/src/deriver/tom/long_term.py index c40fcbcf..54d791ea 100644 --- a/src/deriver/tom/long_term.py +++ b/src/deriver/tom/long_term.py @@ -9,7 +9,6 @@ from sentry_sdk.ai.monitoring import ai_track from src.utils import parse_xml_content from src.utils.model_client import ModelClient, ModelProvider -from .embeddings import CollectionEmbeddingStore # Configure logging logger = logging.getLogger(__name__) @@ -29,7 +28,6 @@ MAX_FACT_DISTANCE = 0.85 async def get_user_representation_long_term( chat_history: str, session_id: str, - embedding_store: CollectionEmbeddingStore, user_representation: str = "None", tom_inference: str = "None", facts: Optional[list[str]] = None,