batchwork-ai 0.1.1__tar.gz

batchwork_ai-0.1.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Ajanraj
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

batchwork_ai-0.1.1/PKG-INFO

@@ -0,0 +1,143 @@
+Metadata-Version: 2.4
+Name: batchwork-ai
+Version: 0.1.1
+Summary: Unified async batch API for AI providers
+Keywords: ai,anthropic,batch,gemini,llm,openai
+Author: Ajanraj
+Author-email: Ajanraj <hey@ajanraj.com>
+License-Expression: MIT
+License-File: LICENSE
+Classifier: Development Status :: 3 - Alpha
+Classifier: Framework :: AsyncIO
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Typing :: Typed
+Requires-Dist: httpcore>=1,<2
+Requires-Dist: httpx>=0.28,<1
+Requires-Dist: pydantic>=2.11,<3
+Requires-Dist: upstash-redis>=1.4,<2 ; extra == 'redis'
+Maintainer: Ajanraj
+Maintainer-email: Ajanraj <hey@ajanraj.com>
+Requires-Python: >=3.11
+Project-URL: Documentation, https://batchwork.ajanraj.com
+Provides-Extra: redis
+Description-Content-Type: text/markdown
+
+# Batchwork
+
+Unified async batch API for OpenAI, Anthropic, Google Gemini, Groq, Mistral, Together AI, and xAI. Access provider-native batch pricing—often up to 50% lower than standard synchronous requests—through one typed interface while Batchwork handles provider-specific serialization, submission, polling, and result normalization. Pricing and eligibility remain provider- and model-specific. Built for Python by Ajanraj.
+
+- One typed API for provider-native batch jobs
+- Text, embedding, and image workloads
+- Normalized jobs, results, usage, and errors
+- Messages, tools, structured content, and remote media
+- Polling, persistent stores, and signed webhooks
+- No provider SDK or JavaScript runtime dependencies
+
+📖 **Full documentation: [batchwork.ajanraj.com](https://batchwork.ajanraj.com)**
+
+## Installation
+
+Batchwork requires Python 3.11 or newer.
+
+With uv:
+
+```bash
+uv add batchwork-ai
+```
+
+With pip:
+
+```bash
+pip install batchwork-ai
+```
+
+Then configure a provider credential:
+
+```bash
+export OPENAI_API_KEY="..."
+```
+
+See [Configuration](https://batchwork.ajanraj.com/docs/configuration) for every provider credential, endpoint override, model format, and batch limit.
+
+## Quickstart
+
+```python
+import asyncio
+
+from batchwork import BatchRequest, Batchwork
+
+
+async def main() -> None:
+    async with Batchwork() as client:
+        job = await client.batch(
+            model="openai/gpt-5.6-sol",
+            requests=[BatchRequest(custom_id="hello", prompt="Say hello")],
+        )
+        await job.wait(timeout=3600)
+
+        for result in await job.collect():
+            print(result.custom_id, result.text)
+
+
+asyncio.run(main())
+```
+
+Submitting returns a `BatchJob` immediately. Provider processing is asynchronous and may take minutes or, depending on the provider and workload, up to 24 hours.
+
+Models use `provider/model` form. Results are correlated with requests through `custom_id`; provider output order is not guaranteed.
+
+## Providers
+
+Output workloads supported by Batchwork:
+
+| Provider      | Text | Embeddings | Image generation |
+| ------------- | ---- | ---------- | ---------------- |
+| OpenAI        | Yes  | Yes        | Yes              |
+| Anthropic     | Yes  | No         | No               |
+| Google Gemini | Yes  | Yes        | Yes              |
+| Groq          | Yes  | No         | No               |
+| Mistral       | Yes  | Yes        | No               |
+| Together AI   | Yes  | No         | No               |
+| xAI           | Yes  | No         | Yes              |
+
+Image, PDF, text-file, and audio inputs for text requests vary separately from output modalities. Provider APIs may also impose model-specific limits. See the [provider overview](https://batchwork.ajanraj.com/docs/providers) for input support, submission transport, credentials, and restrictions.
+
+## Optional Redis store
+
+Install the Upstash Redis integration for persistent polling state:
+
+```bash
+uv add "batchwork-ai[redis]"
+# or
+pip install "batchwork-ai[redis]"
+```
+
+The base package does not import or require `upstash-redis`.
+
+## Documentation
+
+- [Installation](https://batchwork.ajanraj.com/docs/installation)
+- [Configuration](https://batchwork.ajanraj.com/docs/configuration)
+- [Jobs](https://batchwork.ajanraj.com/docs/guides/jobs)
+- [Results](https://batchwork.ajanraj.com/docs/guides/results)
+- [Text, embeddings, and images](https://batchwork.ajanraj.com/docs/modalities/text)
+- [Provider overview](https://batchwork.ajanraj.com/docs/providers)
+- [Polling and webhooks](https://batchwork.ajanraj.com/docs/guides/server)
+- [Stores](https://batchwork.ajanraj.com/docs/guides/stores)
+- [Security](https://batchwork.ajanraj.com/docs/guides/security)
+- [Examples](https://batchwork.ajanraj.com/docs/examples)
+- [Public API](https://batchwork.ajanraj.com/docs/api)
+- [FAQ](https://batchwork.ajanraj.com/docs/faq)
+
+## License
+
+[MIT](https://opensource.org/licenses/MIT) © [Ajanraj](https://github.com/ajanraj)
+
+## Acknowledgements
+
+Inspired by [Hayden Bleasel's Batchwork](https://github.com/haydenbleasel/batchwork).

batchwork_ai-0.1.1/README.md

@@ -0,0 +1,114 @@
+# Batchwork
+
+Unified async batch API for OpenAI, Anthropic, Google Gemini, Groq, Mistral, Together AI, and xAI. Access provider-native batch pricing—often up to 50% lower than standard synchronous requests—through one typed interface while Batchwork handles provider-specific serialization, submission, polling, and result normalization. Pricing and eligibility remain provider- and model-specific. Built for Python by Ajanraj.
+
+- One typed API for provider-native batch jobs
+- Text, embedding, and image workloads
+- Normalized jobs, results, usage, and errors
+- Messages, tools, structured content, and remote media
+- Polling, persistent stores, and signed webhooks
+- No provider SDK or JavaScript runtime dependencies
+
+📖 **Full documentation: [batchwork.ajanraj.com](https://batchwork.ajanraj.com)**
+
+## Installation
+
+Batchwork requires Python 3.11 or newer.
+
+With uv:
+
+```bash
+uv add batchwork-ai
+```
+
+With pip:
+
+```bash
+pip install batchwork-ai
+```
+
+Then configure a provider credential:
+
+```bash
+export OPENAI_API_KEY="..."
+```
+
+See [Configuration](https://batchwork.ajanraj.com/docs/configuration) for every provider credential, endpoint override, model format, and batch limit.
+
+## Quickstart
+
+```python
+import asyncio
+
+from batchwork import BatchRequest, Batchwork
+
+
+async def main() -> None:
+    async with Batchwork() as client:
+        job = await client.batch(
+            model="openai/gpt-5.6-sol",
+            requests=[BatchRequest(custom_id="hello", prompt="Say hello")],
+        )
+        await job.wait(timeout=3600)
+
+        for result in await job.collect():
+            print(result.custom_id, result.text)
+
+
+asyncio.run(main())
+```
+
+Submitting returns a `BatchJob` immediately. Provider processing is asynchronous and may take minutes or, depending on the provider and workload, up to 24 hours.
+
+Models use `provider/model` form. Results are correlated with requests through `custom_id`; provider output order is not guaranteed.
+
+## Providers
+
+Output workloads supported by Batchwork:
+
+| Provider      | Text | Embeddings | Image generation |
+| ------------- | ---- | ---------- | ---------------- |
+| OpenAI        | Yes  | Yes        | Yes              |
+| Anthropic     | Yes  | No         | No               |
+| Google Gemini | Yes  | Yes        | Yes              |
+| Groq          | Yes  | No         | No               |
+| Mistral       | Yes  | Yes        | No               |
+| Together AI   | Yes  | No         | No               |
+| xAI           | Yes  | No         | Yes              |
+
+Image, PDF, text-file, and audio inputs for text requests vary separately from output modalities. Provider APIs may also impose model-specific limits. See the [provider overview](https://batchwork.ajanraj.com/docs/providers) for input support, submission transport, credentials, and restrictions.
+
+## Optional Redis store
+
+Install the Upstash Redis integration for persistent polling state:
+
+```bash
+uv add "batchwork-ai[redis]"
+# or
+pip install "batchwork-ai[redis]"
+```
+
+The base package does not import or require `upstash-redis`.
+
+## Documentation
+
+- [Installation](https://batchwork.ajanraj.com/docs/installation)
+- [Configuration](https://batchwork.ajanraj.com/docs/configuration)
+- [Jobs](https://batchwork.ajanraj.com/docs/guides/jobs)
+- [Results](https://batchwork.ajanraj.com/docs/guides/results)
+- [Text, embeddings, and images](https://batchwork.ajanraj.com/docs/modalities/text)
+- [Provider overview](https://batchwork.ajanraj.com/docs/providers)
+- [Polling and webhooks](https://batchwork.ajanraj.com/docs/guides/server)
+- [Stores](https://batchwork.ajanraj.com/docs/guides/stores)
+- [Security](https://batchwork.ajanraj.com/docs/guides/security)
+- [Examples](https://batchwork.ajanraj.com/docs/examples)
+- [Public API](https://batchwork.ajanraj.com/docs/api)
+- [FAQ](https://batchwork.ajanraj.com/docs/faq)
+
+## License
+
+[MIT](https://opensource.org/licenses/MIT) © [Ajanraj](https://github.com/ajanraj)
+
+## Acknowledgements
+
+Inspired by [Hayden Bleasel's Batchwork](https://github.com/haydenbleasel/batchwork).

batchwork_ai-0.1.1/pyproject.toml

@@ -0,0 +1,70 @@
+[build-system]
+requires = ["uv_build>=0.11.29,<0.12"]
+build-backend = "uv_build"
+
+[project]
+name = "batchwork-ai"
+version = "0.1.1"
+description = "Unified async batch API for AI providers"
+readme = "README.md"
+requires-python = ">=3.11"
+license = "MIT"
+license-files = ["LICENSE"]
+authors = [{ name = "Ajanraj", email = "hey@ajanraj.com" }]
+maintainers = [{ name = "Ajanraj", email = "hey@ajanraj.com" }]
+keywords = ["ai", "anthropic", "batch", "gemini", "llm", "openai"]
+classifiers = [
+  "Development Status :: 3 - Alpha",
+  "Framework :: AsyncIO",
+  "Intended Audience :: Developers",
+  "Programming Language :: Python :: 3",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14",
+  "Typing :: Typed",
+]
+dependencies = [
+  "httpcore>=1,<2",
+  "httpx>=0.28,<1",
+  "pydantic>=2.11,<3",
+]
+
+[project.optional-dependencies]
+redis = ["upstash-redis>=1.4,<2"]
+
+[project.urls]
+Documentation = "https://batchwork.ajanraj.com"
+
+[dependency-groups]
+dev = [
+  "pytest>=8.4,<9",
+  "pytest-asyncio>=1,<2",
+  "pytest-cov>=6.2,<7",
+  "ruff>=0.12,<1",
+  "twine>=6.1,<7",
+  "ty==0.0.59",
+]
+
+[tool.uv.build-backend]
+module-name = "batchwork"
+module-root = "src"
+
+[tool.pytest.ini_options]
+addopts = "-q"
+asyncio_mode = "auto"
+testpaths = ["tests"]
+markers = [
+  "live: requires real provider credentials",
+  "redis: requires Upstash Redis credentials",
+]
+
+[tool.ruff]
+line-length = 100
+target-version = "py311"
+
+[tool.ruff.lint]
+select = ["E", "F", "I", "UP", "B", "ASYNC", "RUF"]
+
+[tool.ty.src]
+include = ["src"]

batchwork_ai-0.1.1/src/batchwork/__init__.py

@@ -0,0 +1,261 @@
+"""Unified async batch API for AI providers."""
+
+from .client import Batchwork
+from .errors import (
+    BatchClosedError,
+    BatchStateError,
+    BatchTimeoutError,
+    BatchworkError,
+    MediaResolutionError,
+    MissingDependencyError,
+    UnsupportedProviderError,
+)
+from .job import BatchJob, PollCallback
+from .media import DefaultMediaResolver, MediaResolver, ResolvedMedia
+from .server import (
+    AtomicWebhookReplayStore,
+    BatchPoller,
+    BatchStore,
+    BatchWebhookEvent,
+    CompletionSink,
+    CredentialResolver,
+    ErrorHandler,
+    MemoryBatchStore,
+    MemoryWebhookReplayStore,
+    PinnedWebhookTransport,
+    RedisBatchStore,
+    TickFailure,
+    TickResult,
+    TrackedBatch,
+    TrackTarget,
+    UpstashRedis,
+    VerifiedWebhook,
+    WebhookEventType,
+    WebhookReplayStore,
+    WebhookResponse,
+    WebhookTransport,
+    WebhookUrlValidator,
+    create_batch_poller,
+    create_memory_store,
+    create_redis_store,
+    parse_webhook_url,
+    resolve_public_addresses,
+    sign_webhook,
+    validate_webhook_url,
+    verify_batch_webhook,
+    verify_webhook,
+)
+from .types import (
+    TERMINAL_STATUSES,
+    AssistantContentPart,
+    AssistantMessage,
+    BatchDefaults,
+    BatchEmbeddingRequest,
+    BatchImage,
+    BatchImageDefaults,
+    BatchImageRequest,
+    BatchLimits,
+    BatchProvider,
+    BatchRef,
+    BatchRequest,
+    BatchRequestCounts,
+    BatchRequestSettings,
+    BatchResult,
+    BatchResultError,
+    BatchResultStatus,
+    BatchSnapshot,
+    BatchStatus,
+    BatchUsage,
+    ContentToolOutput,
+    CustomPart,
+    DeprecatedToolOutputFileDataPart,
+    DeprecatedToolOutputFileIdPart,
+    DeprecatedToolOutputFileReferencePart,
+    DeprecatedToolOutputFileUrlPart,
+    DeprecatedToolOutputImageDataPart,
+    DeprecatedToolOutputImageFileIdPart,
+    DeprecatedToolOutputImageFileReferencePart,
+    DeprecatedToolOutputImageUrlPart,
+    ErrorJsonToolOutput,
+    ErrorTextToolOutput,
+    ExecutionDeniedToolOutput,
+    FilePart,
+    FunctionTool,
+    ImagePart,
+    JsonScalar,
+    JsonToolOutput,
+    JsonValue,
+    MediaSource,
+    ModelKind,
+    ModelMessage,
+    ModelSpec,
+    NamedToolChoice,
+    ProviderCredentials,
+    ProviderDefinedTool,
+    ProviderFileReference,
+    ProviderOptions,
+    ProviderToolChoice,
+    ReasoningFilePart,
+    ReasoningMediaSource,
+    ReasoningPart,
+    SystemMessage,
+    TaggedFileData,
+    TaggedFileDataData,
+    TaggedFileDataReference,
+    TaggedFileDataText,
+    TaggedFileDataUrl,
+    TaggedReasoningFileData,
+    TextPart,
+    TextToolOutput,
+    Tool,
+    ToolApprovalRequestPart,
+    ToolApprovalResponsePart,
+    ToolCallPart,
+    ToolChoice,
+    ToolContentPart,
+    ToolMessage,
+    ToolOutput,
+    ToolOutputContentPart,
+    ToolOutputCustomPart,
+    ToolOutputFilePart,
+    ToolOutputTextPart,
+    ToolResultPart,
+    UserContentPart,
+    UserMessage,
+    coerce_credentials,
+    is_terminal_status,
+    provider_from_ref,
+    resolve_model,
+    utc_datetime,
+)
+
+__version__ = "0.1.1"
+
+__all__ = [
+    "TERMINAL_STATUSES",
+    "AssistantContentPart",
+    "AssistantMessage",
+    "AtomicWebhookReplayStore",
+    "BatchClosedError",
+    "BatchDefaults",
+    "BatchEmbeddingRequest",
+    "BatchImage",
+    "BatchImageDefaults",
+    "BatchImageRequest",
+    "BatchJob",
+    "BatchLimits",
+    "BatchPoller",
+    "BatchProvider",
+    "BatchRef",
+    "BatchRequest",
+    "BatchRequestCounts",
+    "BatchRequestSettings",
+    "BatchResult",
+    "BatchResultError",
+    "BatchResultStatus",
+    "BatchSnapshot",
+    "BatchStateError",
+    "BatchStatus",
+    "BatchStore",
+    "BatchTimeoutError",
+    "BatchUsage",
+    "BatchWebhookEvent",
+    "Batchwork",
+    "BatchworkError",
+    "CompletionSink",
+    "ContentToolOutput",
+    "CredentialResolver",
+    "CustomPart",
+    "DefaultMediaResolver",
+    "DeprecatedToolOutputFileDataPart",
+    "DeprecatedToolOutputFileIdPart",
+    "DeprecatedToolOutputFileReferencePart",
+    "DeprecatedToolOutputFileUrlPart",
+    "DeprecatedToolOutputImageDataPart",
+    "DeprecatedToolOutputImageFileIdPart",
+    "DeprecatedToolOutputImageFileReferencePart",
+    "DeprecatedToolOutputImageUrlPart",
+    "ErrorHandler",
+    "ErrorJsonToolOutput",
+    "ErrorTextToolOutput",
+    "ExecutionDeniedToolOutput",
+    "FilePart",
+    "FunctionTool",
+    "ImagePart",
+    "JsonScalar",
+    "JsonToolOutput",
+    "JsonValue",
+    "MediaResolutionError",
+    "MediaResolver",
+    "MediaSource",
+    "MemoryBatchStore",
+    "MemoryWebhookReplayStore",
+    "MissingDependencyError",
+    "ModelKind",
+    "ModelMessage",
+    "ModelSpec",
+    "NamedToolChoice",
+    "PinnedWebhookTransport",
+    "PollCallback",
+    "ProviderCredentials",
+    "ProviderDefinedTool",
+    "ProviderFileReference",
+    "ProviderOptions",
+    "ProviderToolChoice",
+    "ReasoningFilePart",
+    "ReasoningMediaSource",
+    "ReasoningPart",
+    "RedisBatchStore",
+    "ResolvedMedia",
+    "SystemMessage",
+    "TaggedFileData",
+    "TaggedFileDataData",
+    "TaggedFileDataReference",
+    "TaggedFileDataText",
+    "TaggedFileDataUrl",
+    "TaggedReasoningFileData",
+    "TextPart",
+    "TextToolOutput",
+    "TickFailure",
+    "TickResult",
+    "Tool",
+    "ToolApprovalRequestPart",
+    "ToolApprovalResponsePart",
+    "ToolCallPart",
+    "ToolChoice",
+    "ToolContentPart",
+    "ToolMessage",
+    "ToolOutput",
+    "ToolOutputContentPart",
+    "ToolOutputCustomPart",
+    "ToolOutputFilePart",
+    "ToolOutputTextPart",
+    "ToolResultPart",
+    "TrackTarget",
+    "TrackedBatch",
+    "UnsupportedProviderError",
+    "UpstashRedis",
+    "UserContentPart",
+    "UserMessage",
+    "VerifiedWebhook",
+    "WebhookEventType",
+    "WebhookReplayStore",
+    "WebhookResponse",
+    "WebhookTransport",
+    "WebhookUrlValidator",
+    "__version__",
+    "coerce_credentials",
+    "create_batch_poller",
+    "create_memory_store",
+    "create_redis_store",
+    "is_terminal_status",
+    "parse_webhook_url",
+    "provider_from_ref",
+    "resolve_model",
+    "resolve_public_addresses",
+    "sign_webhook",
+    "utc_datetime",
+    "validate_webhook_url",
+    "verify_batch_webhook",
+    "verify_webhook",
+]