pvanand commited on
Commit
6474977
1 Parent(s): 4730670

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -3
main.py CHANGED
@@ -507,6 +507,7 @@ async def followup_agent(query: FollowupQueryModel, background_tasks: Background
507
  logger.info(f"Completed followup agent response for query: {query.query}, send result: {result}")
508
 
509
  return StreamingResponse(process_response(), media_type="text/event-stream")
 
510
  @app.post("/v2/followup-tools-agent")
511
  def followup_agent(query: FollowupQueryModel, background_tasks: BackgroundTasks, api_key: str = Depends(verify_api_key)):
512
  """
@@ -721,7 +722,7 @@ async def followup_agent(query: FollowupQueryModel, background_tasks: Background
721
 
722
 
723
  @app.post("v2/digiyatra-followup")
724
- async def followup_agent_v2(query: FollowupQueryModel, background_tasks: BackgroundTasks, api_key: str = Depends(verify_api_key)):
725
  """
726
  Followup agent endpoint that provides helpful responses or generates clarifying questions based on user queries.
727
  Requires API Key authentication via X-API-Key header.
@@ -747,8 +748,7 @@ async def followup_agent_v2(query: FollowupQueryModel, background_tasks: Backgro
747
 
748
  logger.info(f"LLM RAW response for query: {query.query}: {full_response}")
749
  response_content, interact,tools = parse_followup_and_tools(full_response)
750
-
751
-
752
  result = {
753
  "response": response_content,
754
  "clarification": interact
@@ -763,6 +763,7 @@ async def followup_agent_v2(query: FollowupQueryModel, background_tasks: Backgro
763
  logger.info(f"Completed followup agent response for query: {query.query}, send result: {result}")
764
 
765
  return StreamingResponse(process_response(), media_type="text/event-stream")
 
766
 
767
 
768
  from fastapi.middleware.cors import CORSMiddleware
 
507
  logger.info(f"Completed followup agent response for query: {query.query}, send result: {result}")
508
 
509
  return StreamingResponse(process_response(), media_type="text/event-stream")
510
+
511
  @app.post("/v2/followup-tools-agent")
512
  def followup_agent(query: FollowupQueryModel, background_tasks: BackgroundTasks, api_key: str = Depends(verify_api_key)):
513
  """
 
722
 
723
 
724
  @app.post("v2/digiyatra-followup")
725
+ async def digi_followup_agent_v2(query: FollowupQueryModel, background_tasks: BackgroundTasks, api_key: str = Depends(verify_api_key)):
726
  """
727
  Followup agent endpoint that provides helpful responses or generates clarifying questions based on user queries.
728
  Requires API Key authentication via X-API-Key header.
 
748
 
749
  logger.info(f"LLM RAW response for query: {query.query}: {full_response}")
750
  response_content, interact,tools = parse_followup_and_tools(full_response)
751
+
 
752
  result = {
753
  "response": response_content,
754
  "clarification": interact
 
763
  logger.info(f"Completed followup agent response for query: {query.query}, send result: {result}")
764
 
765
  return StreamingResponse(process_response(), media_type="text/event-stream")
766
+
767
 
768
 
769
  from fastapi.middleware.cors import CORSMiddleware