dspy-codex-auth 0.1.0__tar.gz

dspy_codex_auth-0.1.0/PKG-INFO

@@ -0,0 +1,238 @@
+Metadata-Version: 2.4
+Name: dspy-codex-auth
+Version: 0.1.0
+Summary: DSPy integration for ChatGPT Codex subscription-backed language models
+Keywords: codex,dspy,language-models,openai
+Author: hrbatra
+Author-email: hrbatra <hrbatra@utexas.edu>
+License-Expression: MIT
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Dist: dspy>=3.2.0
+Requires-Dist: litellm>=1.82.6
+Requires-Dist: requests>=2.33.1
+Requires-Python: >=3.12
+Project-URL: Homepage, https://github.com/hrbatra/dspy-codex-auth
+Project-URL: Issues, https://github.com/hrbatra/dspy-codex-auth/issues
+Project-URL: Repository, https://github.com/hrbatra/dspy-codex-auth
+Description-Content-Type: text/markdown
+
+# dspy-codex-auth
+
+DSPy integration for using ChatGPT/Codex subscription credentials as a DSPy
+language model.
+
+This package is intentionally narrow:
+
+- It includes ChatGPT/Codex OAuth login, token refresh, and Pi-compatible
+  credential storage.
+- It installs a DSPy `LM` wrapper for `codex/...` model strings.
+- It fixes Codex Responses streaming shapes that DSPy 3.2 cannot parse from
+  the current Codex backend response stream.
+
+## Install
+
+```bash
+uv add dspy-codex-auth
+```
+
+Until the first PyPI release is installed in your environment, install from the
+GitHub repo:
+
+```bash
+uv add "dspy-codex-auth @ git+https://github.com/hrbatra/dspy-codex-auth.git@main"
+```
+
+## Login
+
+If you already have Codex credentials in `~/.pi/agent/auth.json`, no extra
+login is needed.
+
+Otherwise:
+
+```bash
+uv run python -c "import dspy_codex_auth; dspy_codex_auth.login()"
+```
+
+## Basic Usage
+
+```python
+import dspy
+import dspy_codex_auth
+
+dspy_codex_auth.install()
+
+lm = dspy.LM("codex/gpt-5.5", cache=False)
+dspy.configure(lm=lm, adapter=dspy.JSONAdapter())
+```
+
+`cache=False` is recommended for Codex while iterating because stale DSPy cache
+entries can preserve old empty-output responses across package upgrades.
+
+## Reasoning Summary
+
+Pass `reasoning_effort` as usual. This package also supports
+`reasoning_summary`, which maps to the Responses API `reasoning.summary` field.
+
+```python
+lm = dspy.LM(
+    "codex/gpt-5.5",
+    cache=False,
+    reasoning_effort="medium",
+    reasoning_summary="detailed",
+)
+```
+
+DSPy predictions expose declared output fields. The lower-level LM history can
+also include a returned reasoning summary:
+
+```python
+summary = lm.history[-1]["outputs"][0].get("reasoning_content")
+```
+
+## OpenAI-Style Model String With Codex Auth
+
+If you prefer to keep an `openai/...` model string and select Codex auth
+explicitly:
+
+```python
+lm = dspy_codex_auth.LM(
+    "openai/gpt-5.5",
+    auth_provider="codex",
+    cache=False,
+    reasoning_effort="medium",
+    reasoning_summary="detailed",
+)
+```
+
+## What It Fixes
+
+The ChatGPT Codex backend streams useful output events, but the completed
+LiteLLM Responses object can arrive with `response.output == []`. DSPy expects
+Responses output items to contain final message text, function calls, and
+reasoning summaries. This package reconstructs those output items from stream
+events before DSPy parses the response.
+
+It currently handles:
+
+- `response.output_item.done`
+- `response.output_text.done`
+- `response.output_text.delta`
+- `response.reasoning_summary_text.done`
+- `response.reasoning_summary_text.delta`
+- streamed function-call output items
+
+It also strips output-token cap fields that the Codex backend currently rejects:
+
+- `max_tokens`
+- `max_output_tokens`
+- `max_completion_tokens`
+
+## French Example
+
+```python
+import dspy
+import dspy_codex_auth
+
+dspy_codex_auth.install()
+
+lm = dspy.LM("codex/gpt-5.5", cache=False)
+dspy.configure(lm=lm, adapter=dspy.JSONAdapter())
+
+
+class TranslateFrenchToEnglish(dspy.Signature):
+    """Translate the French input into short, natural English."""
+
+    french: str = dspy.InputField(desc="French sentence")
+    english: str = dspy.OutputField(desc="Natural English translation")
+
+
+translator = dspy.Predict(TranslateFrenchToEnglish)
+print(translator(french="merci beaucoup").english)
+```
+
+## Math Example With Reasoning Summary
+
+```python
+import dspy
+import dspy_codex_auth
+
+dspy_codex_auth.install()
+
+lm = dspy.LM(
+    "codex/gpt-5.5",
+    cache=False,
+    reasoning_effort="medium",
+    reasoning_summary="detailed",
+)
+dspy.configure(lm=lm, adapter=dspy.JSONAdapter())
+
+
+class SolveMath(dspy.Signature):
+    """Solve the math problem. Return a concise numeric answer and a brief explanation."""
+
+    problem: str = dspy.InputField(desc="Math problem")
+    answer: str = dspy.OutputField(desc="Concise final answer")
+    explanation: str = dspy.OutputField(desc="Brief explanation")
+
+
+solver = dspy.Predict(SolveMath)
+pred = solver(
+    problem=(
+        "Compute the integral of the standard normal probability density "
+        "function from 0 to 1.5."
+    )
+)
+
+print(pred.answer)
+print(pred.explanation)
+print(lm.history[-1]["outputs"][0].get("reasoning_content"))
+```
+
+## Attribution
+
+`dspy-codex-auth` includes and adapts MIT-licensed auth and DSPy integration
+code from `dspy-lm-auth`:
+
+https://github.com/MaximeRivest/dspy-lm-auth
+
+The streamed-output reconstruction addresses a DSPy/Codex Responses streaming
+compatibility issue that was also discussed in `dspy-lm-auth` PR #2:
+
+https://github.com/MaximeRivest/dspy-lm-auth/pull/2
+
+`dspy-lm-auth` is MIT-licensed. The original copyright notice is preserved in
+`THIRD_PARTY_NOTICES.md`.
+
+## Development
+
+```bash
+uv sync --dev
+uv run pytest
+uv run ruff check .
+uv build --no-sources
+```
+
+## Release
+
+Releases are published from GitHub Actions with PyPI Trusted Publishing.
+
+For the first release, create a pending PyPI publisher for:
+
+- PyPI project: `dspy-codex-auth`
+- Owner: `hrbatra`
+- Repository: `dspy-codex-auth`
+- Workflow: `publish.yml`
+- Environment: `pypi`
+
+Then tag and push:
+
+```bash
+git tag v0.1.0
+git push origin v0.1.0
+```

dspy_codex_auth-0.1.0/README.md

@@ -0,0 +1,214 @@
+# dspy-codex-auth
+
+DSPy integration for using ChatGPT/Codex subscription credentials as a DSPy
+language model.
+
+This package is intentionally narrow:
+
+- It includes ChatGPT/Codex OAuth login, token refresh, and Pi-compatible
+  credential storage.
+- It installs a DSPy `LM` wrapper for `codex/...` model strings.
+- It fixes Codex Responses streaming shapes that DSPy 3.2 cannot parse from
+  the current Codex backend response stream.
+
+## Install
+
+```bash
+uv add dspy-codex-auth
+```
+
+Until the first PyPI release is installed in your environment, install from the
+GitHub repo:
+
+```bash
+uv add "dspy-codex-auth @ git+https://github.com/hrbatra/dspy-codex-auth.git@main"
+```
+
+## Login
+
+If you already have Codex credentials in `~/.pi/agent/auth.json`, no extra
+login is needed.
+
+Otherwise:
+
+```bash
+uv run python -c "import dspy_codex_auth; dspy_codex_auth.login()"
+```
+
+## Basic Usage
+
+```python
+import dspy
+import dspy_codex_auth
+
+dspy_codex_auth.install()
+
+lm = dspy.LM("codex/gpt-5.5", cache=False)
+dspy.configure(lm=lm, adapter=dspy.JSONAdapter())
+```
+
+`cache=False` is recommended for Codex while iterating because stale DSPy cache
+entries can preserve old empty-output responses across package upgrades.
+
+## Reasoning Summary
+
+Pass `reasoning_effort` as usual. This package also supports
+`reasoning_summary`, which maps to the Responses API `reasoning.summary` field.
+
+```python
+lm = dspy.LM(
+    "codex/gpt-5.5",
+    cache=False,
+    reasoning_effort="medium",
+    reasoning_summary="detailed",
+)
+```
+
+DSPy predictions expose declared output fields. The lower-level LM history can
+also include a returned reasoning summary:
+
+```python
+summary = lm.history[-1]["outputs"][0].get("reasoning_content")
+```
+
+## OpenAI-Style Model String With Codex Auth
+
+If you prefer to keep an `openai/...` model string and select Codex auth
+explicitly:
+
+```python
+lm = dspy_codex_auth.LM(
+    "openai/gpt-5.5",
+    auth_provider="codex",
+    cache=False,
+    reasoning_effort="medium",
+    reasoning_summary="detailed",
+)
+```
+
+## What It Fixes
+
+The ChatGPT Codex backend streams useful output events, but the completed
+LiteLLM Responses object can arrive with `response.output == []`. DSPy expects
+Responses output items to contain final message text, function calls, and
+reasoning summaries. This package reconstructs those output items from stream
+events before DSPy parses the response.
+
+It currently handles:
+
+- `response.output_item.done`
+- `response.output_text.done`
+- `response.output_text.delta`
+- `response.reasoning_summary_text.done`
+- `response.reasoning_summary_text.delta`
+- streamed function-call output items
+
+It also strips output-token cap fields that the Codex backend currently rejects:
+
+- `max_tokens`
+- `max_output_tokens`
+- `max_completion_tokens`
+
+## French Example
+
+```python
+import dspy
+import dspy_codex_auth
+
+dspy_codex_auth.install()
+
+lm = dspy.LM("codex/gpt-5.5", cache=False)
+dspy.configure(lm=lm, adapter=dspy.JSONAdapter())
+
+
+class TranslateFrenchToEnglish(dspy.Signature):
+    """Translate the French input into short, natural English."""
+
+    french: str = dspy.InputField(desc="French sentence")
+    english: str = dspy.OutputField(desc="Natural English translation")
+
+
+translator = dspy.Predict(TranslateFrenchToEnglish)
+print(translator(french="merci beaucoup").english)
+```
+
+## Math Example With Reasoning Summary
+
+```python
+import dspy
+import dspy_codex_auth
+
+dspy_codex_auth.install()
+
+lm = dspy.LM(
+    "codex/gpt-5.5",
+    cache=False,
+    reasoning_effort="medium",
+    reasoning_summary="detailed",
+)
+dspy.configure(lm=lm, adapter=dspy.JSONAdapter())
+
+
+class SolveMath(dspy.Signature):
+    """Solve the math problem. Return a concise numeric answer and a brief explanation."""
+
+    problem: str = dspy.InputField(desc="Math problem")
+    answer: str = dspy.OutputField(desc="Concise final answer")
+    explanation: str = dspy.OutputField(desc="Brief explanation")
+
+
+solver = dspy.Predict(SolveMath)
+pred = solver(
+    problem=(
+        "Compute the integral of the standard normal probability density "
+        "function from 0 to 1.5."
+    )
+)
+
+print(pred.answer)
+print(pred.explanation)
+print(lm.history[-1]["outputs"][0].get("reasoning_content"))
+```
+
+## Attribution
+
+`dspy-codex-auth` includes and adapts MIT-licensed auth and DSPy integration
+code from `dspy-lm-auth`:
+
+https://github.com/MaximeRivest/dspy-lm-auth
+
+The streamed-output reconstruction addresses a DSPy/Codex Responses streaming
+compatibility issue that was also discussed in `dspy-lm-auth` PR #2:
+
+https://github.com/MaximeRivest/dspy-lm-auth/pull/2
+
+`dspy-lm-auth` is MIT-licensed. The original copyright notice is preserved in
+`THIRD_PARTY_NOTICES.md`.
+
+## Development
+
+```bash
+uv sync --dev
+uv run pytest
+uv run ruff check .
+uv build --no-sources
+```
+
+## Release
+
+Releases are published from GitHub Actions with PyPI Trusted Publishing.
+
+For the first release, create a pending PyPI publisher for:
+
+- PyPI project: `dspy-codex-auth`
+- Owner: `hrbatra`
+- Repository: `dspy-codex-auth`
+- Workflow: `publish.yml`
+- Environment: `pypi`
+
+Then tag and push:
+
+```bash
+git tag v0.1.0
+git push origin v0.1.0
+```

dspy_codex_auth-0.1.0/pyproject.toml

@@ -0,0 +1,40 @@
+[project]
+name = "dspy-codex-auth"
+version = "0.1.0"
+description = "DSPy integration for ChatGPT Codex subscription-backed language models"
+readme = "README.md"
+authors = [
+    { name = "hrbatra", email = "hrbatra@utexas.edu" }
+]
+license = "MIT"
+requires-python = ">=3.12"
+keywords = ["codex", "dspy", "language-models", "openai"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+]
+dependencies = [
+    "dspy>=3.2.0",
+    "litellm>=1.82.6",
+    "requests>=2.33.1",
+]
+
+[project.urls]
+Homepage = "https://github.com/hrbatra/dspy-codex-auth"
+Repository = "https://github.com/hrbatra/dspy-codex-auth"
+Issues = "https://github.com/hrbatra/dspy-codex-auth/issues"
+
+[build-system]
+requires = ["uv_build>=0.9.17,<0.10.0"]
+build-backend = "uv_build"
+
+[dependency-groups]
+dev = [
+    "pytest>=9.0.3",
+    "ruff>=0.15.12",
+]

dspy_codex_auth-0.1.0/src/dspy_codex_auth/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 hrbatra
+
+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.

dspy_codex_auth-0.1.0/src/dspy_codex_auth/THIRD_PARTY_NOTICES.md

@@ -0,0 +1,38 @@
+# Third-Party Notices
+
+`dspy-codex-auth` includes and adapts MIT-licensed auth and DSPy integration
+code from `dspy-lm-auth`:
+
+https://github.com/MaximeRivest/dspy-lm-auth
+
+`dspy-lm-auth` is licensed under the MIT License.
+
+The Codex streamed-output reconstruction implemented here addresses a
+DSPy/Codex Responses streaming compatibility issue that was also discussed in
+`dspy-lm-auth` pull request #2:
+
+https://github.com/MaximeRivest/dspy-lm-auth/pull/2
+
+## dspy-lm-auth MIT License
+
+MIT License
+
+Copyright (c) 2026 Maxime Rivest
+
+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.

dspy_codex_auth-0.1.0/src/dspy_codex_auth/__init__.py

@@ -0,0 +1,74 @@
+from dspy_codex_auth.auth import (
+    OPENAI_CODEX_PROVIDER,
+    ApiKeyCredential,
+    AuthStorage,
+    Credential,
+    OAuthCredential,
+    OpenAICodexOAuthProvider,
+    build_openai_codex_authorization_url,
+    clear_command_cache,
+    exchange_openai_codex_code,
+    extract_chatgpt_account_id,
+    get_default_auth_storage,
+    get_oauth_provider,
+    getauthtoken,
+    login,
+    login_openai_codex,
+    logout,
+    normalize_provider_id,
+    parse_authorization_input,
+    refresh_openai_codex_token,
+    register_oauth_provider,
+    resolve_config_value,
+    set_default_auth_storage,
+)
+
+from dspy_codex_auth.lm import (
+    DEFAULT_CODEX_API_BASE,
+    DEFAULT_CODEX_INSTRUCTIONS,
+    DEFAULT_CODEX_MODEL,
+    DEFAULT_CODEX_ORIGINATOR,
+    LM,
+    codex_headers,
+    install,
+    register_model_alias,
+    resolve_lm_route,
+    uninstall,
+    unregister_model_alias,
+)
+
+__all__ = [
+    "DEFAULT_CODEX_API_BASE",
+    "DEFAULT_CODEX_INSTRUCTIONS",
+    "DEFAULT_CODEX_MODEL",
+    "DEFAULT_CODEX_ORIGINATOR",
+    "OPENAI_CODEX_PROVIDER",
+    "ApiKeyCredential",
+    "AuthStorage",
+    "Credential",
+    "LM",
+    "OAuthCredential",
+    "OpenAICodexOAuthProvider",
+    "build_openai_codex_authorization_url",
+    "clear_command_cache",
+    "codex_headers",
+    "exchange_openai_codex_code",
+    "extract_chatgpt_account_id",
+    "get_default_auth_storage",
+    "get_oauth_provider",
+    "getauthtoken",
+    "install",
+    "login",
+    "login_openai_codex",
+    "logout",
+    "normalize_provider_id",
+    "parse_authorization_input",
+    "refresh_openai_codex_token",
+    "register_model_alias",
+    "register_oauth_provider",
+    "resolve_config_value",
+    "resolve_lm_route",
+    "set_default_auth_storage",
+    "uninstall",
+    "unregister_model_alias",
+]