smallestai 4.3.6

No project description provided

SmallestAi Python Library

The SmallestAi Python library provides convenient access to the SmallestAi APIs from Python.

Table of Contents

• Installation
• Reference
• Usage
• Async Client
• Exception Handling
• Streaming
• Websockets
• Advanced
  • Access Raw Response Data
  • Retries
  • Timeouts
  • Custom Client
• Contributing

Installation

pip install smallestai

Reference

A full reference for this library is available here.

Usage

Instantiate and use the client with the following:

from smallestai import SmallestAI

client = SmallestAI(
    token="YOUR_TOKEN",
)
client.atoms.agent_templates.create_agent_from_template(
    agent_name="agentName",
    template_id="templateId",
)

Async Client

The SDK also exports an async client so that you can make non-blocking calls to our API. Note that if you are constructing an Async httpx client class to pass into this client, use httpx.AsyncClient() instead of httpx.Client() (e.g. for the httpx_client parameter of this client).

import asyncio

from smallestai import AsyncSmallestAI

client = AsyncSmallestAI(
    token="YOUR_TOKEN",
)


async def main() -> None:
    await client.atoms.agent_templates.create_agent_from_template(
        agent_name="agentName",
        template_id="templateId",
    )


asyncio.run(main())

Exception Handling

When the API returns a non-success status code (4xx or 5xx response), a subclass of the following error will be thrown.

from smallestai.core.api_error import ApiError

try:
    client.atoms.agent_templates.create_agent_from_template(...)
except ApiError as e:
    print(e.status_code)
    print(e.body)

Streaming

The SDK supports streaming responses, as well, the response will be a generator that you can loop over.

from smallestai import SmallestAI

client = SmallestAI(
    token="YOUR_TOKEN",
)
response = client.waves.lightning_large.stream_lightning_large_speech(
    text="text",
    voice_id="voice_id",
)
for chunk in response.data:
    yield chunk

Websockets

The SDK supports both sync and async websocket connections for real-time, low-latency communication. Sockets can be created using the connect method, which returns a context manager. You can either iterate through the returned SocketClient to process messages as they arrive, or attach handlers to respond to specific events.

# Connect to the websocket (Sync)
import threading

from smallestai import SmallestAI

client = SmallestAI(...)

with client.streaming_tts.connect() as socket:
    # Iterate over the messages as they arrive
    for message in socket
        print(message)

    # Or, attach handlers to specific events
    socket.on(EventType.OPEN, lambda _: print("open"))
    socket.on(EventType.MESSAGE, lambda message: print("received message", message))
    socket.on(EventType.CLOSE, lambda _: print("close"))
    socket.on(EventType.ERROR, lambda error: print("error", error))


    # Start the listening loop in a background thread
    listener_thread = threading.Thread(target=socket.start_listening, daemon=True)
    listener_thread.start()

# Connect to the websocket (Async)
import asyncio

from smallestai import AsyncSmallestAI

client = AsyncSmallestAI(...)

async with client.streaming_tts.connect() as socket:
    # Iterate over the messages as they arrive
    async for message in socket
        print(message)

    # Or, attach handlers to specific events
    socket.on(EventType.OPEN, lambda _: print("open"))
    socket.on(EventType.MESSAGE, lambda message: print("received message", message))
    socket.on(EventType.CLOSE, lambda _: print("close"))
    socket.on(EventType.ERROR, lambda error: print("error", error))


    # Start listening for events in an asyncio task
    listen_task = asyncio.create_task(socket.start_listening())

Advanced

Access Raw Response Data

The SDK provides access to raw response data, including headers, through the .with_raw_response property. The .with_raw_response property returns a "raw" client that can be used to access the .headers and .data attributes.

from smallestai import SmallestAI

client = SmallestAI(
    ...,
)
response = (
    client.atoms.agent_templates.with_raw_response.create_agent_from_template(
        ...
    )
)
print(response.headers)  # access the response headers
print(response.status_code)  # access the response status code
print(response.data)  # access the underlying object
with client.waves.lightning_large.with_raw_response.stream_lightning_large_speech(
    ...
) as response:
    print(
        response.headers
    )  # access the response headersprint(response.status_code)  # access the response status code
    for chunk in response.data:
        print(chunk)  # access the underlying object(s)

Retries

The SDK is instrumented with automatic retries with exponential backoff. A request will be retried as long as the request is deemed retryable and the number of retry attempts has not grown larger than the configured retry limit (default: 2).

A request is deemed retryable when any of the following HTTP status codes is returned:

• 408 (Timeout)
• 429 (Too Many Requests)
• 5XX (Internal Server Errors)

Use the max_retries request option to configure this behavior.

client.atoms.agent_templates.create_agent_from_template(..., request_options={
    "max_retries": 1
})

Timeouts

The SDK defaults to a 60 second timeout. You can configure this with a timeout option at the client or request level.

from smallestai import SmallestAI

client = SmallestAI(
    ...,
    timeout=20.0,
)


# Override timeout for a specific method
client.atoms.agent_templates.create_agent_from_template(..., request_options={
    "timeout_in_seconds": 1
})

Custom Client

You can override the httpx client to customize it for your use-case. Some common use-cases include support for proxies and transports.

import httpx
from smallestai import SmallestAI

client = SmallestAI(
    ...,
    httpx_client=httpx.Client(
        proxy="http://my.test.proxy.example.com",
        transport=httpx.HTTPTransport(local_address="0.0.0.0"),
    ),
)

Contributing

While we value open-source contributions to this SDK, this library is generated programmatically. Additions made directly to this library would have to be moved over to our generation code, otherwise they would be overwritten upon the next generated release. Feel free to open a PR as a proof of concept, but know that we will not be able to merge it as-is. We suggest opening an issue first to discuss with us!

On the other hand, contributions to the README are always very welcome!