remove log messages, robustify AMA

Author: stephenfreund

mxgraph_component/mxgraph_component/frontend/src/diagram.tsx

@@ -496,7 +496,7 @@ class CustomHierarchicalLayout extends mx.mxHierarchicalLayout {
 export function layoutDiagram(graph: mxGraph) {
     graph.getModel().beginUpdate();
     try {
-        console.log(graph)
+        // console.log(graph)
 
         // add an implicit edge from each node to the group of the target if that gropu is not the default parent
         const cells = graph.getChildCells(graph.getDefaultParent(), true, true);
@@ -538,7 +538,7 @@ export function layoutDiagram(graph: mxGraph) {
                 graph.getModel().remove(edge);
             }
         }
-        console.log(graph)
+        // console.log(graph)
     } finally {
         graph.getModel().endUpdate();
     }

mxgraph_component/mxgraph_component/frontend/src/index.tsx

@@ -1294,11 +1294,11 @@ function onRender(event: Event): void {
   const data = (event as CustomEvent<RenderData>).detail;
   const initial_render = currentDiagram === undefined;
 
-  console.log("currentDiagram === undefined", initial_render)
-  console.log("forced", data.args["forced"])
-  console.log("clear", data.args["clear"])
-  console.log("!editable", !data.args["editable"])
-  console.log("Force", initial_render || !data.args['editable'] || data.args["forced"] || data.args["clear"])
+  // console.log("currentDiagram === undefined", initial_render)
+  // console.log("forced", data.args["forced"])
+  // console.log("clear", data.args["clear"])
+  // console.log("!editable", !data.args["editable"])
+  // console.log("Force", initial_render || !data.args['editable'] || data.args["forced"] || data.args["clear"])
 
 
   setEditable(false);

src/flowco/dataflow/dfg_update.py

@@ -43,7 +43,7 @@ def update_dataflow_graph(
     Returns a new DataFlowGraph instance with the updates applied.
     """
 
-    print(diagram_update)
+    # print(diagram_update)
 
     # Create deep copies of current nodes and edges to avoid mutation
     new_nodes_dict: Dict[str, Node] = {
@@ -188,7 +188,7 @@ def update_dataflow_graph(
             )
             for group in diagram_update.groups
         ]
-        print(groups)
+        # print(groups)
 
         new_graph = DataFlowGraph(
             description=current_graph.description,

src/flowco/dataflow/extended_type.py

@@ -1133,7 +1133,7 @@ def rep_to_summary(the_type: TypeRepresentation) -> str:
             return "Record"
         elif t == "pd.DataFrame":
             return "DataFrame"
-        elif t in ["np.ndarray", "list", "set", "pd.Series"]:
+        elif t in ["np.ndarray", "List", "Set", "pd.Series"]:
             return f"{t.capitalize()}[{rep_to_summary(the_type.element_type)}]"
         elif t == "class":
             return ext_type.the_type.name

src/flowco/page/output.py

@@ -18,6 +18,8 @@
     ImageURL,
 )
 
+from flowco.util.text import sandwich_text
+
 
 class OutputType(str, enum.Enum):
     text = "text"
@@ -284,8 +286,7 @@ class ResultOutput(BaseModel):
     def to_content_part(self) -> ChatCompletionContentPartParam:
         if self.output_type == OutputType.text:
             return ChatCompletionContentPartTextParam(
-                type="text",
-                text=self.data,
+                type="text", text=sandwich_text(self.data, 50, 25)
             )
         elif self.output_type == OutputType.image:
             return ChatCompletionContentPartImageParam(

src/flowco/util/text.py

@@ -354,3 +354,16 @@ def black_format(code: List[str], line_length=70):
     except Exception as e:
         formatted_code = code_block
     return formatted_code.split("\n")
+
+
+def sandwich_text(text: str, max_lines: int = 1024, top_proportion: float = 0.5) -> str:
+    if max_lines == None:
+        return text
+    lines = text.splitlines()
+    if len(lines) <= max_lines:
+        return text
+    else:
+        total_len = max_lines - 1  # some slop for the ...
+        top_len = int(top_proportion * total_len)
+        bot_start = len(lines) - (total_len - top_len)
+        return "\n".join(lines[0:top_len] + [" [...] "] + lines[bot_start:])

src/llm/assistant.py

@@ -232,16 +232,15 @@ def make_call(
         self, tool_call: ChatCompletionMessageToolCall | ParsedFunctionToolCall
     ) -> str:
         call_text = ""
-
         call_id = tool_call.id
         function = tool_call.function
         function_name = function.name
-        args = json.loads(function.arguments)
-        function_def = self.functions[function_name]
+        try:
+            args = json.loads(function.arguments)
+            function_def = self.functions[function_name]
 
-        self.logger.log(f"Tool call: {function_name}")
+            self.logger.log(f"Tool call: {function_name}")
 
-        try:
             result = function_def.function(**args)
         except Exception as e:
             self.logger.error(f"Error in tool call {function_name}: {e}")