VcRlAgent commited on
Commit
701643e
·
1 Parent(s): 93102ce

Initial WorkWise backend merged main.py -> app.py

Browse files
Files changed (2) hide show
  1. app/routes/ask_routes.py +2 -1
  2. requirements.txt +1 -0
app/routes/ask_routes.py CHANGED
@@ -6,12 +6,13 @@ from app.services.generator import generator
6
  from app.utils.response_builder import build_query_response, extract_chart_intent
7
  from app.utils.logger import setup_logger
8
  from collections import Counter
 
9
 
10
  logger = setup_logger(__name__)
11
  router = APIRouter()
12
 
13
- @router.post("/ask", response_model=QueryResponse)
14
  @spaces.GPU
 
15
  async def ask_question(request: QueryRequest):
16
  """
17
  Answer natural language questions using RAG
 
6
  from app.utils.response_builder import build_query_response, extract_chart_intent
7
  from app.utils.logger import setup_logger
8
  from collections import Counter
9
+ import spaces
10
 
11
  logger = setup_logger(__name__)
12
  router = APIRouter()
13
 
 
14
  @spaces.GPU
15
+ @router.post("/ask", response_model=QueryResponse)
16
  async def ask_question(request: QueryRequest):
17
  """
18
  Answer natural language questions using RAG
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  fastapi==0.109.0
2
  uvicorn[standard]==0.27.0
3
  python-dotenv==1.0.0
 
1
+ spaces
2
  fastapi==0.109.0
3
  uvicorn[standard]==0.27.0
4
  python-dotenv==1.0.0