diff --git a/veadk/cli/studio/fast_api.py b/veadk/cli/studio/fast_api.py
index 85af90a0..7f4c9bb9 100644
--- a/veadk/cli/studio/fast_api.py
+++ b/veadk/cli/studio/fast_api.py
@@ -22,6 +22,7 @@
 from fastapi.staticfiles import StaticFiles
 from google.adk.agents.run_config import RunConfig, StreamingMode
 from google.genai import types
+from pydantic import BaseModel
 
 from veadk.cli.studio.models import (
     GetAgentResponse,
@@ -151,6 +152,14 @@ async def runner_run_sse(user_text: str):
             yield f"data: {json.dumps(TOOL_INPUT_AVAILABLE)}\n\n"
 
         for function_response in event.get_function_responses():
+            # for load_memory and load_knowledgebase tools, the output is based BaseModel
+            # we have to convert it
+            for key, value in function_response.response.items():
+                if isinstance(value, BaseModel):
+                    function_response.response[key] = value.model_dump(
+                        exclude_none=False
+                    )
+
             TOOL_OUTPUT_AVAILABLE = {
                 "type": "tool-output-available",
                 "toolCallId": function_response.id,