pvanand commited on
Commit
41880ba
1 Parent(s): 32d7366

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -46,7 +46,7 @@ sys_prompts = {
46
  }
47
 
48
  class QueryModel(BaseModel):
49
- query: str = Query(default="market research", description="input query to generate Report")
50
  description: str = Query(default="", description="additional context for report")
51
  user_id: str = Query(default="", description="unique user id")
52
  user_name: str = Query(default="", description="user name")
@@ -56,7 +56,7 @@ class QueryModel(BaseModel):
56
 
57
  @app.post("/generate_report")
58
  async def generate_report(request: Request, query: QueryModel):
59
- query_str = query.query
60
  description = query.description
61
  user_id = query.user_id
62
  internet = "online" if query.internet else "offline"
 
46
  }
47
 
48
  class QueryModel(BaseModel):
49
+ topic: str = Query(default="market research", description="input query to generate Report")
50
  description: str = Query(default="", description="additional context for report")
51
  user_id: str = Query(default="", description="unique user id")
52
  user_name: str = Query(default="", description="user name")
 
56
 
57
  @app.post("/generate_report")
58
  async def generate_report(request: Request, query: QueryModel):
59
+ query_str = query.topic
60
  description = query.description
61
  user_id = query.user_id
62
  internet = "online" if query.internet else "offline"