feat: refactor agent memory representation and modify routes for editing blocks

alembic/versions/5987401b40ae_refactor_agent_memory.py

@@ -0,0 +1,34 @@
+"""Refactor agent memory
+
+Revision ID: 5987401b40ae
+Revises: 1c8880d671ee
+Create Date: 2024-11-25 14:35:00.896507
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "5987401b40ae"
+down_revision: Union[str, None] = "1c8880d671ee"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column("agents", "tools", new_column_name="tool_names")
+    op.drop_column("agents", "memory")
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column("agents", "tool_names", new_column_name="tools")
+    op.add_column("agents", sa.Column("memory", postgresql.JSON(astext_type=sa.Text()), autoincrement=False, nullable=True))
+    # ### end Alembic commands ###

examples/docs/tools.py

@@ -45,7 +45,7 @@ def roll_d20() -> str:
         TerminalToolRule(tool_name="send_message"),
     ],
 )
-print(f"Created agent with name {agent_state.name} with tools {agent_state.tools}")
+print(f"Created agent with name {agent_state.name} with tools {agent_state.tool_names}")
 
 # Message an agent
 response = client.send_message(agent_id=agent_state.id, role="user", message="roll a dice")

examples/swarm/swarm.py

@@ -3,7 +3,7 @@
 
 import typer
 
-from letta import AgentState, EmbeddingConfig, LLMConfig, create_client
+from letta import EmbeddingConfig, LLMConfig, PersistedAgentState, create_client
 from letta.schemas.agent import AgentType
 from letta.schemas.memory import BasicBlockMemory, Block
 
@@ -32,7 +32,7 @@ def create_agent(
         include_base_tools: Optional[bool] = True,
         # instructions
         instructions: str = "",
-    ) -> AgentState:
+    ) -> PersistedAgentState:
 
         # todo: process tools for agent handoff
         persona_value = (

examples/tool_rule_usage.py

@@ -3,7 +3,7 @@
 
 from letta import create_client
 from letta.schemas.letta_message import FunctionCallMessage
-from letta.schemas.tool_rule import InitToolRule, TerminalToolRule, ToolRule
+from letta.schemas.tool_rule import ChildToolRule, InitToolRule, TerminalToolRule
 from tests.helpers.endpoints_helper import (
     assert_invoked_send_message_with_keyword,
     setup_agent,
@@ -100,10 +100,10 @@ def main():
     # 3. Create the tool rules. It must be called in this order, or there will be an error thrown.
     tool_rules = [
         InitToolRule(tool_name="first_secret_word"),
-        ToolRule(tool_name="first_secret_word", children=["second_secret_word"]),
-        ToolRule(tool_name="second_secret_word", children=["third_secret_word"]),
-        ToolRule(tool_name="third_secret_word", children=["fourth_secret_word"]),
-        ToolRule(tool_name="fourth_secret_word", children=["send_message"]),
+        ChildToolRule(tool_name="first_secret_word", children=["second_secret_word"]),
+        ChildToolRule(tool_name="second_secret_word", children=["third_secret_word"]),
+        ChildToolRule(tool_name="third_secret_word", children=["fourth_secret_word"]),
+        ChildToolRule(tool_name="fourth_secret_word", children=["send_message"]),
         TerminalToolRule(tool_name="send_message"),
     ]
 

letta/__init__.py

@@ -4,7 +4,7 @@
 from letta.client.client import LocalClient, RESTClient, create_client
 
 # imports for easier access
-from letta.schemas.agent import AgentState
+from letta.schemas.agent import AgentState, PersistedAgentState
 from letta.schemas.block import Block
 from letta.schemas.embedding_config import EmbeddingConfig
 from letta.schemas.enums import JobStatus