routes/chat.py

Rama main · réplica de lectura del repositorio original.

"""
routes/chat.py — Lightweight text completion for canvas pages.

POST /api/chat  { message: "..." }  →  { response: "..." }

Uses Z.AI (glm-5-turbo) for text rewriting / enhancement.
NEVER Groq — Groq is TTS only, never for LLM.
"""

import logging
import os

from flask import Blueprint, jsonify, request

from services.zai_direct import zai_messages_post

logger = logging.getLogger(__name__)

chat_bp = Blueprint('chat', __name__)


@chat_bp.route('/api/chat', methods=['POST'])
def chat_complete():
    data = request.get_json(silent=True) or {}
    message = (data.get('message') or '').strip()
    if not message:
        return jsonify({'error': 'No message provided'}), 400

    api_key = os.environ.get('ZAI_API_KEY', '')
    fallback_key = os.environ.get('ZAI_FALLBACK_API_KEY', '')
    if not api_key and not fallback_key:
        return jsonify({'error': 'ZAI_API_KEY not configured'}), 500

    try:
        r = zai_messages_post(
            {
                'model': 'glm-5-turbo',
                'messages': [{'role': 'user', 'content': message}],
                'max_tokens': 1024,
            },
            timeout=30,
        )
        r.raise_for_status()
        text = r.json()['content'][0]['text'].strip()
        return jsonify({'response': text})
    except Exception as exc:
        logger.exception('chat completion error')
        return jsonify({'error': str(exc)}), 500