gradio ui spaces hf v1

app.py

@@ -38,9 +38,18 @@ query_parser_agent = Agent(
 # Need to define QueryAnalysisOutput class here as it's used by the task
 class QueryAnalysisOutput(BaseModel):
     """Structured output for the query analysis task."""
-    symbols: list[str] = Field(..., description="List of stock ticker symbols (e.g., ['TSLA', 'AAPL']).")
-    timeframe: str = Field(..., description="Time period (e.g., '1d', '1mo', '1y').")
-    action: str = Field(..., description="Action to be performed (e.g., 'fetch', 'plot').")
+    symbols: list[str] = Field(
+        ...,
+        json_schema_extra={"description": "List of stock ticker symbols (e.g., ['TSLA', 'AAPL'])."}
+    )
+    timeframe: str = Field(
+        ...,
+        json_schema_extra={"description": "Time period (e.g., '1d', '1mo', '1y')."}
+    )
+    action: str = Field(
+        ...,
+        json_schema_extra={"description": "Action to be performed (e.g., 'fetch', 'plot')."}
+    )
 
 
 query_parsing_task = Task(