pvanand commited on
Commit
c32bfde
·
verified ·
1 Parent(s): c41eb74

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -3
main.py CHANGED
@@ -65,9 +65,8 @@ app.add_middleware(
65
  # Create a Pydantic model to handle the input data
66
  class TopicInput(BaseModel):
67
  user_input: str
68
- num_topics: int
69
- previous_query: str
70
-
71
 
72
  @app.get("/",tags=["Home"])
73
  def api_home():
 
65
  # Create a Pydantic model to handle the input data
66
  class TopicInput(BaseModel):
67
  user_input: str
68
+ num_topics: int = Query(default=5, description="Number of subtopics to generate (default: 5)")
69
+ previous_query: str = Query(default="", description="Previous query for context (default: empty string)")
 
70
 
71
  @app.get("/",tags=["Home"])
72
  def api_home():