Update presentation_api.py

presentation_api.py

@@ -22,28 +22,38 @@ router = APIRouter(
 )
 
 
-@tool
+@tool(parse_docstring=True)
 def plan(input: dict) -> str:
-    """Create a presentation plan with numbered slides and their descriptions.
+    """Create a 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
-        Eg: {"1":"title page for ..", "2":"introduction .."}
-    Returns:
-        A dictionary with slide numbers as keys and descriptions as values
+        input: Dictionary containing presentation details, Example: {"1": "title page for ..", "2": "introduction .."}
     """
     return f"Plan created"
 
-@tool
+@tool(parse_docstring=True)
 def create_slide(slideno: int, content: str) -> str:
-    """Create a single presentation slide.
+    """Create a single presentation slide. Returns a confirmation message indicating that the slide has been created.
+
     Args:
-        slideno: The slide number to create
-        content: The content for the slide
-    Returns:
-        Confirmation of slide creation
+        slideno: The slide number to create.
+        content: The content for the slide.
     """
     return f"slide {slideno} created"
 
+@tool(parse_docstring=True)
+def execute_python(expression: str) -> str:
+    """Execute a python mathematic expression. Returns the result of the expression or an error message if execution fails.
+
+    Args:
+        expression: The python expression to execute.
+    """
+    try:
+        result = eval(expression)
+        return f"The result of the expression is {result}"
+    except Exception as e:
+        return f"Error executing the expression: {str(e)}"
+
 memory = MemorySaver()
 model = ChatOpenAI(model="gpt-4o-mini", streaming=True)
 prompt = ChatPromptTemplate.from_messages([
@@ -78,7 +88,7 @@ def state_modifier(state) -> list[BaseMessage]:
 # Create the agent with presentation tools
 agent = create_react_agent(
     model,
-    tools=[plan, create_slide],
+    tools=[plan, create_slide, execute_python],
     checkpointer=memory,
     state_modifier=state_modifier,
 )
@@ -114,11 +124,11 @@ async def chat(input_data: ChatInput):
                     yield f"{json.dumps({'type': 'token', 'content': content})}\n"
 
             elif kind == "on_tool_start":
-                tool_input = event['data'].get('input', '')
+                tool_input = event['data'].get('input', '').pretty_repr()
                 yield f"{json.dumps({'type': 'tool_start', 'tool': event['name'], 'input': tool_input})}\n"
             
             elif kind == "on_tool_end":
-                tool_output = event['data'].get('output', '')
+                tool_output = event['data'].get('output', '').pretty_repr()
                 yield f"{json.dumps({'type': 'tool_end', 'tool': event['name'], 'output': tool_output})}\n"
 
     return EventSourceResponse(
@@ -130,3 +140,9 @@ async def chat(input_data: ChatInput):
 async def health_check():
     return {"status": "healthy"}
 
+
+app = FastAPI()
+app.include_router(router)
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)