VcRlAgent commited on
Commit
24e6b59
Β·
1 Parent(s): 92f791e

Reranking back

Browse files
app/main.py CHANGED
@@ -2,7 +2,7 @@
2
  from fastapi import FastAPI
3
  from fastapi.middleware.cors import CORSMiddleware
4
  from app.config import settings
5
- from app.routes import ingest_routes, ask_routes, metrics_routes
6
  from app.services.vector_store import vector_store
7
  from app.utils.logger import setup_logger
8
 
@@ -27,7 +27,7 @@ app.add_middleware(
27
  app.include_router(ingest_routes.router, prefix="/api", tags=["Ingestion"])
28
  app.include_router(ask_routes.router, prefix="/api", tags=["Query"])
29
  app.include_router(metrics_routes.router, prefix="/api", tags=["Metrics"])
30
- #app.include_router(debug_routes.router, prefix="/api", tags=["Debug"])
31
 
32
  logger.info("βœ… Routers initialized ::")
33
  for route in app.routes:
 
2
  from fastapi import FastAPI
3
  from fastapi.middleware.cors import CORSMiddleware
4
  from app.config import settings
5
+ from app.routes import ingest_routes, ask_routes, metrics_routes, debug_routes
6
  from app.services.vector_store import vector_store
7
  from app.utils.logger import setup_logger
8
 
 
27
  app.include_router(ingest_routes.router, prefix="/api", tags=["Ingestion"])
28
  app.include_router(ask_routes.router, prefix="/api", tags=["Query"])
29
  app.include_router(metrics_routes.router, prefix="/api", tags=["Metrics"])
30
+ app.include_router(debug_routes.router, prefix="/api", tags=["Debug"])
31
 
32
  logger.info("βœ… Routers initialized ::")
33
  for route in app.routes:
app/routes/ask_routes.py CHANGED
@@ -4,7 +4,7 @@ from fastapi import APIRouter, HTTPException
4
  from app.models.jira_schema import QueryRequest, QueryResponse
5
  from app.services.retriever import retriever
6
  from app.services.generator import generator
7
- #from app.services.reranker import reranker
8
  from app.utils.response_builder import build_query_response, extract_chart_intent
9
  from app.utils.logger import setup_logger
10
  from collections import Counter
@@ -35,13 +35,13 @@ async def ask_question(request: QueryRequest):
35
 
36
 
37
  # Format context
38
- context = retriever.format_context(results)
39
 
40
  # 🧠 Re-rank results
41
- #logger.info("[RERANKER] Starting re-ranking process....")
42
- #reranked_results = reranker.rerank(request.query, results, top_k=5)
43
  # Format context, Use reranked results for context
44
- #context = retriever.format_context(reranked_results) # ## Bug IO Error
45
 
46
  # Generate answer
47
  answer = generator.generate_rag_response(request.query, context)
 
4
  from app.models.jira_schema import QueryRequest, QueryResponse
5
  from app.services.retriever import retriever
6
  from app.services.generator import generator
7
+ from app.services.reranker import reranker
8
  from app.utils.response_builder import build_query_response, extract_chart_intent
9
  from app.utils.logger import setup_logger
10
  from collections import Counter
 
35
 
36
 
37
  # Format context
38
+ #context = retriever.format_context(results)
39
 
40
  # 🧠 Re-rank results
41
+ logger.info("[RERANKER] Starting re-ranking process....")
42
+ reranked_results = reranker.rerank(request.query, results, top_k=5)
43
  # Format context, Use reranked results for context
44
+ context = retriever.format_context(reranked_results) # ## Bug IO Error
45
 
46
  # Generate answer
47
  answer = generator.generate_rag_response(request.query, context)
app/routes/{debug_routes.py.bak β†’ debug_routes.py} RENAMED
File without changes
app/services/{reranker.py.bak β†’ reranker.py} RENAMED
File without changes