sogni-client 5.27.1__tar.gz

sogni_client-5.27.1/.gitignore

@@ -0,0 +1,19 @@
+.venv/
+.pytest_cache/
+.ruff_cache/
+__pycache__/
+*.py[cod]
+*.egg-info/
+.coverage
+htmlcov/
+build/
+dist/
+.env
+.env.*
+!.env.example
+*.key
+*.pem
+*.p12
+*.pfx
+.pypirc
+.DS_Store

sogni_client-5.27.1/LICENSE

@@ -0,0 +1,15 @@
+ISC License
+
+Copyright (c) 2026 Sogni AI
+
+Permission to use, copy, modify, and/or distribute this software for any
+purpose with or without fee is hereby granted, provided that the above
+copyright notice and this permission notice appear in all copies.
+
+THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
+REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
+OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+PERFORMANCE OF THIS SOFTWARE.

sogni_client-5.27.1/PKG-INFO

@@ -0,0 +1,284 @@
+Metadata-Version: 2.5
+Name: sogni-client
+Version: 5.27.1
+Summary: Async Python SDK for image, video, audio, and LLM inference on the Sogni Supernet
+Project-URL: Homepage, https://www.sogni.ai
+Project-URL: Documentation, https://docs.sogni.ai/sogni-sdk/python/
+Project-URL: Repository, https://github.com/Sogni-AI/sogni-client-python
+Author: Sogni AI
+License-Expression: ISC
+License-File: LICENSE
+Keywords: generative-ai,image-generation,llm,sogni,video-generation
+Classifier: Development Status :: 4 - Beta
+Classifier: Framework :: AsyncIO
+Classifier: License :: OSI Approved :: ISC License (ISCL)
+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: Programming Language :: Python :: 3.14
+Requires-Python: >=3.10
+Requires-Dist: eth-account<1,>=0.13
+Requires-Dist: httpx<1,>=0.27
+Requires-Dist: websockets<17,>=13
+Provides-Extra: dev
+Requires-Dist: build>=1.2; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.24; extra == 'dev'
+Requires-Dist: pytest>=8.3; extra == 'dev'
+Requires-Dist: ruff>=0.9; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# Sogni Client for Python
+
+An async Python SDK for image, video, audio, and LLM inference on the Sogni
+Supernet. It follows the public surface and wire protocol of the TypeScript
+`sogni-client`, while using Python naming conventions and async iterators.
+
+> The Python port is currently beta. Keep credentials in environment variables
+> or your system keychain; never commit them to source control.
+
+[Official quickstart](https://docs.sogni.ai/sogni-sdk/python/) ·
+[Examples](https://github.com/Sogni-AI/sogni-client-python/tree/main/examples) ·
+[Sogni API reference](https://docs.sogni.ai/api-reference/)
+
+## Install
+
+Install the latest beta directly from the official GitHub repository:
+
+```bash
+python -m pip install "sogni-client @ git+https://github.com/Sogni-AI/sogni-client-python.git@main"
+```
+
+For an editable source checkout:
+
+```bash
+git clone https://github.com/Sogni-AI/sogni-client-python.git
+cd sogni-client-python
+python -m pip install -e .
+```
+
+Python 3.10 or newer is required.
+
+## Create an image
+
+```python
+import asyncio
+import os
+
+from sogni_client import SogniClient
+
+
+async def main() -> None:
+    async with await SogniClient.create(api_key=os.environ["SOGNI_API_KEY"]) as sogni:
+        project = await sogni.projects.create(
+            type="image",
+            model_id="z_image_turbo_bf16",
+            positive_prompt="A tiny observatory above a sea of clouds",
+            negative_prompt="text, watermark",
+            number_of_media=1,
+            width=1024,
+            height=1024,
+            steps=8,
+        )
+        print(await project.wait_for_completion())
+
+
+asyncio.run(main())
+```
+
+`SogniClient.create()` generates a unique application ID when one is not
+provided. Pass `app_id="..."` when you deliberately need a stable socket
+identity.
+
+## Edit an image with Krea 2 Identity Edit
+
+Pass one or two local reference images through `context_images`. For two-image
+edits, place the base scene first and the identity or detail reference second.
+
+```python
+project = await sogni.projects.create(
+    type="image",
+    model_id="krea2_identity_edit_v1_2",
+    positive_prompt=(
+        "Change only the jacket to vivid sapphire blue. Preserve the exact "
+        "facial identity, expression, framing, background, and lighting."
+    ),
+    number_of_media=1,
+    width=1024,
+    height=1024,
+    steps=10,
+    guidance=1,
+    token_type="spark",
+    context_images=["reference.png"],
+)
+print(await project.wait_for_completion(timeout=900))
+```
+
+The runnable example accepts one or two image paths and can also create a batch:
+
+```bash
+python examples/krea_identity_edit.py reference.png \
+  --prompt "Change only the jacket to vivid sapphire blue; preserve identity."
+
+python examples/krea_identity_edit.py scene.png identity.png \
+  --prompt "Use the first image as the base scene and the second for identity." \
+  --count 4
+```
+
+## Chat
+
+Socket-backed completion:
+
+```python
+result = await sogni.chat.completions.create(
+    model="qwen3.6-35b-a3b-gguf-iq4xs",
+    messages=[{"role": "user", "content": "Give me three visual concepts."}],
+)
+print(result["content"])
+```
+
+Hosted OpenAI-compatible completion:
+
+```python
+result = await sogni.chat.hosted.create(
+    model="qwen3.6-35b-a3b-gguf-iq4xs",
+    messages=[{"role": "user", "content": "Describe a surreal album cover."}],
+)
+```
+
+For streaming socket chat, pass `stream=True` and iterate over the returned
+`ChatStream` with `async for`.
+
+## Durable workflows
+
+```python
+workflow = await sogni.workflows.start(
+    input={"prompt": "Create a four-panel character turnaround"},
+    idempotency_key="turnaround-001",
+)
+
+async for event in sogni.workflows.stream_events(workflow["id"]):
+    print(event["event"], event["data"])
+```
+
+The client also exposes:
+
+- `sogni.account` for authentication, balances, rewards, transactions, and subscriptions
+- `sogni.projects` for generation, uploads, model discovery, and estimates
+- `sogni.chat` for socket, hosted, tool, and durable-run APIs
+- `sogni.workflows` and `sogni.workflows.templates`
+- `sogni.replay` and `sogni.stats`
+
+Python `snake_case` arguments are preferred. Common JavaScript-style aliases
+remain accepted to simplify migration.
+
+## Resuming projects after a reconnect
+
+Generation keeps running on the Supernet while your socket is down. A dropped
+connection is a transport gap, not a failure: tracked projects stay alive, the
+client reconnects with capped exponential backoff for as long as the session is
+authenticated, and on every `authenticated` handshake it reconciles with the
+server. Whatever the client missed is replayed through the normal `project` /
+`job` events, so listeners attached before the gap keep receiving updates and
+`wait_for_completion()` still resolves.
+
+Projects the server knows about but this client does not (a restart, a second
+client sharing the account, cleared local state) are rebuilt as tracked
+`Project` instances with `project.recovered is True`. Their `params` are
+reconstructed from the original request; asset inputs are not recoverable.
+
+```python
+# Every reconciliation reports what changed. `snapshot` is the raw server view,
+# for apps that keep their own project store.
+sogni.projects.on("projectsSynced", lambda r: print(r["reason"], r["active"], r["lost"]))
+
+# In-flight projects this client was not tracking; they are tracked now, so
+# `project` / `job` events follow as usual.
+sogni.projects.on("activeProjectsRecovered", lambda projects: ...)
+
+# Projects that finished while this client was away, result URLs already resolved.
+sogni.projects.on("completedProjectsRecovered", lambda projects: ...)
+
+# Ask for a fresh reconciliation yourself, e.g. after waking from sleep.
+await sogni.projects.sync()
+```
+
+A project the server no longer lists is looked up on the REST API (which only
+stores finished projects) a few times before it is declared lost; it then fails
+with an error where `is_project_lost_error(error)` is `True`. Apps that persist
+project ids themselves can run the same lookup with
+`sogni.projects.resolve_missing(ids)`.
+
+The same snapshot answers "is anything rendering elsewhere on this account?" —
+`sogni.projects.list_projects_elsewhere()` returns those in-flight projects
+read-only (`appSource`, `status`, `model`, per-job step counts). The socket
+rate-limits it to 20 calls per 10s per account, so poll on the order of tens of
+seconds.
+
+Recovery is per app instance: the server hands projects back to the `appId` that
+created them, so persist your `appId` and reuse it across restarts.
+
+## Sensitive content
+
+`job.is_nsfw` means the server **withheld** the media: the render ran with the
+Sensitive Content Filter on, a signal fired, and there is nothing to download.
+When the artist turns the filter off the media is delivered and merely labelled
+— that case reports `job.nsfw_detected` with `job.nsfw_sources` (`prompt`
+and/or `image`), has a `result_url` like any other result, and leaves
+`job.is_nsfw` false. Use `job.has_result_media` (or `job.is_withheld`) to decide
+whether media exists, and the viewer's own filter setting to decide whether to
+blur it.
+
+## Compatibility
+
+This release tracks the current TypeScript source at `5.27.1`. The
+REST, WebSocket, and SSE contracts are covered by credential-free protocol
+tests, including authentication refresh, uploads, project state recovery,
+streaming chat, workflows, templates, replay, and the canonical 25 hosted-tool
+schemas.
+
+Current model and transport coverage includes LTX 2.5, MiniMax H3 in all four
+tiers (Standard, 8-step Balanced, 4-step LightX2V Turbo, and the separate
+FastH3 `fastvideo-int8` Turbo engine), Seedance 2.5, Wan 3 and Wan 3.0 Enhanced,
+RTX VSR, MiniMax Music 3, LoRA catalog discovery, queue start estimates,
+live-benchmarked render/total time on cost quotes, in-flight project recovery
+across reconnects, confirmed cancellation, and connection/workload attribution.
+
+The Python API is async-first; `AsyncSogniClient` is an alias of
+`SogniClient`, not a synchronous wrapper. Browser-only cookie coordination and
+multi-tab behavior have no Python equivalent. Local image references are
+uploaded with their detected MIME type, but the TypeScript client's optional
+browser-side image resizing is not reproduced. All 25 canonical tool schemas
+are exposed; the local project-backed executor handles the six direct media
+generation tools, while the remaining tools run through the hosted or durable
+chat APIs. Live, credentialed smoke tests are intentionally separate from the
+default test suite.
+
+## Token authentication
+
+```python
+sogni = await SogniClient.create(auth_type="token")
+await sogni.set_tokens(token=access_token, refresh_token=refresh_token)
+```
+
+Username/password login and signing are available through `sogni.account.login`.
+API-key use does not require storing a wallet password.
+
+## Development
+
+```bash
+python -m pip install -e '.[dev]'
+pytest
+ruff check sogni_client tests
+ruff format --check sogni_client tests
+python -m build
+```
+
+Live integration tests require explicit credentials and are not run by default.
+
+## Documentation
+
+- [Python SDK quickstart](https://docs.sogni.ai/sogni-sdk/python/)
+- [Sogni SDK overview](https://docs.sogni.ai/sogni-sdk/)
+- [REST API reference](https://docs.sogni.ai/api-reference/)

sogni_client-5.27.1/README.md

@@ -0,0 +1,253 @@
+# Sogni Client for Python
+
+An async Python SDK for image, video, audio, and LLM inference on the Sogni
+Supernet. It follows the public surface and wire protocol of the TypeScript
+`sogni-client`, while using Python naming conventions and async iterators.
+
+> The Python port is currently beta. Keep credentials in environment variables
+> or your system keychain; never commit them to source control.
+
+[Official quickstart](https://docs.sogni.ai/sogni-sdk/python/) ·
+[Examples](https://github.com/Sogni-AI/sogni-client-python/tree/main/examples) ·
+[Sogni API reference](https://docs.sogni.ai/api-reference/)
+
+## Install
+
+Install the latest beta directly from the official GitHub repository:
+
+```bash
+python -m pip install "sogni-client @ git+https://github.com/Sogni-AI/sogni-client-python.git@main"
+```
+
+For an editable source checkout:
+
+```bash
+git clone https://github.com/Sogni-AI/sogni-client-python.git
+cd sogni-client-python
+python -m pip install -e .
+```
+
+Python 3.10 or newer is required.
+
+## Create an image
+
+```python
+import asyncio
+import os
+
+from sogni_client import SogniClient
+
+
+async def main() -> None:
+    async with await SogniClient.create(api_key=os.environ["SOGNI_API_KEY"]) as sogni:
+        project = await sogni.projects.create(
+            type="image",
+            model_id="z_image_turbo_bf16",
+            positive_prompt="A tiny observatory above a sea of clouds",
+            negative_prompt="text, watermark",
+            number_of_media=1,
+            width=1024,
+            height=1024,
+            steps=8,
+        )
+        print(await project.wait_for_completion())
+
+
+asyncio.run(main())
+```
+
+`SogniClient.create()` generates a unique application ID when one is not
+provided. Pass `app_id="..."` when you deliberately need a stable socket
+identity.
+
+## Edit an image with Krea 2 Identity Edit
+
+Pass one or two local reference images through `context_images`. For two-image
+edits, place the base scene first and the identity or detail reference second.
+
+```python
+project = await sogni.projects.create(
+    type="image",
+    model_id="krea2_identity_edit_v1_2",
+    positive_prompt=(
+        "Change only the jacket to vivid sapphire blue. Preserve the exact "
+        "facial identity, expression, framing, background, and lighting."
+    ),
+    number_of_media=1,
+    width=1024,
+    height=1024,
+    steps=10,
+    guidance=1,
+    token_type="spark",
+    context_images=["reference.png"],
+)
+print(await project.wait_for_completion(timeout=900))
+```
+
+The runnable example accepts one or two image paths and can also create a batch:
+
+```bash
+python examples/krea_identity_edit.py reference.png \
+  --prompt "Change only the jacket to vivid sapphire blue; preserve identity."
+
+python examples/krea_identity_edit.py scene.png identity.png \
+  --prompt "Use the first image as the base scene and the second for identity." \
+  --count 4
+```
+
+## Chat
+
+Socket-backed completion:
+
+```python
+result = await sogni.chat.completions.create(
+    model="qwen3.6-35b-a3b-gguf-iq4xs",
+    messages=[{"role": "user", "content": "Give me three visual concepts."}],
+)
+print(result["content"])
+```
+
+Hosted OpenAI-compatible completion:
+
+```python
+result = await sogni.chat.hosted.create(
+    model="qwen3.6-35b-a3b-gguf-iq4xs",
+    messages=[{"role": "user", "content": "Describe a surreal album cover."}],
+)
+```
+
+For streaming socket chat, pass `stream=True` and iterate over the returned
+`ChatStream` with `async for`.
+
+## Durable workflows
+
+```python
+workflow = await sogni.workflows.start(
+    input={"prompt": "Create a four-panel character turnaround"},
+    idempotency_key="turnaround-001",
+)
+
+async for event in sogni.workflows.stream_events(workflow["id"]):
+    print(event["event"], event["data"])
+```
+
+The client also exposes:
+
+- `sogni.account` for authentication, balances, rewards, transactions, and subscriptions
+- `sogni.projects` for generation, uploads, model discovery, and estimates
+- `sogni.chat` for socket, hosted, tool, and durable-run APIs
+- `sogni.workflows` and `sogni.workflows.templates`
+- `sogni.replay` and `sogni.stats`
+
+Python `snake_case` arguments are preferred. Common JavaScript-style aliases
+remain accepted to simplify migration.
+
+## Resuming projects after a reconnect
+
+Generation keeps running on the Supernet while your socket is down. A dropped
+connection is a transport gap, not a failure: tracked projects stay alive, the
+client reconnects with capped exponential backoff for as long as the session is
+authenticated, and on every `authenticated` handshake it reconciles with the
+server. Whatever the client missed is replayed through the normal `project` /
+`job` events, so listeners attached before the gap keep receiving updates and
+`wait_for_completion()` still resolves.
+
+Projects the server knows about but this client does not (a restart, a second
+client sharing the account, cleared local state) are rebuilt as tracked
+`Project` instances with `project.recovered is True`. Their `params` are
+reconstructed from the original request; asset inputs are not recoverable.
+
+```python
+# Every reconciliation reports what changed. `snapshot` is the raw server view,
+# for apps that keep their own project store.
+sogni.projects.on("projectsSynced", lambda r: print(r["reason"], r["active"], r["lost"]))
+
+# In-flight projects this client was not tracking; they are tracked now, so
+# `project` / `job` events follow as usual.
+sogni.projects.on("activeProjectsRecovered", lambda projects: ...)
+
+# Projects that finished while this client was away, result URLs already resolved.
+sogni.projects.on("completedProjectsRecovered", lambda projects: ...)
+
+# Ask for a fresh reconciliation yourself, e.g. after waking from sleep.
+await sogni.projects.sync()
+```
+
+A project the server no longer lists is looked up on the REST API (which only
+stores finished projects) a few times before it is declared lost; it then fails
+with an error where `is_project_lost_error(error)` is `True`. Apps that persist
+project ids themselves can run the same lookup with
+`sogni.projects.resolve_missing(ids)`.
+
+The same snapshot answers "is anything rendering elsewhere on this account?" —
+`sogni.projects.list_projects_elsewhere()` returns those in-flight projects
+read-only (`appSource`, `status`, `model`, per-job step counts). The socket
+rate-limits it to 20 calls per 10s per account, so poll on the order of tens of
+seconds.
+
+Recovery is per app instance: the server hands projects back to the `appId` that
+created them, so persist your `appId` and reuse it across restarts.
+
+## Sensitive content
+
+`job.is_nsfw` means the server **withheld** the media: the render ran with the
+Sensitive Content Filter on, a signal fired, and there is nothing to download.
+When the artist turns the filter off the media is delivered and merely labelled
+— that case reports `job.nsfw_detected` with `job.nsfw_sources` (`prompt`
+and/or `image`), has a `result_url` like any other result, and leaves
+`job.is_nsfw` false. Use `job.has_result_media` (or `job.is_withheld`) to decide
+whether media exists, and the viewer's own filter setting to decide whether to
+blur it.
+
+## Compatibility
+
+This release tracks the current TypeScript source at `5.27.1`. The
+REST, WebSocket, and SSE contracts are covered by credential-free protocol
+tests, including authentication refresh, uploads, project state recovery,
+streaming chat, workflows, templates, replay, and the canonical 25 hosted-tool
+schemas.
+
+Current model and transport coverage includes LTX 2.5, MiniMax H3 in all four
+tiers (Standard, 8-step Balanced, 4-step LightX2V Turbo, and the separate
+FastH3 `fastvideo-int8` Turbo engine), Seedance 2.5, Wan 3 and Wan 3.0 Enhanced,
+RTX VSR, MiniMax Music 3, LoRA catalog discovery, queue start estimates,
+live-benchmarked render/total time on cost quotes, in-flight project recovery
+across reconnects, confirmed cancellation, and connection/workload attribution.
+
+The Python API is async-first; `AsyncSogniClient` is an alias of
+`SogniClient`, not a synchronous wrapper. Browser-only cookie coordination and
+multi-tab behavior have no Python equivalent. Local image references are
+uploaded with their detected MIME type, but the TypeScript client's optional
+browser-side image resizing is not reproduced. All 25 canonical tool schemas
+are exposed; the local project-backed executor handles the six direct media
+generation tools, while the remaining tools run through the hosted or durable
+chat APIs. Live, credentialed smoke tests are intentionally separate from the
+default test suite.
+
+## Token authentication
+
+```python
+sogni = await SogniClient.create(auth_type="token")
+await sogni.set_tokens(token=access_token, refresh_token=refresh_token)
+```
+
+Username/password login and signing are available through `sogni.account.login`.
+API-key use does not require storing a wallet password.
+
+## Development
+
+```bash
+python -m pip install -e '.[dev]'
+pytest
+ruff check sogni_client tests
+ruff format --check sogni_client tests
+python -m build
+```
+
+Live integration tests require explicit credentials and are not run by default.
+
+## Documentation
+
+- [Python SDK quickstart](https://docs.sogni.ai/sogni-sdk/python/)
+- [Sogni SDK overview](https://docs.sogni.ai/sogni-sdk/)
+- [REST API reference](https://docs.sogni.ai/api-reference/)

sogni_client-5.27.1/examples/generate_image.py

@@ -0,0 +1,27 @@
+"""Generate one image with an API key from the environment."""
+
+import asyncio
+import os
+
+from sogni_client import SogniClient
+
+
+async def main() -> None:
+    api_key = os.environ["SOGNI_API_KEY"]
+    async with await SogniClient.create(api_key=api_key) as sogni:
+        project = await sogni.projects.create(
+            type="image",
+            model_id="z_image_turbo_bf16",
+            positive_prompt="A glass greenhouse drifting above Singapore at dawn",
+            negative_prompt="text, watermark",
+            number_of_media=1,
+            width=1024,
+            height=1024,
+            steps=8,
+        )
+        for url in await project.wait_for_completion():
+            print(url)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

sogni_client-5.27.1/examples/krea_identity_edit.py

@@ -0,0 +1,59 @@
+"""Edit one or two local reference images with Krea 2 Identity Edit v1.2."""
+
+from __future__ import annotations
+
+import argparse
+import asyncio
+import os
+from pathlib import Path
+
+from sogni_client import SogniClient
+
+
+def parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument(
+        "reference_images",
+        nargs="+",
+        type=Path,
+        help="One or two local reference images; place the base scene first.",
+    )
+    parser.add_argument("--prompt", required=True, help="The edit instruction.")
+    parser.add_argument("--count", type=int, default=1, help="Number of outputs (default: 1).")
+    parser.add_argument("--width", type=int, default=1024)
+    parser.add_argument("--height", type=int, default=1024)
+    parser.add_argument("--steps", type=int, default=10)
+    args = parser.parse_args()
+    if not 1 <= len(args.reference_images) <= 2:
+        parser.error("Krea 2 Identity Edit requires one or two reference images")
+    if args.count < 1:
+        parser.error("--count must be at least 1")
+    for reference in args.reference_images:
+        if not reference.is_file():
+            parser.error(f"reference image does not exist: {reference}")
+    return args
+
+
+async def main(args: argparse.Namespace) -> None:
+    async with await SogniClient.create(api_key=os.environ["SOGNI_API_KEY"]) as sogni:
+        project = await sogni.projects.create(
+            type="image",
+            model_id="krea2_identity_edit_v1_2",
+            positive_prompt=args.prompt,
+            number_of_media=args.count,
+            width=args.width,
+            height=args.height,
+            steps=args.steps,
+            guidance=1,
+            token_type="spark",
+            context_images=args.reference_images,
+        )
+        # No client-side timeout: a timeout here would abandon a render that is
+        # still running on the Supernet. The project carries its own runtime
+        # budget, and the SDK resumes it across reconnects.
+        for url in await project.wait_for_completion():
+            print(url)
+
+
+if __name__ == "__main__":
+    asyncio.run(main(parse_args()))

sogni_client-5.27.1/examples/stream_chat.py

@@ -0,0 +1,23 @@
+"""Stream an LLM response over the Sogni socket transport."""
+
+import asyncio
+import os
+
+from sogni_client import SogniClient
+
+
+async def main() -> None:
+    api_key = os.environ["SOGNI_API_KEY"]
+    async with await SogniClient.create(api_key=api_key) as sogni:
+        stream = await sogni.chat.completions.create(
+            model="qwen3.6-35b-a3b-gguf-iq4xs",
+            messages=[{"role": "user", "content": "Pitch three surreal album covers."}],
+            stream=True,
+        )
+        async for chunk in stream:
+            print(chunk.get("content", ""), end="", flush=True)
+        print()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())