add todo and rename files

JustinHsu1019 · Nov 10, 2024 · e24908b · e24908b
1 parent 1eb9275
commit e24908b
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 2 deletions.
diff --git a/src/data/conbine_result.py → src/data/conbine_readpdf_result.py b/src/data/conbine_result.py → src/data/conbine_readpdf_result.py
diff --git a/src/data/read_data_noocr.py → src/data/read_pdf_noocr.py b/src/data/read_data_noocr.py → src/data/read_pdf_noocr.py
diff --git a/src/data/ocr_process.py → src/data/read_pdf_ocr.py b/src/data/ocr_process.py → src/data/read_pdf_ocr.py
diff --git a/src/flask_app.py b/src/flask_app.py
@@ -6,8 +6,7 @@
 from flask_limiter.util import get_remote_address
 from flask_restx import Api, Resource, fields
 
-# from utils.weaviate_op import search_do
-from utils.weaviatexreranker import search_do
+from utils.retrieval_agent import search_do
 from werkzeug.security import check_password_hash, generate_password_hash
 
 config, logger, CONFIG_PATH = config_log.setup_config_and_logging()
@@ -61,6 +60,7 @@ def get(self):
         return response
 
 
+# TODO: Modify the output format for general RAG purposes
 @ns.route('/chat')
 class ChatBot(Resource):
     @api.doc('chat_bot')
@@ -70,8 +70,11 @@ def post(self):
         source = request.json.get('source')
         question = request.json.get('query')
         category = request.json.get('category')
+
+        # for alpha testing
         # alpha = request.json.get('alpha')
 
+        # input template
         # {
         # "qid": 1,
         # "source": [442, 115, 440, 196, 431, 392, 14, 51],

diff --git a/src/utils/weaviatexreranker.py → src/utils/retrieval_agent.py b/src/utils/weaviatexreranker.py → src/utils/retrieval_agent.py