fix bug for no media (#168)

landing-ai · Jul 12, 2024 · 0d5b843 · 0d5b843
1 parent e7b3181
commit 0d5b843
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
@@ -655,7 +655,15 @@ def chat_with_workflow(
  int_chat = cast(
  List[Message],
  [
- {"role": c["role"], "content": c["content"], "media": c["media"]}
+ (
+ {
+ "role": c["role"],
+ "content": c["content"],
+ "media": c["media"],
+ }
+ if "media" in c
+ else {"role": c["role"], "content": c["content"]}
+ )
  for c in chat
  ],
  )