From 8405e0597ed6a48fd416302a1719f1cf4753a717 Mon Sep 17 00:00:00 2001 From: FacerAin Date: Thu, 7 Dec 2023 02:29:39 +0900 Subject: [PATCH] feat: add test to FakeRetriever --- tests/test_client.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/tests/test_client.py b/tests/test_client.py index bc5e072..a21cd07 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -6,6 +6,15 @@ 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) @@ -13,11 +22,8 @@ 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)