chore: update test pinecone to mock

FacerAin · Dec 6, 2023 · 985516a · 985516a
1 parent 6925fc5
commit 985516a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -10,7 +10,7 @@
 
 
 def test_completion(monkeypatch):
-    def mockreturn(a, b):
+    def mockreturn(*args, **kwargs):
         return "Hello"
 
     monkeypatch.setattr(ExecutorAgent, "run", mockreturn)
@@ -22,10 +22,11 @@ def mockreturn(a, b):
 
 
 def test_similarity_search(monkeypatch):
-    def mockreturn(a, b):
+    def mockreturn(*args, **kwargs):
         return "Hello"
 
     monkeypatch.setattr(PineconeRetriever, "get_relevant_doc_string", mockreturn)
+    monkeypatch.setattr(PineconeRetriever, "get_pinecone_index", mockreturn)
     monkeypatch.setattr(pinecone, "init", mockreturn)
 
     response = client.get("/api/v1/chat/similarity-search?query=Hi")