v0.1.1

src/main.py

@@ -16,7 +16,9 @@ from typing import List, Dict, Any, Generator, Optional, cast, Callable
 from chromaIntf import ChromaIntf
 import baseInfra.dropbox_handler as dbh
 import traceback
+from pathlib import Path
 
+logger=logging.getLogger("root")
 chromaIntf=ChromaIntf()
 
 class PathRequest(BaseModel):
@@ -66,17 +68,24 @@ async def get_matching_docs(inStr: str, kwargs: Dict [Any, Any] ,background_task
     TODO: Add parameter for type of query and number of docs to return
     TODO: Add parameter to return the source information as well
     """
-    print("Reaching get matchin docs ")
-    return await chromaIntf.getRelevantDocs(inStr,kwargs)
+    logger.info("================================================")
+    logger.info("Received new request at /getMatchingDocs:"+inStr)
+    retVal=await chromaIntf.getRelevantDocs(inStr,kwargs)
+    logger.info("Returning:"+retVal)
+    logger.info("================================================")
+    return retVal
 
 @app.post(api_base+"/addTextDocument")
 async def add_text_document(inDoc: DocWithMeta ) -> Any:
     """
      Add text and metadata to the db
     """
-    print("Received request for")
-    print(inDoc) 
-    return await chromaIntf.addText(inDoc.text,inDoc.metadata)
+    logger.info("================================================")
+    logger.info("Received new request at /addTextDocuments:"+inDoc)
+    retVal= await chromaIntf.addText(inDoc.text,inDoc.metadata)
+    logger.info("Returning:"+retVal)
+    logger.info("================================================")
+    return retVal
 
 @app.get(api_base+"/persist")
 async def persist_db():
@@ -111,4 +120,5 @@ async def list_docs():
 print(__name__)
 
 if __name__ == '__main__' or __name__ == "src.main":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(f"{Path(__file__).stem}:app", host="0.0.0.0", port=8000,workers=4)
+    #uvicorn.run(app, host="0.0.0.0", port=8000)