Upload folder using huggingface_hub

sanatan_assistant.py

@@ -145,3 +145,44 @@ def query_by_metadata_field(
             response["documents"], response["metadatas"], response["ids"]
         )
     )
+
+def query_by_metadata_field(
+    collection_name: allowedCollections,
+    query: str,
+    metadata_field: str,
+    metadata_search_operator: Literal[
+        "$in", "$eq", "$gt", "$gte", "$lt", "$lte", "$ne", "$nin"
+    ],
+    metadata_value: Any,
+    n_results=5,
+):
+    """
+    Search a scripture collection by metadata. Do NOT use this for semantic search. Only use when a specific metadata field is provided.
+
+    Parameters:
+    - collection_name (str): The name of the scripture collection to search. ...
+    - query (str): The search query.
+    - metadata_field (str) : The name of the metadata field. e.g. azhwar_name
+    - metadata_search_operator (str) : The search operator e.g. $eq or $in
+    - metadata_value : Value to search for can be any primitive datatype like str or int (or a list[str] if metadata_search_operator = '$in'). for e.g. Thirumangai Azhwar or '2233' or 2233
+    - n_results (int): Number of results to return. Default is 5.
+
+    Returns:
+    - A list of matching results.
+    """
+    logger.info("Searching collection [%s] for [%s]", collection_name, query)
+    response = sanatanDatabase.search_by_metadata(
+        collection_name=collection_name,
+        query=query,
+        metadata_field=metadata_field,
+        metadata_search_operator=metadata_search_operator,
+        metadata_value=metadata_value,
+        n_results=n_results,
+    )
+
+    return "\n\n".join(
+        f"Document: {doc}\nMetadata: {meta}\nID: {id_}"
+        for doc, meta, id_ in zip(
+            response["documents"], response["metadatas"], response["ids"]
+        )
+    )