chore: update pinecone test mock

FacerAin · Dec 6, 2023 · 6925fc5 · 6925fc5
1 parent 665ef94
commit 6925fc5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -1,3 +1,4 @@
+import pinecone
 import pytest
 from fastapi.testclient import TestClient
 
@@ -25,6 +26,8 @@ def mockreturn(a, b):
         return "Hello"
 
     monkeypatch.setattr(PineconeRetriever, "get_relevant_doc_string", mockreturn)
+    monkeypatch.setattr(pinecone, "init", mockreturn)
+
     response = client.get("/api/v1/chat/similarity-search?query=Hi")
     assert response.status_code == 200
     assert isinstance(response.text, str)