diff --git a/src/ai_client.py b/src/ai_client.py index 028c4b14..858caf1d 100644 --- a/src/ai_client.py +++ b/src/ai_client.py @@ -2561,7 +2561,8 @@ 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 isinstance(fi, models.FileItem) else models.FileItem.from_dict(fi) + from src.models import FileItem as _FIC + fi_item = fi if isinstance(fi, _FIC) else _FIC.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}"}) @@ -2804,7 +2805,8 @@ 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 isinstance(fi, models.FileItem) else models.FileItem.from_dict(fi) + from src.models import FileItem as _FIC + fi_item = fi if isinstance(fi, _FIC) else _FIC.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}"}) @@ -2896,7 +2898,8 @@ 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 isinstance(fi, models.FileItem) else models.FileItem.from_dict(fi) + from src.models import FileItem as _FIC + fi_item = fi if isinstance(fi, _FIC) else _FIC.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}"})