Upload duckdb_react_agent.py

duckdb_react_agent.py

@@ -306,12 +306,12 @@ def node_decide_viz(state: AgentState, llm: ChatOpenAI) -> AgentState:
     resp = llm.invoke(msgs)  # type: ignore
 
     spec = {"make_plot": False}
-    try:
-        spec = json.loads(resp.content)  # type: ignore
-        if not isinstance(spec, dict) or "make_plot" not in spec:
-            spec = {"make_plot": False}
-    except Exception:
-        spec = {"make_plot": False}
+    # try:
+    #     spec = json.loads(resp.content)  # type: ignore
+    #     if not isinstance(spec, dict) or "make_plot" not in spec:
+    #         spec = {"make_plot": False}
+    # except Exception:
+    #     spec = {"make_plot": False}
 
     return {**state, "_viz_spec": spec}
 
@@ -625,4 +625,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()