fix: address second conversation issue (#241)

* fix: address second conversation issue

* update poetry lock

* fix

pyproject.toml

@@ -24,6 +24,7 @@ requests = "2.*"
 tqdm = ">=4.64.0,<5.0.0"
 pandas = "2.*"
 openai = "1.*"
+flake8 = "^7.0.0"
 typing_extensions = "4.*"
 opencv-python = "4.*"
 tabulate = "^0.9.0"
@@ -47,7 +48,6 @@ av = "^11.0.0"
 autoflake = "1.*"
 pytest = "7.*"
 black = ">=23,<25"
-flake8 = "5.*"
 isort = "5.*"
 responses = "^0.23.1"
 mypy = "<1.8.0"

vision_agent/agent/vision_agent.py

@@ -276,8 +276,11 @@ def chat_with_code(
  # sometimes it gets stuck in a loop, so we force it to exit
  if last_response == response:
  response["let_user_respond"] = True
-
- self.streaming_message({"role": "assistant", "content": response})
+ self.streaming_message(
+ {"role": "assistant", "error": "Stuck in loop"}
+ )
+ else:
+ self.streaming_message({"role": "assistant", "content": response})
 
  if response["let_user_respond"]:
  break

vision_agent/utils/execute.py

@@ -104,7 +104,7 @@ class Result:
  latex: Optional[str] = None
  json: Optional[Dict[str, Any]] = None
  javascript: Optional[str] = None
- artifact_name: Optional[str] = None
+ artifact: Optional[str] = None
  extra: Optional[Dict[str, Any]] = None
  "Extra data that can be included. Not part of the standard types."