fixed source doc issue

app/api/v1/routes.py

@@ -212,16 +212,19 @@ async def query_document(
         # Extract source documents from results
         sources = []
         if hasattr(session.rag_service, 'result') and session.rag_service.result:
-            matches = session.rag_service.result.get('matches', [])
+            matches = session.rag_service.result
             for match in matches[:3]:  # Top 3 sources
-                metadata = match.get('metadata', {})
-                sources.append(SourceDocument(
-                    doc_id=metadata.get('doc_id', match.get('id', '')),
-                    page=metadata.get('page_no', metadata.get('page', 0)),
-                    text=metadata.get('text', ''),
-                    score=match.get('score', 0.0),
-                    metadata=metadata
+                metadata = match.metadata
+                score = session.rag_service.metadataservice.cosine_similarity(vec1=session.rag_service.embedding_model.embed_query(match.page_content), vec2=session.rag_service.embedding_model.embed_query(
+                    session.rag_service.query
                 ))
+                sources.append(SourceDocument(
+                            doc_id=metadata['doc_id'],
+                            page=metadata['page_no'],
+                            text=match.page_content,
+                            score=score,
+                            metadata=metadata
+                        ))
 
         return QueryResponse(
             session_id=session_id,

app/services/RAG_service.py

@@ -91,6 +91,7 @@ class RAGService:
 
     def create_query_embedding(self, query: str):
         print("[RAGService] Creating query embedding...")
+        self.query = query
         self.query_embedder = QueryEmbedding(query=query, embedding_model=self.embedding_model)
         self.query_embedding = self.query_embedder.get_embedding()
         print(f"[RAGService] Query embedding created: {self.query_embedding}")