Update presentation_api.py

presentation_api.py

@@ -25,20 +25,17 @@ router = APIRouter(
 from typing import Dict
 from pydantic import BaseModel, Field
 
-# First, let's create a proper input schema for the plan tool
-class PlanInput(BaseModel):
-    input: Dict[str, str] = Field(
-        ...,  # This makes the field required
-        description="Dictionary containing slide numbers as keys and descriptions as values",
-        example={
-            "1": "Title page for presentation",
-            "2": "Introduction section with key points",
-            "3": "Main content overview"
-        }
-    )
+class Slide(BaseModel):
+    slide_number: int
+    title: str
+    description: str
+
+# Define the slides container
+class SlidesInput(BaseModel):
+    slides: List[Slide]
 
 @tool(parse_docstring=True)
-def plan(input: PlanInput) -> str:
+def plan(input: SlidesInput) -> str:
     """Input presentation plan with numbered slides and their descriptions. 
     Returns a confirmation message indicating that the plan has been created.