Update artifacts

presentation_api.py

@@ -15,6 +15,7 @@ from datetime import datetime
 from fastapi import APIRouter
 from langchain_core.runnables import RunnableConfig
 from langchain_core.prompts import ChatPromptTemplate
+from typing import Any
 
 router = APIRouter(
     prefix="/presentation",
@@ -23,20 +24,35 @@ router = APIRouter(
 
 import json
 
-@tool
-def plan(slides_json: str) -> str:
+@tool(response_format="content_and_artifact")
+def plan(slides_json: str) -> tuple[str, dict]:
     """Create a presentation plan from a JSON string of slides (keys=slide numbers, values=content)."""
     try:
         slides = json.loads(slides_json)
-        return f"Plan created with {len(slides)} slides: {', '.join(slides.keys())}."
-    except json.JSONDecodeError as e:
-        return f"Invalid JSON format. Please provide a valid JSON string {str(e)[:100]}."
+        print(slides)
+        return (
+            f"Plan created with {len(slides)} slides: {', '.join(slides.keys())}.",
+            {"slides_plan_json": slides_json}
+        )
+    except Exception as e:
+        return (
+            f"Invalid JSON format. Please provide a valid JSON string {str(e)[:100]}.",
+            None
+        )
 
-@tool
-def create_slide(slide_number: int, content: str) -> str:
+@tool(response_format="content_and_artifact") 
+def create_slide(slide_number: int, content: str, config: RunnableConfig) -> tuple[str, dict]:
     """Create a slide with the given number and content."""
     # Integration with slide creation API or template would go here
-    return f"Slide {slide_number} created: {content}"
+    slide = {
+        "number": slide_number,
+        "content": content,
+        "created_at": datetime.now().isoformat()
+    }
+    return (
+        f"Slide {slide_number} created",
+        {"slide": slide}
+    )
 
 @tool(parse_docstring=True)
 def execute_python(expression: str) -> str:
@@ -113,7 +129,6 @@ async def chat(input_data: ChatInput):
             version="v2"
         ):
             kind = event["event"]
-            print(event)
             
             if kind == "on_chat_model_stream":
                 content = event["data"]["chunk"].content
@@ -125,12 +140,11 @@ async def chat(input_data: ChatInput):
                 yield f"{json.dumps({'type': 'tool_start', 'tool': event['name'], 'input': tool_input})}\n"
             
             elif kind == "on_tool_end":
+                print(event['data'])
                 tool_output = event['data'].get('output', '')
-                print(type(tool_output))
-                print(type(dir(tool_output)))
-                tool_output = tool_output.pretty_repr()
-                yield f"{json.dumps({'type': 'tool_end', 'tool': event['name'], 'output': tool_output})}\n"
-            
+                artifact_output = tool_output.artifact if tool_output.artifact else None
+                yield f"{json.dumps({'type': 'tool_end', 'tool': event['name'], 'output': tool_output.pretty_repr(), 'artifacts_data': artifact_output})}\n"
+                print(tool_output.pretty_repr())
     return EventSourceResponse(
         generate(),
         media_type="text/event-stream"