fix lint

landing-ai · Sep 10, 2024 · 4e0bfd9 · 4e0bfd9
1 parent 2a8c779
commit 4e0bfd9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
@@ -272,7 +272,7 @@ def chat_with_code(
  artifacts_loaded = artifacts.show()
  int_chat.append({"role": "observation", "content": artifacts_loaded})
  orig_chat.append({"role": "observation", "content": artifacts_loaded})
- self.callback_message({"role": "observation", "content": artifacts_loaded})
+ self.streaming_message({"role": "observation", "content": artifacts_loaded})
 
  while not finished and iterations < self.max_iterations:
  response = run_conversation(self.agent, int_chat)
@@ -285,7 +285,7 @@ def chat_with_code(
  if last_response == response:
  response["let_user_respond"] = True
 
- self.callback_message({"role": "assistant", "content": response})
+ self.streaming_message({"role": "assistant", "content": response})
 
  if response["let_user_respond"]:
  break
@@ -306,7 +306,7 @@ def chat_with_code(
  orig_chat.append(
  {"role": "observation", "content": obs, "execution": result}
  )
- self.callback_message(
+ self.streaming_message(
  {
  "role": "observation",
  "content": obs,