agent_server/models.py

import os

import agent_server.helpers


def models_payload() -> dict:
    """
    Returns the /v1/models response payload.
    """
    MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen3-1.7B")
    now = agent_server.helpers.now_ts()
    return {
        "object": "list",
        "data": [
            {
                "id": MODEL_NAME,
                "object": "model",
                "created": now,
                "owned_by": "upstream",
            },
            {
                "id": f"{MODEL_NAME}-nothink",
                "object": "model",
                "created": now,
                "owned_by": "upstream",
            },
            {
                "id": "code-writing-agent-without-tools",
                "object": "model",
                "created": now,
                "owned_by": "you",
            },
            {
                "id": "code-writing-agent-with-search",
                "object": "model",
                "created": now,
                "owned_by": "you",
            },
            {
                "id": "tool-calling-agent-with-search-and-code",
                "object": "model",
                "created": now,
                "owned_by": "you",
            },
            {
                "id": "custom-agent-with-beam-design-tools",
                "object": "model",
                "created": now,
                "owned_by": "you",
            },
            {
                "id": "generator-with-managed-critic",
                "object": "model",
                "created": now,
                "owned_by": "you",
            },
            {
                "id": "manager-with-heterogeneous-agents",
                "object": "model",
                "created": now,
                "owned_by": "you",
            },
        ],
    }