adding filter

app.py

@@ -18,7 +18,7 @@ except Exception as e:
 def retrieve(
     query:str, 
     collection_name:str = None,
-    filter_metadata:list[dict] = None,
+    filter_metadata:dict = None,
 ) -> list:
     """
     Retrieve semantically similar documents from the vector database for MCP clients.
@@ -71,12 +71,16 @@ with gr.Blocks() as ui:
                 placeholder="country",
                 info="Filter by document source type (leave empty for all)"
             )
-            subtype_input = gr.Textbox(
+            sources_value = gr.Textbox(
                 label="Value in filter to be looked for(optional)", 
                 lines=1, 
                 placeholder="Ecuador, Guatemala",
                 info="Filter by document subtype (leave empty for all)"
             )
+            if sources_input & sources_value:
+                filter_metadata = {sources_input:sources_value}
+            else: 
+                filter_metadata = None
             
             submit_btn = gr.Button("Submit", variant="primary")
         
@@ -91,7 +95,7 @@ with gr.Blocks() as ui:
     # UI event handler
     submit_btn.click(
         fn=retrieve,
-        inputs=[query_input,collection_name],
+        inputs=[query_input,collection_name, filter_metadata],
         outputs=output,
         api_name="retrieve"
     )

utils/retriever.py

@@ -70,7 +70,7 @@ def get_vectorstore() -> VectorStoreInterface:
     return vectorstore
 
 def create_filter(
-    filter_metadata:list[Dict] = None,
+    filter_metadata:dict = None,
 ) -> Optional[rest.Filter]:
     """
     Create a Qdrant filter based on metadata criteria.
@@ -89,21 +89,21 @@ def create_filter(
     
     conditions = []
     logging.info(f"Defining filters for {filter_metadata}")
-    for condition in filter_metadata:
-        for key, val in condition:
-            if isinstance(val, str):
-                conditions.append(rest.FieldCondition(
-                key=f"metadata.{key}",
-                match=rest.MatchValue(value=val)
-                )
+    
+    for key, val in filter_metadata.items():
+        if isinstance(val, str):
+            conditions.append(rest.FieldCondition(
+            key=f"metadata.{key}",
+            match=rest.MatchValue(value=val)
             )
-            else:
-                conditions.append(
-            rest.FieldCondition(
-                key=f"metadata.{key}",
-                match=rest.MatchAny(any=val)
+        )
+        else:
+            conditions.append(
+        rest.FieldCondition(
+            key=f"metadata.{key}",
+            match=rest.MatchAny(any=val)
+        )
             )
-                )
     return conditions