cleanup requirements, fix api

api.py

@@ -1,3 +1,4 @@
+from bson import ObjectId
 from fastapi import FastAPI, Response, Body, Security
 from fastapi.security import APIKeyHeader
 from pydantic import BaseModel, model_validator
@@ -85,7 +86,7 @@ async def qa(api_key: str = Security(api_key_header), data: QModel = Body(...)):
             "rewrite_prompt_id": rewrite_prompt_id,
             "check_result": check_result,
             "temperature": data.temperature,
-            "retrieve_document_count": data.retrieval_count,
+            "retrieve_document_count": str(data.retrieval_count),
         }
     )
 

conversation/conversation_store.py

@@ -1,5 +1,6 @@
 import os
 
+from bson import ObjectId
 from dotenv import load_dotenv
 from pymongo import MongoClient
 
@@ -52,7 +53,7 @@ class ConversationStore:
 
     storage = ConversationStorage(URI, DB_NAME, COLLECTION_NAME)
 
-    def save_content(self, q, a=None, sources: list[str] = None, params: dict[str, str] = None):
+    def save_content(self, q, a=None, sources: list[str] = None, params: dict[str, str] = None) -> ObjectId:
         return self.storage.store_document_mongodb(
             Conversation(
                 conversation=[
@@ -64,7 +65,7 @@ class ConversationStore:
                 ],
                 params=params
             )
-        )
+        ).__str__()
 
     def get_all(self):
         return self.storage.get_all()

requirements.txt

@@ -8,14 +8,14 @@ langchain-google-genai
 langchain-together
 langchain-mongodb
 
-fitz
-pypdf
-tools
+#fitz
+#pypdf
+#tools
 python-dotenv
 pymongo
 pydantic
-chromadb
-bs4
+#chromadb
+#bs4
 fastapi
 gptcache
 
@@ -23,5 +23,5 @@ fastapi
 requests
 uvicorn[standard]
 
-sentence-transformers
-text_generation
+#sentence-transformers
+#text_generation