Spaces:

awinml
/

2-qa-earnings-sentencewise

Build error

App Files Files Community

awinml commited on May 30, 2023

Commit

e0df9e7

•

1 Parent(s): eb6952b

Upload 17 files

Browse files

Files changed (1) hide show

utils/retriever.py +6 -6

utils/retriever.py CHANGED Viewed

@@ -37,7 +37,7 @@ def query_pinecone_sparse(
                     "Quarter": {"$in": ["Q1", "Q2", "Q3", "Q4"]},
                     "Ticker": {"$eq": ticker},
                     "QA_Flag": {"$eq": participant},
-                    '$and': filter_dict
                 },
                 include_metadata=True,
             )
@@ -59,7 +59,7 @@ def query_pinecone_sparse(
                     "Quarter": {"$eq": quarter},
                     "Ticker": {"$eq": ticker},
                     "QA_Flag": {"$eq": participant},
-                    '$and': filter_dict
                 },
                 include_metadata=True,
             )
@@ -74,7 +74,7 @@ def query_pinecone_sparse(
                 "Quarter": {"$eq": quarter},
                 "Ticker": {"$eq": ticker},
                 "QA_Flag": {"$eq": participant},
-                '$and': filter_dict
             },
             include_metadata=True,
         )
@@ -125,7 +125,7 @@ def query_pinecone(
                     "Quarter": {"$in": ["Q1", "Q2", "Q3", "Q4"]},
                     "Ticker": {"$eq": ticker},
                     "QA_Flag": {"$eq": participant},
-                    '$and': filter_dict
                 },
                 include_metadata=True,
             )
@@ -146,7 +146,7 @@ def query_pinecone(
                     "Quarter": {"$eq": quarter},
                     "Ticker": {"$eq": ticker},
                     "QA_Flag": {"$eq": participant},
-                    '$and': filter_dict
                 },
                 include_metadata=True,
             )
@@ -160,7 +160,7 @@ def query_pinecone(
                 "Quarter": {"$eq": quarter},
                 "Ticker": {"$eq": ticker},
                 "QA_Flag": {"$eq": participant},
-                '$and': filter_dict
             },
             include_metadata=True,
         )

                     "Quarter": {"$in": ["Q1", "Q2", "Q3", "Q4"]},
                     "Ticker": {"$eq": ticker},
                     "QA_Flag": {"$eq": participant},
+                    "Keywords": {"$in": keywords}
                 },
                 include_metadata=True,
             )
                     "Quarter": {"$eq": quarter},
                     "Ticker": {"$eq": ticker},
                     "QA_Flag": {"$eq": participant},
+                    "Keywords": {"$in": keywords}
                 },
                 include_metadata=True,
             )
                 "Quarter": {"$eq": quarter},
                 "Ticker": {"$eq": ticker},
                 "QA_Flag": {"$eq": participant},
+                "Keywords": {"$in": keywords}
             },
             include_metadata=True,
         )
                     "Quarter": {"$in": ["Q1", "Q2", "Q3", "Q4"]},
                     "Ticker": {"$eq": ticker},
                     "QA_Flag": {"$eq": participant},
+                    "Keywords": {"$in": keywords}
                 },
                 include_metadata=True,
             )
                     "Quarter": {"$eq": quarter},
                     "Ticker": {"$eq": ticker},
                     "QA_Flag": {"$eq": participant},
+                    "Keywords": {"$in": keywords}
                 },
                 include_metadata=True,
             )
                 "Quarter": {"$eq": quarter},
                 "Ticker": {"$eq": ticker},
                 "QA_Flag": {"$eq": participant},
+                "Keywords": {"$in": keywords}
             },
             include_metadata=True,
         )