remove prints

This commit is contained in:
Xi Yan 2025-02-28 12:50:09 -08:00
parent 7bb5f2bf0e
commit 3c8dd98c03

View file

@ -107,7 +107,6 @@ class AgentPersistence:
return ToolExecutionStep(**json.loads(value)) if value else None return ToolExecutionStep(**json.loads(value)) if value else None
async def set_num_infer_iters_in_turn(self, session_id: str, turn_id: str, num_infer_iters: int): async def set_num_infer_iters_in_turn(self, session_id: str, turn_id: str, num_infer_iters: int):
print(f"setting num_infer_iters_in_turn: {num_infer_iters}")
await self.kvstore.set( await self.kvstore.set(
key=f"num_infer_iters_in_turn:{self.agent_id}:{session_id}:{turn_id}", key=f"num_infer_iters_in_turn:{self.agent_id}:{session_id}:{turn_id}",
value=str(num_infer_iters), value=str(num_infer_iters),
@ -117,5 +116,4 @@ class AgentPersistence:
value = await self.kvstore.get( value = await self.kvstore.get(
key=f"num_infer_iters_in_turn:{self.agent_id}:{session_id}:{turn_id}", key=f"num_infer_iters_in_turn:{self.agent_id}:{session_id}:{turn_id}",
) )
print(f"getting num_infer_iters_in_turn: {value}")
return int(value) if value else None return int(value) if value else None