services/protocol/openai_v1_models.py

from __future__ import annotations

from typing import Any

from services.openai_backend_api import OpenAIBackendAPI
from utils.helper import IMAGE_MODELS


def list_models() -> dict[str, Any]:
    result = OpenAIBackendAPI().list_models()
    data = result.get("data")
    if not isinstance(data, list):
        return result
    seen = {str(item.get("id") or "").strip() for item in data if isinstance(item, dict)}
    for model in sorted(IMAGE_MODELS):
        if model not in seen:
            data.append({
                "id": model,
                "object": "model",
                "created": 0,
                "owned_by": "chatgpt2api",
                "permission": [],
                "root": model,
                "parent": None,
            })
    return result