update model in return chunk_id in API

config.py

@@ -7,7 +7,7 @@ load_dotenv()  # Tải biến môi trường từ file .env
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "YOUR_API_KEY_HERE")
 
 # Embedding model
-EMBEDDING_MODEL = "intfloat/multilingual-e5-base"
+EMBEDDING_MODEL = "linhha2705/multilingual-e5-age"
 
 # ChromaDB settings - sử dụng absolute path
 CHROMA_PERSIST_DIRECTORY = os.path.join(os.getcwd(), "chroma_db")

core/rag_pipeline.py

@@ -65,7 +65,8 @@ class RAGPipeline:
                 
                 # Thêm source reference
                 source_info = {
-                    "title": metadata.get('title', metadata.get('chapter', 'Tài liệu dinh dưỡng')),
+                    "chunk_id": metadata.get('chunk_id', 'unknown'),
+                    "title": metadata.get('title', metadata.get('chapter', 'Tài liệu dinh dưỡng')),  # Giữ title nếu cần
                     "pages": metadata.get('pages'),
                     "content_type": metadata.get('content_type', 'text')
                 }
@@ -126,7 +127,9 @@ class RAGPipeline:
             
             # Thêm thông tin metadata
             context_str = f"[Tài liệu {i}]"
-            if metadata.get('title'):
+            if metadata.get('chunk_id'):
+                context_str += f" - ID: {metadata['chunk_id']}"
+            elif metadata.get('title'):
                 context_str += f" - {metadata['title']}"
             if metadata.get('pages'):
                 context_str += f" (Trang {metadata['pages']})"