fix unpacking problem

cpacker · Oct 15, 2024 · ae9b0e9 · ae9b0e9
1 parent eeb1460
commit ae9b0e9
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/letta/agent.py b/letta/agent.py
@@ -744,13 +744,11 @@ def step(
         total_usage = UsageStatistics()
         step_count = 0
         while True:
+            kwargs["ms"] = ms
+            kwargs["first_message"] = False
+            kwargs["return_dicts"] = False
             step_response = self.inner_step(
                 messages=next_input_message,
-                first_message=False,
-                skip_verify=skip_verify,
-                return_dicts=False,
-                stream=stream,
-                ms=ms,
                 **kwargs,
             )
             step_response.messages