feat: Rewrite agents (#2232)
This commit is contained in:
@@ -74,7 +74,7 @@ def main():
|
||||
"""
|
||||
|
||||
# Create an agent
|
||||
agent = client.create_agent(name=agent_uuid, memory=ChatMemory(human="My name is Matt.", persona=persona), tools=[tool.name])
|
||||
agent = client.create_agent(name=agent_uuid, memory=ChatMemory(human="My name is Matt.", persona=persona), tool_ids=[tool.id])
|
||||
print(f"Created agent: {agent.name} with ID {str(agent.id)}")
|
||||
|
||||
# Send a message to the agent
|
||||
|
||||
@@ -29,7 +29,7 @@ agent_state = client.create_agent(
|
||||
# whether to include base letta tools (default: True)
|
||||
include_base_tools=True,
|
||||
# list of additional tools (by name) to add to the agent
|
||||
tools=[],
|
||||
tool_ids=[],
|
||||
)
|
||||
print(f"Created agent with name {agent_state.name} and unique ID {agent_state.id}")
|
||||
|
||||
|
||||
@@ -36,7 +36,7 @@ print(f"Created tool with name {tool.name}")
|
||||
# create a new agent
|
||||
agent_state = client.create_agent(
|
||||
# create the agent with an additional tool
|
||||
tools=[tool.name],
|
||||
tool_ids=[tool.id],
|
||||
# add tool rules that terminate execution after specific tools
|
||||
tool_rules=[
|
||||
# exit after roll_d20 is called
|
||||
@@ -45,7 +45,7 @@ agent_state = client.create_agent(
|
||||
TerminalToolRule(tool_name="send_message"),
|
||||
],
|
||||
)
|
||||
print(f"Created agent with name {agent_state.name} with tools {agent_state.tool_names}")
|
||||
print(f"Created agent with name {agent_state.name} with tools {[t.name for t in agent_state.tools]}")
|
||||
|
||||
# Message an agent
|
||||
response = client.send_message(agent_id=agent_state.id, role="user", message="roll a dice")
|
||||
@@ -61,7 +61,8 @@ client.add_tool_to_agent(agent_id=agent_state.id, tool_id=tool.id)
|
||||
client.delete_agent(agent_id=agent_state.id)
|
||||
|
||||
# create an agent with only a subset of default tools
|
||||
agent_state = client.create_agent(include_base_tools=False, tools=[tool.name, "send_message"])
|
||||
send_message_tool = client.get_tool_id("send_message")
|
||||
agent_state = client.create_agent(include_base_tools=False, tool_ids=[tool.id, send_message_tool])
|
||||
|
||||
# message the agent to search archival memory (will be unable to do so)
|
||||
response = client.send_message(agent_id=agent_state.id, role="user", message="search your archival memory")
|
||||
|
||||
@@ -67,7 +67,9 @@ def main():
|
||||
"""
|
||||
|
||||
# Create an agent
|
||||
agent_state = client.create_agent(name=agent_uuid, memory=ChatMemory(human="My name is Matt.", persona=persona), tools=[tool_name])
|
||||
agent_state = client.create_agent(
|
||||
name=agent_uuid, memory=ChatMemory(human="My name is Matt.", persona=persona), tool_ids=[wikipedia_query_tool.id]
|
||||
)
|
||||
print(f"Created agent: {agent_state.name} with ID {str(agent_state.id)}")
|
||||
|
||||
# Send a message to the agent
|
||||
|
||||
@@ -108,7 +108,7 @@ def main():
|
||||
]
|
||||
|
||||
# 4. Create the agent
|
||||
agent_state = setup_agent(client, config_file, agent_uuid=agent_uuid, tools=[t.name for t in tools], tool_rules=tool_rules)
|
||||
agent_state = setup_agent(client, config_file, agent_uuid=agent_uuid, tool_ids=[t.id for t in tools], tool_rules=tool_rules)
|
||||
|
||||
# 5. Ask for the final secret word
|
||||
response = client.user_message(agent_id=agent_state.id, message="What is the fourth secret word?")
|
||||
|
||||
Reference in New Issue
Block a user