Merge pull request #3 from metricinsights/charts-bugfix

bugfix charts

Author: Nayjest

mcp_server.py

@@ -10,7 +10,6 @@
 from fastmcp import FastMCP, Context
 from microcore import ui
 
-
 setup_logging()
 
 mcp = FastMCP(
@@ -82,7 +81,11 @@ async def ask(agent: str, question: str, ctx: Context) -> str:
             json_serialized_data = json.dumps(
                 dict(
                     data_type="chart",
-                    data=i.data["chart_spec"],
+                    data=(
+                        json.loads(i.data["chart_spec"])
+                        if isinstance(i.data["chart_spec"], str)
+                        else i.data["chart_spec"]
+                    ),
                 ),
                 indent=2,
             )

snowflake_mcp/config.py

@@ -3,11 +3,11 @@
 
 @dataclass
 class Config:
-    account: str = field(default='')
-    token: str = field(default='')
-    database: str = field(default='')
-    schema: str = field(default='')
-    agent: str = field(default='')
+    account: str = field(default="")
+    token: str = field(default="")
+    database: str = field(default="")
+    schema: str = field(default="")
+    agent: str = field(default="")
     defaults: "Config" = field(default=None)
 
     def __post_init__(self):

test_agent.py

@@ -16,13 +16,9 @@
         res = i.data["text"]
         print(ui.blue(res), end="", flush=True)
     elif i.event_type == "response.tool_result.status":
-        print(ui.yellow(
-            f"[Status: {i.data['status']}]: {i.data.get('message', '')}"
-        ))
+        print(ui.yellow(f"[Status: {i.data['status']}]: {i.data.get('message', '')}"))
     elif i.event_type == "response.status":
-        print(ui.yellow(
-            f"[Status: {i.data['status']}]: {i.data.get('message', '')}"
-        ))
+        print(ui.yellow(f"[Status: {i.data['status']}]: {i.data.get('message', '')}"))
     elif i.event_type == "response.thinking.delta":
         res = i.data["text"]
         print(ui.gray(res), end="", flush=True)