Spaces:
Paused
Paused
File size: 1,959 Bytes
201c475 deab47c 201c475 5266ca3 201c475 15a2751 201c475 bab921d 201c475 bab921d 15a2751 bab921d 15a2751 bab921d deab47c bab921d 201c475 15a2751 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
from abc import abstractmethod
from typing import TypedDict
from agent import Agent
from python.helpers.print_style import PrintStyle
from python.helpers import files, messages
class Response:
def __init__(self, message: str, break_loop: bool) -> None:
self.message = message
self.break_loop = break_loop
class Tool:
def __init__(self, agent: Agent, name: str, args: dict[str,str], message: str, **kwargs) -> None:
self.agent = agent
self.name = name
self.args = args
self.message = message
@abstractmethod
def execute(self,**kwargs) -> Response:
pass
def before_execution(self, **kwargs):
PrintStyle(font_color="#1B4F72", padding=True, background_color="white", bold=True).print(f"{self.agent.agent_name}: Using tool '{self.name}':")
if self.args and isinstance(self.args, dict):
for key, value in self.args.items():
PrintStyle(font_color="#85C1E9", bold=True).stream(self.nice_key(key)+": ")
PrintStyle(font_color="#85C1E9", padding=isinstance(value,str) and "\n" in value).stream(value)
PrintStyle().print()
def after_execution(self, response: Response, **kwargs):
text = messages.truncate_text(response.message.strip(), self.agent.config.max_tool_response_length)
msg_response = files.read_file("./prompts/fw.tool_response.md", tool_name=self.name, tool_response=text)
self.agent.append_message(msg_response, human=True)
PrintStyle(font_color="#1B4F72", background_color="white", padding=True, bold=True).print(f"{self.agent.agent_name}: Response from tool '{self.name}':")
PrintStyle(font_color="#85C1E9").print(response.message)
def nice_key(self, key:str):
words = key.split('_')
words = [words[0].capitalize()] + [word.lower() for word in words[1:]]
result = ' '.join(words)
return result |