diff --git a/src/ai_client.py b/src/ai_client.py index 12171924..70282116 100644 --- a/src/ai_client.py +++ b/src/ai_client.py @@ -2562,7 +2562,7 @@ def _send_grok(md_content: str, user_message: str, base_dir: str, if file_items: for fi in file_items: if fi.get("is_image") and fi.get("base64_data"): - fi_item = fi if hasattr(fi, 'path') else models.FileItem(path=fi.get('path', 'attachment')) + fi_item = fi if isinstance(fi, models.FileItem) else models.FileItem.from_dict(fi) user_content = f"[IMAGE: {fi_item.path or 'attachment'}]\n{user_content}" if discussion_history and not history: history.append({"role": "user", "content": f"[DISCUSSION HISTORY]\n\n{discussion_history}\n\n---\n\n{user_message}"}) @@ -2805,7 +2805,7 @@ def _send_qwen(md_content: str, user_message: str, base_dir: str, if file_items: for fi in file_items: if fi.get("is_image") and fi.get("base64_data"): - fi_item = fi if hasattr(fi, 'path') else models.FileItem(path=fi.get('path', 'attachment')) + fi_item = fi if isinstance(fi, models.FileItem) else models.FileItem.from_dict(fi) user_content = f"[IMAGE: {fi_item.path or 'attachment'}]\n{user_content}" if discussion_history and not history: history.append({"role": "user", "content": f"[DISCUSSION HISTORY]\n\n{discussion_history}\n\n---\n\n{user_message}"}) @@ -2897,7 +2897,7 @@ def _send_llama(md_content: str, user_message: str, base_dir: str, if file_items: for fi in file_items: if fi.get("is_image") and fi.get("base64_data"): - fi_item = fi if hasattr(fi, 'path') else models.FileItem(path=fi.get('path', 'attachment')) + fi_item = fi if isinstance(fi, models.FileItem) else models.FileItem.from_dict(fi) user_content = f"[IMAGE: {fi_item.path or 'attachment'}]\n{user_content}" if discussion_history and not history: history.append({"role": "user", "content": f"[DISCUSSION HISTORY]\n\n{discussion_history}\n\n---\n\n{user_message}"})