store thoughts in agent info without serialization

Author: ollmer

src/agentlab/agents/tapeagent/agent.py

@@ -4,7 +4,6 @@
 
 import bgym
 import hydra
-from pydantic import json
 from tapeagents.agent import Agent
 from tapeagents.core import Action, Observation, Tape, Thought
 
@@ -25,6 +24,11 @@ def make_agent(self) -> bgym.Agent:
         return TapeAgent(agent=agent, tape=Tape(steps=[]))
 
 
+@dataclass
+class TapeAgentInfo(bgym.AgentInfo):
+    thoughts: list[Thought] = None
+
+
 class TapeAgent(bgym.Agent):
     agent: Agent
     tape: Tape
@@ -34,8 +38,8 @@ def __init__(self, agent: Agent, tape: Tape):
         self.agent = agent
         self.tape = tape
 
-    def obs_preprocessor(self, obs: dict) -> Any:
-        logger.info(f"Preprocessing observation: {obs}")
+    def obs_preprocessor(self, obs: Any) -> Any:
+        logger.info(f"Observation: {obs}")
         return obs
 
     def get_action(
@@ -57,4 +61,4 @@ def get_action(
                 action = event.step.llm_view()
                 logger.info(f"Action: {action}")
                 break  # we stop at the first action
-        return (action, bgym.AgentInfo(think=json.dumps(thoughts), stats={}))
+        return (action, TapeAgentInfo(thoughts=thoughts))