clean up

letta-ai · Oct 10, 2024 · 046f53c · 046f53c
1 parent 8c900b7
commit 046f53c
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 20 deletions.
diff --git a/letta/agent.py b/letta/agent.py
@@ -1338,10 +1338,6 @@ def retry_message(self) -> List[Message]:
 
 def save_agent(agent: Agent, ms: MetadataStore):
     """Save agent to metadata store"""
-    # from letta.o1_agent import O1Agent
-    # if isinstance(agent, O1Agent):
-    #    assert isinstance(agent.agent.agent_state.memory, Memory), f"Memory object is not of type Memory: {type(letta_agent.agent_state.memory)}"
-
     agent.update_state()
     agent_state = agent.agent_state
     agent_id = agent_state.id

diff --git a/letta/o1_agent.py b/letta/o1_agent.py
@@ -50,8 +50,6 @@ def __init__(
     ):
         super().__init__(interface, agent_state, tools)
         self.max_thinking_steps = max_thinking_steps
-        # self.interface = interface
-        # self.agent_state = agent_state
         self.tools = tools
         self.first_message_verify_mono = first_message_verify_mono
 
@@ -75,9 +73,6 @@ def step(
             tools=self.tools,
             first_message_verify_mono=self.first_message_verify_mono,
         )
-        print("O1 AGENT", self.agent_state.system)
-        print("O1 AGENT", self.agent_state)
-
         response = thinking_agent.step(
             user_message,
             first_message,
@@ -91,11 +86,7 @@ def step(
             ms,
         )
 
-        print("THINKING AGENT", thinking_agent.agent_state.system)
         for _ in range(self.max_thinking_steps):
-            # assert isinstance(self.agent_state.memory, Memory), f"Memory object is not of type Memory: {type(self.agent.agent_state.memory)}"
-
-            # assert isinstance(thinking_agent.agent_state.memory, Memory), f"Memory object is not of type Memory"
             response = thinking_agent.step(
                 None,
                 first_message,
@@ -112,10 +103,3 @@ def step(
             if response.messages[-1].name == "send_final_message":
                 break
         return response
-
-    # def rebuild_memory(self, force=False, update_timestamp=True, ms: Optional[MetadataStore] = None):
-    #    self.agent.rebuild_memory(force, update_timestamp, ms)
-
-    # def update_state(self) -> AgentState:
-    #    updated_state = self.agent.update_state()
-    #    return updated_state