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

ciayn

This commit is contained in:
AI Christianson 2024-12-28 14:41:39 -05:00
parent d8a3c88624
commit 535be97c1f
2 changed files with 17 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
ra_aid/agents/ciayn_agent.py
View File

@ -1,11 +1,7 @@
import inspect import inspect
from typing import Dict, Any, Generator, List, Optional from typing import Dict, Any, Generator, List, Optional
from langchain_core.messages import AIMessage, HumanMessage from langchain_core.messages import AIMessage, HumanMessage
from rich.panel import Panel from ra_aid.exceptions import ToolExecutionError
from rich.markdown import Markdown
from rich.console import Console
console = Console()
class CiaynAgent: class CiaynAgent:
def _get_function_info(self, func): def _get_function_info(self, func):
@ -78,7 +74,7 @@ Output **ONLY THE CODE** and **NO MARKDOWN BACKTICKS**"""
except Exception as e: except Exception as e:
error_msg = f"Error executing code: {str(e)}" error_msg = f"Error executing code: {str(e)}"
console.print(f"[red]Error:[/red] {error_msg}") console.print(f"[red]Error:[/red] {error_msg}")
return error_msg raise ToolExecutionError(error_msg)
def _create_agent_chunk(self, content: str) -> Dict[str, Any]: def _create_agent_chunk(self, content: str) -> Dict[str, Any]:
"""Create an agent chunk in the format expected by print_agent_output.""" """Create an agent chunk in the format expected by print_agent_output."""
@ -107,16 +103,15 @@ Output **ONLY THE CODE** and **NO MARKDOWN BACKTICKS**"""
base_prompt = self._build_prompt(None if first_iteration else last_result) base_prompt = self._build_prompt(None if first_iteration else last_result)
chat_history.append(HumanMessage(content=base_prompt)) chat_history.append(HumanMessage(content=base_prompt))
try: full_history = initial_messages + chat_history
full_history = initial_messages + chat_history response = self.model.invoke(full_history)
response = self.model.invoke(full_history)
try:
last_result = self._execute_tool(response.content) last_result = self._execute_tool(response.content)
chat_history.append(response) chat_history.append(response)
first_iteration = False first_iteration = False
yield {} yield {}
except Exception as e: except ToolExecutionError as e:
error_msg = f"Error: {str(e)}" yield self._create_error_chunk(str(e))
yield self._create_error_chunk(error_msg)
break break

9
ra_aid/exceptions.py
View File

@ -7,3 +7,12 @@ class AgentInterrupt(Exception):
separate from KeyboardInterrupt which is reserved for top-level handling. separate from KeyboardInterrupt which is reserved for top-level handling.
""" """
pass pass
class ToolExecutionError(Exception):
"""Exception raised when a tool execution fails.
This exception is used to distinguish tool execution failures
from other types of errors in the agent system.
"""
pass