"""
TextGen service module for interacting with the Text Generation WebUI API.
"""

from .client import TextGenClient
from .models import (
    ChatMessage,
    ChatCompletionRequest,
    ChatCompletionResponse,
    CompletionRequest,
    CompletionResponse,
    ModelInfo,
)

__all__ = [
    "TextGenClient",
    "ChatMessage",
    "ChatCompletionRequest",
    "ChatCompletionResponse",
    "CompletionRequest",
    "CompletionResponse",
    "ModelInfo",
]