zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Return reason for failure.

This commit is contained in:
AI Christianson 2024-12-21 11:04:57 -05:00
parent 1fc64c5151
commit a9c6ea93b4
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ra_aid/tools/agent.py
View File

@ -29,6 +29,7 @@ def request_research(query: str) -> Dict[str, Any]:
- facts: Current key facts - facts: Current key facts
- files: Related files - files: Related files
- success: Whether completed or interrupted - success: Whether completed or interrupted
- reason: Reason for failure, if any
""" """
# Initialize model and memory # Initialize model and memory
model = initialize_llm("anthropic", "claude-3-sonnet-20240229") model = initialize_llm("anthropic", "claude-3-sonnet-20240229")
@ -79,17 +80,21 @@ def request_research(query: str) -> Dict[str, Any]:
) )
success = True success = True
reason = None
except KeyboardInterrupt: except KeyboardInterrupt:
console.print("\n[yellow]Research interrupted by user[/yellow]") console.print("\n[yellow]Research interrupted by user[/yellow]")
success = False success = False
reason = "cancelled_by_user"
except Exception as e: except Exception as e:
console.print(f"\n[red]Error during research: {str(e)}[/red]") console.print(f"\n[red]Error during research: {str(e)}[/red]")
success = False success = False
reason = f"error: {str(e)}"
# Gather results # Gather results
return { return {
"facts": get_memory_value("key_facts"), "facts": get_memory_value("key_facts"),
"files": list(get_related_files()), "files": list(get_related_files()),
"notes": get_memory_value("research_notes"), "notes": get_memory_value("research_notes"),
"success": success "success": success,
"reason": reason
} }