aboutsummaryrefslogtreecommitdiff
path: root/agent/orchestrator/models/__init__.py
blob: 10b2b5a940fd966da1da74b58f5f0ffe0776c37c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
"""The public surface for model invocation.

Other modules import `call_model` and use it without knowing which backend is
configured. Switching backends means changing one assignment here.
"""

from .base import Message, ModelBackend, ModelRequest, ModelResponse
from .claude_code import ClaudeCodeBackend

_backend: ModelBackend = ClaudeCodeBackend()


def call_model(request: ModelRequest) -> ModelResponse:
    """Invoke the configured model backend."""
    return _backend.call(request)


__all__ = [
    "Message",
    "ModelBackend",
    "ModelRequest",
    "ModelResponse",
    "call_model",
]