From fc9f41c436af13cf26a37a83d4a7f49494a46915 Mon Sep 17 00:00:00 2001 From: Pankaj Telang Date: Tue, 10 Dec 2024 17:15:14 +0000 Subject: [PATCH] Modify package filtering to use to case insensitve matching --- src/codegate/pipeline/codegate_context_retriever/codegate.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/codegate/pipeline/codegate_context_retriever/codegate.py b/src/codegate/pipeline/codegate_context_retriever/codegate.py index 8341c144..fbd86bd6 100644 --- a/src/codegate/pipeline/codegate_context_retriever/codegate.py +++ b/src/codegate/pipeline/codegate_context_retriever/codegate.py @@ -86,6 +86,7 @@ async def process( # Extract packages from the user message last_user_message_str, last_user_idx = last_user_message packages = await self.__lookup_packages(last_user_message_str, context) + packages = [pkg.lower() for pkg in packages] # If user message does not reference any packages, then just return if len(packages) == 0: @@ -103,7 +104,7 @@ async def process( # since Weaviate performs substring match in the filter. updated_searched_objects = [] for searched_object in searched_objects: - if searched_object.properties["name"] in packages: + if searched_object.properties["name"].lower() in packages: updated_searched_objects.append(searched_object) searched_objects = updated_searched_objects