codex-ai 0.2.4__tar.gz → 0.2.5__tar.gz

{codex_ai-0.2.4 → codex_ai-0.2.5}/CHANGELOG.md

@@ -4,6 +4,24 @@ All notable changes to this project will be documented in this file.
 
 ## [Unreleased]
 
+## [0.2.5] - 2026-07-19
+
+### Added
+- Added `ImageInput` and optional `input_images` support to Gemini `generate_image_bytes()` for image reference/edit workflows.
+- Added OpenAI structured output generation through `generate_json()` with native Pydantic parsing.
+- Added typed OpenAI Responses API text streaming through `stream_text()`.
+
+### Changed
+- Replaced the OpenAI Chat Completions implementation with the Responses API.
+- Updated the OpenAI SDK requirement to `openai>=2.0,<3.0`.
+- Changed the OpenAI default model to `gpt-5.6-luna` with `reasoning={"effort": "none"}` and `store=False` defaults.
+
+## [0.2.4] - 2026-05-21
+
+### Changed
+- Widened the supported `codex-core` range to include the `0.4.x` series.
+- Clarified the Gemini-first provider focus in the package documentation.
+
 ## [0.2.3] - 2026-05-17
 
 ### Added

{codex_ai-0.2.4 → codex_ai-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codex-ai
-Version: 0.2.4
+Version: 0.2.5
 Summary: Gemini-first and OpenAI provider helpers for Codex
 Project-URL: Homepage, https://github.com/codexdlc/codex-ai
 Project-URL: Documentation, https://codexdlc.github.io/codex-ai/
@@ -20,13 +20,13 @@ Requires-Dist: codex-core<0.5.0,>=0.2.2
 Requires-Dist: pydantic<3.0,>=2.0
 Provides-Extra: all
 Requires-Dist: google-genai==2.3.0; extra == 'all'
-Requires-Dist: openai<2.0,>=1.0; extra == 'all'
+Requires-Dist: openai<3.0,>=2.0; extra == 'all'
 Provides-Extra: dev
 Requires-Dist: bandit>=1.7; extra == 'dev'
 Requires-Dist: detect-secrets>=1.5; extra == 'dev'
 Requires-Dist: google-genai==2.3.0; extra == 'dev'
 Requires-Dist: mypy>=1.10; extra == 'dev'
-Requires-Dist: openai<2.0,>=1.0; extra == 'dev'
+Requires-Dist: openai<3.0,>=2.0; extra == 'dev'
 Requires-Dist: pip-audit>=2.7; extra == 'dev'
 Requires-Dist: pre-commit>=3.0; extra == 'dev'
 Requires-Dist: pytest-asyncio>=0.23; extra == 'dev'
@@ -42,7 +42,7 @@ Requires-Dist: mkdocstrings[python]>=0.24; extra == 'docs'
 Provides-Extra: gemini
 Requires-Dist: google-genai==2.3.0; extra == 'gemini'
 Provides-Extra: openai
-Requires-Dist: openai<2.0,>=1.0; extra == 'openai'
+Requires-Dist: openai<3.0,>=2.0; extra == 'openai'
 Description-Content-Type: text/markdown
 
 # codex-ai <!-- Type: LANDING -->
@@ -52,7 +52,7 @@ Description-Content-Type: text/markdown
 [![CI](https://github.com/codexdlc/codex-ai/actions/workflows/ci.yml/badge.svg?branch=develop)](https://github.com/codexdlc/codex-ai/actions/workflows/ci.yml)
 [![License](https://img.shields.io/badge/license-Apache%202.0-blue.svg)](https://github.com/codexdlc/codex-ai/blob/main/LICENSE)
 
-Gemini-first API helpers for the Codex ecosystem. The active surface is direct Gemini text, JSON, Gemini image, and Imagen generation. OpenAI remains as a small text-only adapter, and the router/dispatcher layer is kept for legacy text workflows.
+Gemini-first API helpers for the Codex ecosystem. The active surface is direct Gemini text, JSON, Gemini image, and Imagen generation. OpenAI provides modern Responses API text, structured JSON, and streaming helpers. The router/dispatcher layer is kept for legacy text workflows.
 
 ## Install
 
@@ -68,9 +68,11 @@ Requires Python 3.12 or newer.
 ## Gemini Direct API
 
 ```python
+from pathlib import Path
+
 from pydantic import BaseModel
 
-from codex_ai import GeminiProvider
+from codex_ai import GeminiProvider, ImageInput
 
 
 class LootItem(BaseModel):
@@ -89,6 +91,13 @@ image_bytes, content_type = await gemini.generate_image_bytes(
     image_config={"aspect_ratio": "1:1", "image_size": "4K"},
 )
 
+source_image_bytes = Path("source.png").read_bytes()
+edited_bytes, edited_content_type = await gemini.generate_image_bytes(
+    "Keep the composition, but redraw it as a watercolor map.",
+    response_mime_type="image/png",
+    input_images=[ImageInput(data=source_image_bytes, mime_type="image/png")],
+)
+
 imagen_bytes, imagen_content_type = await gemini.generate_imagen_bytes(
     "A fantasy clan banner, game icon style.",
     response_mime_type="image/jpeg",
@@ -101,10 +110,39 @@ imagen_bytes, imagen_content_type = await gemini.generate_imagen_bytes(
 `response_mime_type` as a preferred/fallback MIME type. It does not pass image MIME values
 to Gemini's text `response_mime_type` config field. Pass Gemini image controls such as
 `aspect_ratio` and `image_size` with `image_config`; if a `4K` request is rejected,
-the Gemini provider retries once with `2K`. Use `generate_imagen_bytes()` for
+the Gemini provider retries once with `2K`. Pass `input_images` with `ImageInput`
+items to provide reference/edit-source images for Gemini image models. Use `generate_imagen_bytes()` for
 Imagen models; that path uses `generate_images` and passes the requested MIME as
 `output_mime_type`.
 
+## OpenAI Responses API
+
+```python
+from pydantic import BaseModel
+
+from codex_ai import OpenAIProvider
+
+
+class LootItem(BaseModel):
+    name: str
+    power: int
+
+
+openai = OpenAIProvider(api_key="sk-...")
+
+text = await openai.generate_text("Write one short tavern rumor.")
+loot = await openai.generate_json("Create one loot item.", schema=LootItem)
+
+async for chunk in openai.stream_text("Tell a short story."):
+    print(chunk, end="")
+```
+
+The OpenAI provider uses the Responses API and the `openai` 2.x SDK. Responses
+are not stored by default. Its default `gpt-5.6-luna` model uses
+`reasoning={"effort": "none"}` to retain a cost- and latency-sensitive role;
+override the model and reasoning options per request when a workload needs more
+capability.
+
 ## Legacy Text Router
 
 ```python
@@ -136,7 +174,7 @@ New Gemini integrations should call `generate_text()`, `generate_json()`,
 | Module | Extra | Description |
 | :--- | :--- | :--- |
 | `codex_ai.providers.gemini` | `[gemini]` | Primary API: Gemini text, JSON, Gemini image, and Imagen generation via pinned `google-genai` |
-| `codex_ai.providers.openai` | `[openai]` | Text-only OpenAI Chat Completions adapter |
+| `codex_ai.providers.openai` | `[openai]` | OpenAI Responses API text, structured JSON, and streaming adapter |
 | `codex_ai.core` | - | Legacy text router/dispatcher contracts and shared provider exceptions |
 
 ## Development

{codex_ai-0.2.4 → codex_ai-0.2.5}/README.md

@@ -5,7 +5,7 @@
 [![CI](https://github.com/codexdlc/codex-ai/actions/workflows/ci.yml/badge.svg?branch=develop)](https://github.com/codexdlc/codex-ai/actions/workflows/ci.yml)
 [![License](https://img.shields.io/badge/license-Apache%202.0-blue.svg)](https://github.com/codexdlc/codex-ai/blob/main/LICENSE)
 
-Gemini-first API helpers for the Codex ecosystem. The active surface is direct Gemini text, JSON, Gemini image, and Imagen generation. OpenAI remains as a small text-only adapter, and the router/dispatcher layer is kept for legacy text workflows.
+Gemini-first API helpers for the Codex ecosystem. The active surface is direct Gemini text, JSON, Gemini image, and Imagen generation. OpenAI provides modern Responses API text, structured JSON, and streaming helpers. The router/dispatcher layer is kept for legacy text workflows.
 
 ## Install
 
@@ -21,9 +21,11 @@ Requires Python 3.12 or newer.
 ## Gemini Direct API
 
 ```python
+from pathlib import Path
+
 from pydantic import BaseModel
 
-from codex_ai import GeminiProvider
+from codex_ai import GeminiProvider, ImageInput
 
 
 class LootItem(BaseModel):
@@ -42,6 +44,13 @@ image_bytes, content_type = await gemini.generate_image_bytes(
     image_config={"aspect_ratio": "1:1", "image_size": "4K"},
 )
 
+source_image_bytes = Path("source.png").read_bytes()
+edited_bytes, edited_content_type = await gemini.generate_image_bytes(
+    "Keep the composition, but redraw it as a watercolor map.",
+    response_mime_type="image/png",
+    input_images=[ImageInput(data=source_image_bytes, mime_type="image/png")],
+)
+
 imagen_bytes, imagen_content_type = await gemini.generate_imagen_bytes(
     "A fantasy clan banner, game icon style.",
     response_mime_type="image/jpeg",
@@ -54,10 +63,39 @@ imagen_bytes, imagen_content_type = await gemini.generate_imagen_bytes(
 `response_mime_type` as a preferred/fallback MIME type. It does not pass image MIME values
 to Gemini's text `response_mime_type` config field. Pass Gemini image controls such as
 `aspect_ratio` and `image_size` with `image_config`; if a `4K` request is rejected,
-the Gemini provider retries once with `2K`. Use `generate_imagen_bytes()` for
+the Gemini provider retries once with `2K`. Pass `input_images` with `ImageInput`
+items to provide reference/edit-source images for Gemini image models. Use `generate_imagen_bytes()` for
 Imagen models; that path uses `generate_images` and passes the requested MIME as
 `output_mime_type`.
 
+## OpenAI Responses API
+
+```python
+from pydantic import BaseModel
+
+from codex_ai import OpenAIProvider
+
+
+class LootItem(BaseModel):
+    name: str
+    power: int
+
+
+openai = OpenAIProvider(api_key="sk-...")
+
+text = await openai.generate_text("Write one short tavern rumor.")
+loot = await openai.generate_json("Create one loot item.", schema=LootItem)
+
+async for chunk in openai.stream_text("Tell a short story."):
+    print(chunk, end="")
+```
+
+The OpenAI provider uses the Responses API and the `openai` 2.x SDK. Responses
+are not stored by default. Its default `gpt-5.6-luna` model uses
+`reasoning={"effort": "none"}` to retain a cost- and latency-sensitive role;
+override the model and reasoning options per request when a workload needs more
+capability.
+
 ## Legacy Text Router
 
 ```python
@@ -89,7 +127,7 @@ New Gemini integrations should call `generate_text()`, `generate_json()`,
 | Module | Extra | Description |
 | :--- | :--- | :--- |
 | `codex_ai.providers.gemini` | `[gemini]` | Primary API: Gemini text, JSON, Gemini image, and Imagen generation via pinned `google-genai` |
-| `codex_ai.providers.openai` | `[openai]` | Text-only OpenAI Chat Completions adapter |
+| `codex_ai.providers.openai` | `[openai]` | OpenAI Responses API text, structured JSON, and streaming adapter |
 | `codex_ai.core` | - | Legacy text router/dispatcher contracts and shared provider exceptions |
 
 ## Development

{codex_ai-0.2.4 → codex_ai-0.2.5}/docs/en/architecture/providers/README.md

@@ -2,7 +2,7 @@
 
 ## Purpose
 
-`codex_ai.providers` contains concrete SDK adapters, not a broad interchangeable provider framework. Gemini is the product focus; OpenAI is retained only for text generation.
+`codex_ai.providers` contains concrete SDK adapters, not a broad interchangeable provider framework. Gemini is the product focus; OpenAI is a separate modern Responses API adapter.
 
 Gemini is the primary target and exposes direct methods for text, JSON, and image generation:
 
@@ -13,7 +13,14 @@ await gemini.generate_image_bytes(...)
 await gemini.generate_imagen_bytes(...)
 ```
 
-OpenAI is kept as a text-only adapter with the same `generate_text(...)` convenience.
+OpenAI exposes direct text, structured JSON, and typed streaming helpers:
+
+```python
+await openai.generate_text(...)
+await openai.generate_json(..., schema=MyModel)
+async for chunk in openai.stream_text(...):
+    ...
+```
 
 ## Architecture
 
@@ -24,7 +31,9 @@ PromptResult/String
       ├── GeminiProvider.generate_json(...)        -> dict | BaseModel
       ├── GeminiProvider.generate_image_bytes(...) -> tuple[bytes, str]
       ├── GeminiProvider.generate_imagen_bytes(...) -> tuple[bytes, str]
-      └── OpenAIProvider.generate_text(...)        -> str
+      ├── OpenAIProvider.generate_text(...)        -> str
+      ├── OpenAIProvider.generate_json(...)        -> dict | BaseModel
+      └── OpenAIProvider.stream_text(...)          -> AsyncIterator[str]
 ```
 
 The legacy router pipeline remains available:
@@ -40,14 +49,19 @@ LLMRouter builder -> PromptResult -> LLMDispatcher.process() -> provider.answer(
 | Component | Class | SDK | Default Model |
 |-----------|-------|-----|---------------|
 | `gemini.py` | `GeminiProvider` | `google-genai` | `gemini-2.5-flash-lite` |
-| `openai.py` | `OpenAIProvider` | `openai` | `gpt-4o-mini` |
+| `openai.py` | `OpenAIProvider` | `openai` 2.x | `gpt-5.6-luna` |
 
 ## Key Design Decisions
 
 - Gemini-specific capabilities are represented directly instead of being hidden behind a broad universal abstraction.
 - JSON generation uses provider-native JSON configuration and still validates locally with `json.loads` and optional Pydantic models.
+- OpenAI uses the Responses API exclusively; the old Chat Completions path is not retained.
+- OpenAI requests default to `store=False` and `reasoning={"effort": "none"}`.
+- OpenAI Pydantic schemas use the SDK's native `responses.parse()` structured-output path.
+- The OpenAI SDK client is injectable behind a small Responses API port for deterministic tests and custom transports.
 - Gemini image generation and Imagen generation are separate explicit methods because they use different SDK calls.
 - `generate_image_bytes()` uses Gemini `generate_content` with image modality. Its `response_mime_type` is only a preferred/fallback MIME type, and Gemini image controls are passed through `image_config`.
+- `generate_image_bytes()` can also accept `input_images=[ImageInput(...)]` for image reference/edit workflows; without `input_images`, the prompt-only path is unchanged.
 - `generate_image_bytes()` retries a rejected `image_config={"image_size": "4K"}` request once as `2K`.
 - `generate_imagen_bytes()` uses Imagen `generate_images` and passes the requested MIME as `output_mime_type`.
 - Anthropic, OpenRouter, and multi-provider failover are not active APIs in this line.

{codex_ai-0.2.4 → codex_ai-0.2.5}/docs/en/architecture/providers/data_flow.md

@@ -33,6 +33,16 @@ prompt: str
     -> (bytes, actual_mime_type)
 ```
 
+With `input_images`, the provider sends multimodal content:
+
+```
+prompt: str + input_images: list[ImageInput]
+    -> text part + inline_data image parts
+    -> GeminiProvider.generate_image_bytes()
+    -> first returned inline_data image part
+    -> (bytes, actual_mime_type)
+```
+
 `response_mime_type` is not passed to `GenerateContentConfig.response_mime_type` on this path; it is only a fallback content type when Gemini omits `inline_data.mime_type`.
 When `image_config.image_size` is `4K` and Gemini rejects the request, the provider retries once with `image_size` changed to `2K`.
 
@@ -51,7 +61,30 @@ prompt: str
 ```
 PromptResult | str
     -> OpenAIProvider.generate_text()
-    -> chat.completions.create()
-    -> choices[0].message.content
+    -> _OpenAIRequestMapper
+    -> responses.create(store=False)
+    -> response.output_text
     -> str
 ```
+
+## OpenAI Structured JSON
+
+```
+PromptResult | str + Pydantic schema
+    -> OpenAIProvider.generate_json()
+    -> responses.parse(text_format=schema)
+    -> response.output_parsed
+    -> BaseModel
+```
+
+Without a schema, the provider requests JSON mode and decodes `response.output_text`.
+
+## OpenAI Streaming
+
+```
+PromptResult | str
+    -> OpenAIProvider.stream_text()
+    -> responses.create(stream=True)
+    -> response.output_text.delta events
+    -> AsyncIterator[str]
+```

{codex_ai-0.2.4 → codex_ai-0.2.5}/docs/ru/architecture/providers/README.md

@@ -2,7 +2,7 @@
 
 ## Назначение
 
-`codex_ai.providers` содержит конкретные SDK-адаптеры, а не широкий interchangeable provider framework. Фокус продукта — Gemini; OpenAI сохранен только для текстовой генерации.
+`codex_ai.providers` содержит конкретные SDK-адаптеры, а не широкий interchangeable provider framework. Фокус продукта — Gemini; OpenAI является отдельным современным адаптером Responses API.
 
 Gemini является основным направлением и дает прямые методы для текста, JSON и картинок:
 
@@ -13,7 +13,14 @@ await gemini.generate_image_bytes(...)
 await gemini.generate_imagen_bytes(...)
 ```
 
-OpenAI оставлен как text-only адаптер с `generate_text(...)`.
+OpenAI предоставляет прямые методы для текста, структурированного JSON и typed streaming:
+
+```python
+await openai.generate_text(...)
+await openai.generate_json(..., schema=MyModel)
+async for chunk in openai.stream_text(...):
+    ...
+```
 
 ## Архитектура
 
@@ -24,7 +31,9 @@ PromptResult/String
       ├── GeminiProvider.generate_json(...)        -> dict | BaseModel
       ├── GeminiProvider.generate_image_bytes(...) -> tuple[bytes, str]
       ├── GeminiProvider.generate_imagen_bytes(...) -> tuple[bytes, str]
-      └── OpenAIProvider.generate_text(...)        -> str
+      ├── OpenAIProvider.generate_text(...)        -> str
+      ├── OpenAIProvider.generate_json(...)        -> dict | BaseModel
+      └── OpenAIProvider.stream_text(...)          -> AsyncIterator[str]
 ```
 
 Старый router pipeline остается доступным:
@@ -40,14 +49,19 @@ LLMRouter builder -> PromptResult -> LLMDispatcher.process() -> provider.answer(
 | Компонент | Класс | SDK | Модель по умолчанию |
 |-----------|-------|-----|---------------------|
 | `gemini.py` | `GeminiProvider` | `google-genai` | `gemini-2.5-flash-lite` |
-| `openai.py` | `OpenAIProvider` | `openai` | `gpt-4o-mini` |
+| `openai.py` | `OpenAIProvider` | `openai` 2.x | `gpt-5.6-luna` |
 
 ## Ключевые решения
 
 - Возможности Gemini представлены напрямую, без широкой универсальной абстракции.
 - JSON generation использует native JSON config провайдера и локальную проверку через `json.loads` и optional Pydantic schema.
+- OpenAI использует только Responses API; старый путь Chat Completions не сохраняется.
+- OpenAI-запросы по умолчанию используют `store=False` и `reasoning={"effort": "none"}`.
+- Pydantic-схемы OpenAI обрабатываются через native structured-output путь `responses.parse()`.
+- OpenAI SDK client можно внедрить через небольшой Responses API port для детерминированных тестов и custom transport.
 - Gemini image generation и Imagen generation разведены в отдельные явные методы, потому что они используют разные SDK calls.
 - `generate_image_bytes()` использует Gemini `generate_content` с image modality. Его `response_mime_type` является только preferred/fallback MIME type, а Gemini image controls передаются через `image_config`.
+- `generate_image_bytes()` также принимает `input_images=[ImageInput(...)]` для сценариев reference/edit; без `input_images` старый prompt-only путь не меняется.
 - `generate_image_bytes()` один раз повторяет отклоненный `image_config={"image_size": "4K"}` запрос как `2K`.
 - `generate_imagen_bytes()` использует Imagen `generate_images` и передает requested MIME как `output_mime_type`.
 - Anthropic, OpenRouter и multi-provider failover не являются активными API в этой линейке.

{codex_ai-0.2.4 → codex_ai-0.2.5}/docs/ru/architecture/providers/data_flow.md

@@ -33,6 +33,16 @@ prompt: str
     -> (bytes, actual_mime_type)
 ```
 
+С `input_images` провайдер отправляет multimodal content:
+
+```
+prompt: str + input_images: list[ImageInput]
+    -> text part + inline_data image parts
+    -> GeminiProvider.generate_image_bytes()
+    -> first returned inline_data image part
+    -> (bytes, actual_mime_type)
+```
+
 `response_mime_type` в этом пути не передается в `GenerateContentConfig.response_mime_type`; он используется только как fallback content type, если Gemini не вернул `inline_data.mime_type`.
 Если `image_config.image_size` равен `4K` и Gemini отклоняет запрос, провайдер один раз повторяет его с `image_size="2K"`.
 
@@ -51,7 +61,30 @@ prompt: str
 ```
 PromptResult | str
     -> OpenAIProvider.generate_text()
-    -> chat.completions.create()
-    -> choices[0].message.content
+    -> _OpenAIRequestMapper
+    -> responses.create(store=False)
+    -> response.output_text
     -> str
 ```
+
+## OpenAI Structured JSON
+
+```
+PromptResult | str + Pydantic schema
+    -> OpenAIProvider.generate_json()
+    -> responses.parse(text_format=schema)
+    -> response.output_parsed
+    -> BaseModel
+```
+
+Без schema провайдер запрашивает JSON mode и декодирует `response.output_text`.
+
+## OpenAI Streaming
+
+```
+PromptResult | str
+    -> OpenAIProvider.stream_text()
+    -> responses.create(stream=True)
+    -> response.output_text.delta events
+    -> AsyncIterator[str]
+```

{codex_ai-0.2.4 → codex_ai-0.2.5}/pyproject.toml

@@ -30,7 +30,7 @@ Repository = "https://github.com/codexdlc/codex-ai"
 Issues = "https://github.com/codexdlc/codex-ai/issues"
 
 [project.optional-dependencies]
-openai = ["openai>=1.0,<2.0"]
+openai = ["openai>=2.0,<3.0"]
 gemini = ["google-genai==2.3.0"]
 all = [
     "codex-ai[openai,gemini]",

{codex_ai-0.2.4 → codex_ai-0.2.5}/src/codex_ai/__init__.py

@@ -2,6 +2,8 @@
 
 from codex_ai.core import (
     ImageGenerationProvider,
+    ImageInput,
+    ImageInputLike,
     ImagenGenerationProvider,
     JsonGenerationProvider,
     LLMDispatcher,
@@ -19,6 +21,8 @@ __all__ = [
     # Core
     "LLMDispatcher",
     "ImageGenerationProvider",
+    "ImageInput",
+    "ImageInputLike",
     "ImagenGenerationProvider",
     "JsonGenerationProvider",
     "LLMMessage",

{codex_ai-0.2.4 → codex_ai-0.2.5}/src/codex_ai/core/__init__.py

@@ -8,6 +8,8 @@ from .dispatcher import LLMDispatcher
 from .exceptions import LLMProviderError
 from .protocol import (
     ImageGenerationProvider,
+    ImageInput,
+    ImageInputLike,
     ImagenGenerationProvider,
     JsonGenerationProvider,
     LLMMessage,
@@ -23,6 +25,8 @@ __all__ = [
     "LLMDispatcher",
     "LLMProviderError",
     "ImageGenerationProvider",
+    "ImageInput",
+    "ImageInputLike",
     "ImagenGenerationProvider",
     "JsonGenerationProvider",
     "LLMMessage",

{codex_ai-0.2.4 → codex_ai-0.2.5}/src/codex_ai/core/dispatcher.py

@@ -14,6 +14,7 @@ from typing import Any
 
 from .protocol import (
     ImageGenerationProvider,
+    ImageInputLike,
     ImagenGenerationProvider,
     JsonGenerationProvider,
     LLMProviderProtocol,
@@ -98,6 +99,7 @@ class LLMDispatcher:
         model: str | None = None,
         response_mime_type: str = "image/webp",
         image_config: dict[str, Any] | None = None,
+        input_images: list[ImageInputLike] | None = None,
         **kwargs: Any,
     ) -> tuple[bytes, str]:
         """
@@ -117,6 +119,7 @@ class LLMDispatcher:
             model=model,
             response_mime_type=response_mime_type,
             image_config=image_config,
+            input_images=input_images,
             **kwargs,
         )
 

{codex_ai-0.2.4 → codex_ai-0.2.5}/src/codex_ai/core/protocol.py

@@ -14,7 +14,7 @@ PromptBuilder — type alias for async builder functions registered via LLMRoute
 
 from __future__ import annotations
 
-from collections.abc import Awaitable, Callable
+from collections.abc import Awaitable, Callable, Mapping
 from typing import TYPE_CHECKING, Any, Literal, Protocol, runtime_checkable
 
 from pydantic import BaseModel, ConfigDict
@@ -64,6 +64,24 @@ class PromptResult(BaseDTO):
     max_tokens: int | None = None
 
 
+class ImageInput(BaseDTO):
+    """
+    Binary image input used as a reference or edit source for image generation.
+
+    Attributes:
+        data: Raw image bytes.
+        mime_type: Image MIME type, for example ``"image/png"`` or ``"image/jpeg"``.
+    """
+
+    model_config = ConfigDict(frozen=True, arbitrary_types_allowed=True)
+
+    data: bytes
+    mime_type: str
+
+
+ImageInputLike = ImageInput | Mapping[str, Any]
+
+
 @runtime_checkable
 class LLMProviderProtocol(Protocol):
     """
@@ -150,6 +168,7 @@ class ImageGenerationProvider(Protocol):
         model: str | None = None,
         response_mime_type: str = "image/webp",
         image_config: dict[str, Any] | None = None,
+        input_images: list[ImageInputLike] | None = None,
         **kwargs: Any,
     ) -> tuple[bytes, str]:
         """
@@ -161,6 +180,8 @@ class ImageGenerationProvider(Protocol):
             response_mime_type: Requested/preferred image MIME type.
             image_config: Optional image generation controls such as
                 ``{"aspect_ratio": "1:1", "image_size": "4K"}``.
+            input_images: Optional reference/source images for image editing or
+                image-conditioned generation.
             **kwargs: Extra provider-specific kwargs.
 
         Returns:

{codex_ai-0.2.4 → codex_ai-0.2.5}/src/codex_ai/providers/__init__.py

@@ -1,7 +1,7 @@
 """
 codex_ai.providers
 ==================
-Provider adapters. Gemini is the primary API; OpenAI is text-only.
+Provider adapters. Gemini is the primary API; OpenAI uses the Responses API.
 
 Providers are lazy-loaded to avoid mandatory dependency on all SDK packages.
 """