diff --git a/src/codegate/pipeline/codegate_context_retriever/codegate.py b/src/codegate/pipeline/codegate_context_retriever/codegate.py index 9bbd8565..eafaa080 100644 --- a/src/codegate/pipeline/codegate_context_retriever/codegate.py +++ b/src/codegate/pipeline/codegate_context_retriever/codegate.py @@ -90,7 +90,8 @@ async def process( user_messages = re.sub(r"```.*?```", "", user_messages, flags=re.DOTALL) # Vector search to find bad packages - bad_packages = await storage_engine.search(query=user_messages, distance=0.8, limit=100) + storage_engine = StorageEngine() + searched_objects = await storage_engine.search(query=user_messages, distance=0.5, limit=100) # All bad packages all_bad_packages = bad_snippet_packages + bad_packages