feat: add test to FakeRetriever

tests/test_client.py

@@ -6,18 +6,24 @@
 from app.agent.retriever import PineconeRetriever
 from app.main import app
 
+
+class FakeRetriever(object):
+    def __init__(self, *args, **kwargs):
+        pass
+
+    def get_relevant_doc_string(self, *args, **kwargs):
+        return "Hello"
+
+
 client = TestClient(app)
 
 
 def test_completion(monkeypatch):
     def mockreturn(*args, **kwargs):
         return "Hello"
 
-    def mockinitreturn(*args, **kwargs):
-        return None
-
     monkeypatch.setattr(ExecutorAgent, "run", mockreturn)
-    monkeypatch.setattr(PineconeRetriever, "__init__", mockinitreturn)
+    monkeypatch.setattr("app.agent.retriever.PineconeRetriever", FakeRetriever)
 
     req_body = {"query": "Hi"}
     response = client.post("/api/v1/chat/completion", json=req_body)