add

agentgraph/reconstruction/prompt_reconstructor.py

@@ -264,33 +264,70 @@ class PromptReconstructor:
                 # Place system prompt first
                 complete_prompt = system_role
                 
-                # Format tool definitions with better structure to match example
+                # Enhanced tool definitions processing with fallback handling
+                tool_names_for_action = []
                 if tool_definitions:
                     for tool_def in tool_definitions:
-                        # Extract tool name, args, and description to reformat
+                        # Extract tool name, args, and description with improved parsing
                         tool_name = ""
                         tool_args = "{}"
                         tool_desc = ""
                         
-                        if "Tool Name:" in tool_def:
-                            name_start = tool_def.find("Tool Name:") + len("Tool Name:")
-                            name_end = tool_def.find("\n", name_start) if "\n" in tool_def[name_start:] else len(tool_def)
-                            tool_name = tool_def[name_start:name_end].strip()
+                        # Check if tool definition follows expected format
+                        if "Tool Name:" in tool_def and "Tool Arguments:" in tool_def and "Tool Description:" in tool_def:
+                            # Parse structured format
+                            if "Tool Name:" in tool_def:
+                                name_start = tool_def.find("Tool Name:") + len("Tool Name:")
+                                name_end = tool_def.find("\n", name_start) if "\n" in tool_def[name_start:] else len(tool_def)
+                                tool_name = tool_def[name_start:name_end].strip()
+                            
+                            if "Tool Arguments:" in tool_def:
+                                args_start = tool_def.find("Tool Arguments:") + len("Tool Arguments:")
+                                args_end = tool_def.find("\n", args_start) if "\n" in tool_def[args_start:] else len(tool_def)
+                                tool_args = tool_def[args_start:args_end].strip()
+                            
+                            if "Tool Description:" in tool_def:
+                                desc_start = tool_def.find("Tool Description:") + len("Tool Description:")
+                                desc_end = len(tool_def)  # Take rest of the text for description
+                                tool_desc = tool_def[desc_start:desc_end].strip()
+                        else:
+                            # Fallback: Extract from tool entity if structured format not available
+                            # Find the tool entity that matches this definition
+                            for entity_id, entity in self.entities.items():
+                                if entity["type"] == "Tool" and entity.get("raw_prompt") == tool_def:
+                                    tool_name = entity["name"]
+                                    # Generate reasonable arguments based on tool purpose
+                                    tool_desc = tool_def
+                                    # Create basic arguments structure
+                                    if "search" in tool_name.lower():
+                                        tool_args = '{"query": "search_term", "max_results": 10}'
+                                    elif "load" in tool_name.lower() or "read" in tool_name.lower():
+                                        tool_args = '{"file_path": "path/to/file", "format": "auto"}'
+                                    elif "calculat" in tool_name.lower() or "analyz" in tool_name.lower():
+                                        tool_args = '{"data": "input_data", "method": "default"}'
+                                    elif "validat" in tool_name.lower() or "check" in tool_name.lower():
+                                        tool_args = '{"data": "target_data", "criteria": "standard"}'
+                                    else:
+                                        tool_args = '{"input": "data"}'
+                                    break
+                            
+                            # If still no tool name found, use a generic name
+                            if not tool_name:
+                                tool_name = "UnknownTool"
+                                tool_desc = tool_def if tool_def else "Tool description not available"
+                                tool_args = '{"input": "data"}'
                         
-                        if "Tool Arguments:" in tool_def:
-                            args_start = tool_def.find("Tool Arguments:") + len("Tool Arguments:")
-                            args_end = tool_def.find("\n", args_start) if "\n" in tool_def[args_start:] else len(tool_def)
-                            tool_args = tool_def[args_start:args_end].strip()
+                        # Add to tool names list for action format
+                        if tool_name and tool_name != "UnknownTool":
+                            tool_names_for_action.append(tool_name)
                         
-                        if "Tool Description:" in tool_def:
-                            desc_start = tool_def.find("Tool Description:") + len("Tool Description:")
-                            desc_end = tool_def.find("\n", desc_start) if "\n" in tool_def[desc_start:] else len(tool_def)
-                            tool_desc = tool_def[desc_start:desc_end].strip()
-                        
-                        # Format the tool entry more closely to the example format
+                        # Format the tool entry
                         complete_prompt += f"Tool Name: {tool_name}\n"
                         complete_prompt += f"Tool Arguments: {tool_args}\n"
                         complete_prompt += f"Tool Description: {tool_desc}\n\n"
+                else:
+                    # No tools available - provide clear message
+                    complete_prompt += "No tools are currently available for this task.\n\n"
                 
                 # Enhanced response format with realistic agent reasoning patterns
                 complete_prompt += "RESPONSE FORMAT - Follow this structure for each step:\n\n"
@@ -299,7 +336,7 @@ class PromptReconstructor:
                 complete_prompt += "Reasoning: [Analyze the current situation and explain your thinking process]\n"
                 complete_prompt += "Task Analysis: [Break down what needs to be done and identify requirements]\n"
                 complete_prompt += "Tool Selection: [Choose appropriate tool and justify why]\n"
-                complete_prompt += "Action: [Tool name from: " + ", ".join([t.split("Tool Name:")[1].strip().split("\n")[0] for t in tool_definitions if "Tool Name:" in t]) + "]\n"
+                complete_prompt += "Action: [Tool name from: " + ", ".join(tool_names_for_action) + "]\n"
                 complete_prompt += "Action Input: [JSON object with parameters, using \" for keys and values]\n"
                 complete_prompt += "Observation: [Result of the action]\n"
                 complete_prompt += "```\n\n"