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

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -6,6 +6,7 @@ import re
6
  from dotenv import load_dotenv
7
  import os
8
  from fastapi import FastAPI
 
9
  from pydantic import BaseModel
10
  from fastapi.middleware.cors import CORSMiddleware
11
 
@@ -64,7 +65,7 @@ app.add_middleware(
64
 
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
 
 
6
  from dotenv import load_dotenv
7
  import os
8
  from fastapi import FastAPI
9
+ from fastapi import Query
10
  from pydantic import BaseModel
11
  from fastapi.middleware.cors import CORSMiddleware
12
 
 
65
 
66
  # Create a Pydantic model to handle the input data
67
  class TopicInput(BaseModel):
68
+ user_input: str = Query(default="", description="input query to generate subtopics")
69
  num_topics: int = Query(default=5, description="Number of subtopics to generate (default: 5)")
70
  previous_query: str = Query(default="", description="Previous query for context (default: empty string)")
71