Update presentation_api.py

presentation_api.py

@@ -22,14 +22,30 @@ 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"
+        }
+    )
+
 @tool(parse_docstring=True)
-def plan(input: dict) -> str:
-    """Input presentation plan with numbered slides and their descriptions. Returns a confirmation message indicating that the plan has been created.
+def plan(input: PlanInput) -> str:
+    """Input presentation plan with numbered slides and their descriptions. 
+    Returns a confirmation message indicating that the plan has been created.
 
     Args:
-        input: Dictionary containing presentation details, Example: {"1": "title page for ..", "2": "introduction .."}
+        input: PlanInput object containing presentation details
     """
-    return f"Plan created"
+    return f"Plan created with {len(input.input)} slides"
 
 @tool(parse_docstring=True)
 def create_slide(slideno: int, content: str) -> str: