fix message bug

app.py

@@ -9,6 +9,7 @@ from gradio import mount_gradio_app
 import config
 from model import engine
 from ui import create_ui
+from utils import get_clean_text
 
 # --- FastAPI Setup ---
 app = FastAPI()
@@ -29,7 +30,17 @@ async def chat_completions(request: Request):
 
     try:
         data = await request.json()
-        messages = data.get("messages", [])
+        raw_messages = data.get("messages", [])
+
+        messages = []
+        for msg in raw_messages:
+            messages.append(
+                {
+                    "role": msg.get("role", "user"),
+                    "content": get_clean_text(msg.get("content")),
+                }
+            )
+
         stream = data.get("stream", True)
         temperature = data.get("temperature", config.DEFAULT_TEMP)
         max_tokens = data.get("max_tokens", config.DEFAULT_MAX_TOKENS)

utils.py

@@ -1,21 +1,34 @@
-def sanitize_content(content):
+def get_clean_text(content):
     """
-    Гарантирует, что контент - это строка.
-    Исправляет баг Gradio, когда текст приходит как список.
+    Извлекает чистый текст из любого формата (str, dict, list).
+    Поддерживает:
+    - Простые строки
+    - Gradio формат: {'text': '...'}
+    - OpenAI/Zed формат: [{'type': 'text', 'text': '...'}]
     """
-    if isinstance(content, list):
-        return "\n".join(str(item) for item in content)
-    return str(content) if content is not None else ""
+    if content is None:
+        return ""
 
+    # 1. Если это уже строка — возвращаем как есть
+    if isinstance(content, str):
+        return content
 
-def get_clean_text(content):
-    """
-    Извлекает текст из разных форматов Gradio 5 (словарь, список, строка).
-    Решает проблему отображения {'text': ...}
-    """
-    # Если пришел словарь {'text': '...', 'type': 'text'}
-    if isinstance(content, dict) and "text" in content:
-        return str(content["text"])
+    # 2. Если это словарь (Gradio 5 или часть OpenAI message)
+    if isinstance(content, dict):
+        return content.get("text", "")
+
+    # 3. Если это список (Zed, VS Code, OpenAI Vision)
+    if isinstance(content, list):
+        text_parts = []
+        for item in content:
+            if isinstance(item, str):
+                text_parts.append(item)
+            elif isinstance(item, dict):
+                # Извлекаем 'text' если он есть, иначе игнорируем (например, картинки)
+                text_parts.append(item.get("text", ""))
+            else:
+                text_parts.append(str(item))
+        return "".join(text_parts)
 
-    # Если пришла строка или список, используем вашу функцию из utils.py
-    return sanitize_content(content)
+    # 4. Фолбэк для всего остального
+    return str(content)