nlproxy/server/apis/models.py

"""Model discovery endpoints.

Author: IntelliDeep Labs Team
License: BSL 1.1
"""

from __future__ import annotations

from fastapi import APIRouter

router = APIRouter()


@router.get("/v1/models", tags=["Models"])
async def list_models() -> dict:
    return {
        "object": "list",
        "data": [
            {
                "id": "gpt-4",
                "object": "model",
                "created": 1687882800,
                "owned_by": "openai",
                "permission": [],
                "root": "gpt-4",
                "parent": None,
            },
            {
                "id": "gpt-4-turbo",
                "object": "model",
                "created": 1687882800,
                "owned_by": "openai",
                "permission": [],
                "root": "gpt-4-turbo",
                "parent": None,
            },
            {
                "id": "gpt-3.5-turbo",
                "object": "model",
                "created": 1687882800,
                "owned_by": "openai",
                "permission": [],
                "root": "gpt-3.5-turbo",
                "parent": None,
            },
            {
                "id": "claude-3-opus",
                "object": "model",
                "created": 1687882800,
                "owned_by": "anthropic",
                "permission": [],
                "root": "claude-3-opus",
                "parent": None,
            },
            {
                "id": "gemini-pro",
                "object": "model",
                "created": 1687882800,
                "owned_by": "google",
                "permission": [],
                "root": "gemini-pro",
                "parent": None,
            },
        ],
    }