Spaces:
Sleeping
Sleeping
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 |
-
|
| 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 |
-
|
| 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 |
-
|
| 42 |
-
|
| 43 |
# Format context, Use reranked results for context
|
| 44 |
-
|
| 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
|