Add value filtering

config.yaml

@@ -20,7 +20,11 @@ properties:
   filter_key:
     title: "Filter Key"
     type: string
-    description: "If you need to narrow down the document set further, provide a key name you want to filter by"
+    description: "If you need to narrow down the document set further, provide a key name you want to filter by. If value isn't also specified, then it will filter on any documents that have that key."
+  filter_value:
+    title: "Filter Value"
+    type: string
+    description: "If you want to provide a specific key/value pair to narrow on, provide the value here and the key in filter key. "
   prompt:
     title: "Prompt"
     type: string

main.py

@@ -16,11 +16,17 @@ def main(self):
         prompt = self.data.get("prompt")
         limiter = self.data.get("limiter")
         filter_key = self.data.get("filter_key")
+        filter_value = self.data.get("filter_value")
 
         if filter_key is not None:
-            documents = self.client.documents.search(
-                f"+project:{proj_id} -data_{key_name}:* data_{filter_key}:*"
-            )
+            if filter_value is not None:
+                documents = self.client.documents.search(
+                    f"+project:{proj_id} -data_{key_name}:* data_{filter_key}:{filter_value}"
+                )
+            else:
+                documents = self.client.documents.search(
+                    f"+project:{proj_id} -data_{key_name}:* data_{filter_key}:*"
+                )
         else:
             documents = self.client.documents.search(
                 f"+project:{proj_id} -data_{key_name}:*"