pvanand commited on
Commit
a9d9604
·
verified ·
1 Parent(s): 0289fc8

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -762,9 +762,11 @@ async def digi_followup_agent_v2(query: FollowupQueryModel, background_tasks: Ba
762
  logger.info(f"Completed followup agent response for query: {query.query}, send result: {result}")
763
 
764
  return StreamingResponse(process_response(), media_type="text/event-stream")
765
-
766
 
767
 
 
 
 
768
  from fastapi.middleware.cors import CORSMiddleware
769
 
770
  # CORS middleware setup
 
762
  logger.info(f"Completed followup agent response for query: {query.query}, send result: {result}")
763
 
764
  return StreamingResponse(process_response(), media_type="text/event-stream")
 
765
 
766
 
767
+ from document_generator import router as document_generator_router
768
+ app.include_router(document_generator_router, prefix="/api/v1")
769
+
770
  from fastapi.middleware.cors import CORSMiddleware
771
 
772
  # CORS middleware setup