diff --git a/letta/adapters/letta_llm_request_adapter.py b/letta/adapters/letta_llm_request_adapter.py index 5e472a35..49a3899c 100644 --- a/letta/adapters/letta_llm_request_adapter.py +++ b/letta/adapters/letta_llm_request_adapter.py @@ -127,6 +127,7 @@ class LettaLLMRequestAdapter(LettaLLMAdapter): agent_id=self.agent_id, agent_tags=self.agent_tags, run_id=self.run_id, + call_type="agent_step", org_id=self.org_id, user_id=self.user_id, llm_config=self.llm_config.model_dump() if self.llm_config else None, diff --git a/letta/agents/letta_agent.py b/letta/agents/letta_agent.py index ae0d49bd..fee4b902 100644 --- a/letta/agents/letta_agent.py +++ b/letta/agents/letta_agent.py @@ -420,6 +420,7 @@ class LettaAgent(BaseAgent): agent_id=self.agent_id, agent_tags=agent_state.tags, run_id=self.current_run_id, + call_type="agent_step", org_id=self.actor.organization_id, user_id=self.actor.id, llm_config=self.agent_state.llm_config.model_dump() if self.agent_state.llm_config else None, @@ -773,6 +774,7 @@ class LettaAgent(BaseAgent): agent_id=self.agent_id, agent_tags=agent_state.tags, run_id=self.current_run_id, + call_type="agent_step", org_id=self.actor.organization_id, user_id=self.actor.id, llm_config=self.agent_state.llm_config.model_dump() if self.agent_state.llm_config else None, @@ -1250,6 +1252,7 @@ class LettaAgent(BaseAgent): agent_id=self.agent_id, agent_tags=agent_state.tags, run_id=self.current_run_id, + call_type="agent_step", org_id=self.actor.organization_id, user_id=self.actor.id, llm_config=self.agent_state.llm_config.model_dump() if self.agent_state.llm_config else None, diff --git a/letta/llm_api/llm_api_tools.py b/letta/llm_api/llm_api_tools.py index 280b3eb0..2fb9efba 100644 --- a/letta/llm_api/llm_api_tools.py +++ b/letta/llm_api/llm_api_tools.py @@ -245,6 +245,7 @@ def create( request_json=prepare_openai_payload(data), response_json=response.model_json_schema(), step_id=step_id, + call_type="agent_step", ), )