fixed type errors

landing-ai · Aug 29, 2024 · 499733b · 499733b
1 parent 1d0dcc1
commit 499733b
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 6 deletions.
diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
@@ -198,7 +198,7 @@ def chat_with_code(
  for chat_i in int_chat:
  if "media" in chat_i:
  for media in chat_i["media"]:
- media = code_interpreter.upload_file(media)
+ media = code_interpreter.upload_file(cast(str, media))
  chat_i["content"] += f" Media name {media}" # type: ignore
  # Save dummy value for now since we just need to know the path
  # name in the key 'media'. Later on we can add artifact support
@@ -264,8 +264,12 @@ def chat_with_code(
 
  if self.verbosity >= 1:
  _LOGGER.info(obs)
- int_chat.append({"role": "observation", "content": obs, "execution": result})
- orig_chat.append({"role": "observation", "content": obs, "execution": result})
+ int_chat.append(
+ {"role": "observation", "content": obs, "execution": result}
+ )
+ orig_chat.append(
+ {"role": "observation", "content": obs, "execution": result}
+ )
 
  iterations += 1
  last_response = response

diff --git a/vision_agent/agent/vision_agent_coder.py b/vision_agent/agent/vision_agent_coder.py
@@ -718,7 +718,7 @@ def chat_with_workflow(
  for chat_i in chat:
  if "media" in chat_i:
  for media in chat_i["media"]:
- media = code_interpreter.upload_file(media)
+ media = code_interpreter.upload_file(cast(str, media))
  chat_i["content"] += f" Media name {media}" # type: ignore
  media_list.append(str(media))
 

diff --git a/vision_agent/lmm/lmm.py b/vision_agent/lmm/lmm.py
@@ -132,7 +132,7 @@ def chat(
  fixed_c["content"] = [{"type": "text", "text": c["content"]}] # type: ignore
  if "media" in c:
  for media in c["media"]:
- encoded_media = encode_media(media)
+ encoded_media = encode_media(cast(str, media))
 
  fixed_c["content"].append( # type: ignore
  {
@@ -379,7 +379,9 @@ def chat(
  fixed_chat = []
  for message in chat:
  if "media" in message:
- message["images"] = [encode_media(m) for m in message["media"]]
+ message["images"] = [
+ encode_media(cast(str, m)) for m in message["media"]
+ ]
  del message["media"]
  fixed_chat.append(message)
  url = f"{self.url}/chat"