Metadata-Version: 2.4
Name: litellm-td-llm-provider
Version: 0.1.2
Summary: LiteLLM provider for Treasure Data LLM Proxy
Classifier: Development Status :: 2 - Pre-Alpha
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: >=3.10
Requires-Dist: httpx>=0.24.0
Requires-Dist: litellm>=1.0.0
Requires-Dist: python-dotenv>=1.0.1
Description-Content-Type: text/markdown

# LiteLLM TD LLM Proxy Provider

[![Python](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
[![PyPI](https://img.shields.io/pypi/v/litellm-td-llm-provider.svg)](https://pypi.org/project/litellm-td-llm-provider/)
[![CI status](https://github.com/treasure-data-ai-labs/litellm-td-llm-provider/actions/workflows/ci.yml/badge.svg)](https://github.com/treasure-data-ai-labs/litellm-td-llm-provider/actions)

Custom LiteLLM provider for Treasure Data's LLM Proxy API.

## Installation

```bash
pip install litellm-td-llm-provider
```

## Quick Start

```python
import litellm
from litellm_td_llm_provider import register_td_provider

# Register the TD provider
register_td_provider()

# Use with LiteLLM
response = litellm.completion(
    model="td/claude-sonnet-4-5",
    messages=[{"role": "user", "content": "Hello!"}],
    api_key="your-td-api-key",
    api_base="https://llm-proxy.us01.treasuredata.com"
)
```

## Environment Variables

```bash
export TD_LLM_API_KEY="your-api-key"
export TD_LLM_SITE="us01"  # Optional, defaults to us01
```

## Features

- ✅ Streaming support
- ✅ Async completion (`acompletion`)
- ✅ Tool calling support
- ✅ Multi-site support
- ✅ OpenAI-compatible interface

## Documentation

For detailed documentation, examples, and development guide, visit:
[https://github.com/treasure-data-ai-labs/litellm-td-llm-provider](https://github.com/treasure-data-ai-labs/litellm-td-llm-provider)

## License

Apache License 2.0
