initial commit

src/main.py

@@ -9,28 +9,41 @@ from .parameters import *
 from fastapi import FastAPI, Header, HTTPException, BackgroundTasks
 from fastapi.responses import FileResponse
 
+import logging
+import sys
+
+
+
 
 app = FastAPI()
 
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+logger.addHandler(logging.StreamHandler(sys.stdout))
+
 vector_db = VectorDB(emb_model, db_location, full_actions_list_file_path, num_sub_vectors, batch_size)
 open_ai_connector = OpenAIConnector()
 
 @app.get("/find-action")
 async def find_action(query: str):
-
+    #logging.basicConfig(filename='myapp.log', level=logging.INFO)
+    logger.info('Started')
+    #print('start action')
     #data = vector_db.get_embedding_db_as_pandas()
     #print(data)
 
 
     prefiltered_names, prefiltered_descriptions = vector_db.retrieve_prefiltered_hits(query, K)
 
-    print('prefiltered list')
-    print(prefiltered_names)
+    logger.info('prefiltered list')
+    #print(prefiltered_names)
 
-    print('start query openAI')
+    logger.info('start query openAI')
     response = open_ai_connector.query_open_ai(query, prefiltered_names, prefiltered_descriptions)
-    print(response)
+    logger.info(response)
 
+    logger.info('Finished')
     return {'success': True, 'query': query, 'response': response}