ai-stream-proxy-sdk 0.2.3

Client SDK for ai-stream-proxy.

ai-stream-proxy-sdk

Import name stays ai_stream_proxy.

Install

uv sync

Test

uv run pytest

Build

uv build

Publish

uv publish --token <pypi-token>

Sync usage

from ai_stream_proxy import StreamProxyClient

client = StreamProxyClient(server_url="http://127.0.0.1:3000")

client.create_stream(
    stream_id="test",
    message_id="message-1",
    content_type="claude-code-stream-json+include-partial-messages",
)

with client.new_stream(stream_id="test") as stream:
    stream.enqueue(b'{"type":"chunk"}\n')
    stream.finish(abort=False, reason="done")

message = client.get_stream_message(stream_id="test")
client.delete_stream(stream_id="test")
client.close()

Async usage

import asyncio

from ai_stream_proxy import AsyncStreamProxyClient


async def main() -> None:
    client = AsyncStreamProxyClient(server_url="http://127.0.0.1:3000")

    await client.create_stream(
        stream_id="test",
        message_id="message-1",
        content_type="claude-code-stream-json+include-partial-messages",
    )

    async with client.new_stream(stream_id="test") as stream:
        stream.enqueue(b'{"type":"chunk"}\n')
        stream.finish(abort=False, reason="done")

    message = await client.get_stream_message(stream_id="test")
    await client.delete_stream(stream_id="test")
    await client.aclose()


asyncio.run(main())