👀 Add tests to API

README.md

@@ -44,10 +44,11 @@ app = create_app(QnABot("./mydata"))
 - [x] Local data source (directory of documents) or S3 data source
 - [x] FAISS for storing vectors / index
 - [x] Expose bot over API using FastAPI
+- [ ] Integration with [guardrails](https://github.com/ShreyaR/guardrails)
+- [ ] Integration with [GPTCache](https://github.com/zilliztech/GPTCache)
 - [ ] Support for other vector databases (e.g. Weaviate, Pinecone)
 - [ ] Customise prompt
 - [ ] Support for LLaMA model
-- [ ] Support for Anthropic models
 - [ ] CLI / UI
 
 ### Here's how it works

qnabot/tests/test_api.py

@@ -0,0 +1,21 @@
+import json
+from fastapi.testclient import TestClient
+from qnabot import QnABot, create_app
+
+bot = QnABot(directory="./examples/files")
+app = create_app(bot)
+
+client = TestClient(app)
+
+
+def test_successful_response():
+    response = client.get("/v1/ask/What is your name?")
+    assert response.status_code == 200
+    assert "answer" in response.json()
+    assert isinstance(response.json()["answer"], str)
+
+
+def test_missing_question_parameter():
+    response = client.get("/v1/ask/")
+    assert response.status_code == 404
+    assert response.json() == {"detail": "Not Found"}