codex-lb 0.1.2__tar.gz → 0.1.4__tar.gz

{codex_lb-0.1.2 → codex_lb-0.1.4}/.github/release-please-config.json

@@ -6,8 +6,7 @@
       "changelog-path": "CHANGELOG.md",
       "extra-files": [
         "app/__init__.py"
-      ],
-      "skip-github-release": true
+      ]
     }
   }
 }

codex_lb-0.1.4/.github/release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.1.4"
+}

{codex_lb-0.1.2 → codex_lb-0.1.4}/.github/workflows/release-please.yml

@@ -4,6 +4,7 @@ on:
   push:
     branches:
       - main
+  workflow_dispatch:
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref }}

{codex_lb-0.1.2 → codex_lb-0.1.4}/.github/workflows/release.yml

@@ -1,15 +1,21 @@
 name: Release
 
 on:
-  push:
-    tags:
-      - "v*.*.*"
+  release:
+    types: [published]
   workflow_dispatch:
+    inputs:
+      tag:
+        description: "Release tag (e.g. v0.1.2)"
+        required: true
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref }}
   cancel-in-progress: true
 
+env:
+  RELEASE_TAG: ${{ github.event.release.tag_name || inputs.tag }}
+
 jobs:
   build:
     name: Build (sdist + wheel)
@@ -20,6 +26,7 @@ jobs:
         uses: actions/checkout@v4
         with:
           fetch-depth: 0
+          ref: ${{ env.RELEASE_TAG }}
 
       - name: Set up uv
         uses: astral-sh/setup-uv@v5
@@ -40,8 +47,10 @@ jobs:
           except Exception:  # pragma: no cover
               import tomli as tomllib
 
-          tag_ref = os.environ.get("GITHUB_REF", "")
-          tag = tag_ref.split("/")[-1]
+          tag = os.environ.get("RELEASE_TAG", "")
+          if not tag:
+              print("Missing RELEASE_TAG")
+              sys.exit(1)
           m = re.fullmatch(r"v(\d+\.\d+\.\d+)", tag)
           if not m:
               print(f"Invalid tag format: {tag!r}")
@@ -105,10 +114,15 @@ jobs:
     steps:
       - name: Checkout repository
         uses: actions/checkout@v4
+        with:
+          ref: ${{ env.RELEASE_TAG }}
 
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
 
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+
       - name: Log in to GHCR
         uses: docker/login-action@v3
         with:
@@ -133,6 +147,7 @@ jobs:
           context: .
           file: Dockerfile
           push: true
+          platforms: linux/amd64,linux/arm64
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
           cache-from: type=gha
@@ -151,6 +166,7 @@ jobs:
         uses: actions/checkout@v4
         with:
           fetch-depth: 0
+          ref: ${{ env.RELEASE_TAG }}
 
       - name: Download dist artifacts
         uses: actions/download-artifact@v4
@@ -158,62 +174,64 @@ jobs:
           name: dist
           path: dist
 
-      - name: Generate release notes
-        id: notes
+      - name: Resolve previous tag
+        id: prev_tag
         shell: bash
         run: |
           set -euo pipefail
-          TAG_NAME="${GITHUB_REF#refs/tags/}"
-          TAG_VERSION="${TAG_NAME#v}"
-          RELEASE_DATE="$(date -u +%Y-%m-%d)"
-
+          TAG_NAME="${RELEASE_TAG}"
           PREV_TAG=""
           if git describe --tags --abbrev=0 "${TAG_NAME}^" >/dev/null 2>&1; then
             PREV_TAG="$(git describe --tags --abbrev=0 "${TAG_NAME}^")"
           fi
-
           if [ -n "${PREV_TAG}" ]; then
             RANGE="${PREV_TAG}..${TAG_NAME}"
           else
             RANGE="${TAG_NAME}"
           fi
+          echo "tag_name=${TAG_NAME}" >> "$GITHUB_OUTPUT"
+          echo "prev_tag=${PREV_TAG}" >> "$GITHUB_OUTPUT"
+          echo "range=${RANGE}" >> "$GITHUB_OUTPUT"
 
-          COMMIT_COUNT="$(git rev-list --count "${RANGE}")"
-
+      - name: Generate release notes (GitHub)
+        id: notes
+        uses: actions/github-script@v7
+        env:
+          TAG_NAME: ${{ steps.prev_tag.outputs.tag_name }}
+          PREV_TAG: ${{ steps.prev_tag.outputs.prev_tag }}
+        with:
+          script: |
+            const fs = require("fs");
+            const tag = process.env.TAG_NAME;
+            const prev = process.env.PREV_TAG || undefined;
+            if (!tag) {
+              core.setFailed("Missing TAG_NAME");
+              return;
+            }
+            const { data } = await github.rest.repos.generateReleaseNotes({
+              owner: context.repo.owner,
+              repo: context.repo.repo,
+              tag_name: tag,
+              previous_tag_name: prev,
+            });
+            fs.writeFileSync("release_notes.md", `${data.body.trimEnd()}\\n`);
+            core.setOutput("tag_name", tag);
+
+      - name: Append install and contributors
+        shell: bash
+        env:
+          TAG_NAME: ${{ steps.prev_tag.outputs.tag_name }}
+        run: |
+          set -euo pipefail
+          TAG_VERSION="${TAG_NAME#v}"
           {
-            echo "## What's New in ${TAG_NAME}"
-            echo
-            echo "- Release date: ${RELEASE_DATE}"
-            echo "- Commits: ${COMMIT_COUNT}"
-            echo
-            if [ -n "${PREV_TAG}" ]; then
-              echo "### Changes"
-              git log --pretty=format:"- %s" "${RANGE}"
-            else
-              echo "### Changes"
-              git log --pretty=format:"- %s" "${RANGE}"
-            fi
             echo
-            echo
-            echo "### Install / Run"
+            echo "## Install / Run"
             echo '```bash'
             echo 'uvx codex-lb'
-            echo 'pip install codex-lb'
             echo "docker pull ghcr.io/${GITHUB_REPOSITORY}:${TAG_VERSION}"
             echo '```'
-            echo
-            echo "### Artifacts"
-            if ls dist >/dev/null 2>&1; then
-              ls -1 dist | sed 's/^/- dist\\//'
-            else
-              echo "- dist/ (not available)"
-            fi
-            echo
-            echo "### Contributors"
-            git shortlog -s "${RANGE}" | awk '{$1=$1}1' | sed 's/^/- /'
-          } > release_notes.md
-
-          echo "tag_name=${TAG_NAME}" >> "$GITHUB_OUTPUT"
+          } >> release_notes.md
 
       - name: Create GitHub Release
         uses: softprops/action-gh-release@v2

{codex_lb-0.1.2 → codex_lb-0.1.4}/CHANGELOG.md

@@ -1,5 +1,19 @@
 # Changelog
 
+## [0.1.4](https://github.com/Soju06/codex-lb/compare/v0.1.3...v0.1.4) (2026-01-13)
+
+
+### Bug Fixes
+
+* **db:** harden session cleanup on cancellation ([dee3916](https://github.com/Soju06/codex-lb/commit/dee3916efa83dedec1d5ad43e1e14950b8c6e4a7))
+
+## [0.1.3](https://github.com/Soju06/codex-lb/compare/v0.1.2...v0.1.3) (2026-01-12)
+
+
+### Documentation
+
+* use absolute image URLs for PyPI ([5fa65a5](https://github.com/Soju06/codex-lb/commit/5fa65a572980f356738f49be3adf2c62fdc38466))
+
 ## [0.1.2](https://github.com/Soju06/codex-lb/compare/v0.1.1...v0.1.2) (2026-01-12)
 
 

{codex_lb-0.1.2 → codex_lb-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codex-lb
-Version: 0.1.2
+Version: 0.1.4
 Summary: Codex load balancer and proxy for ChatGPT accounts with usage dashboard
 Author-email: Soju06 <qlskssk@gmail.com>
 Maintainer-email: Soju06 <qlskssk@gmail.com>
@@ -56,7 +56,7 @@ Description-Content-Type: text/markdown
 Load balancer for ChatGPT accounts. Pool multiple accounts, track usage, view everything in a dashboard.
 
 <p align="center">
-  <img src="docs/screenshots/dashboard.jpeg" alt="Codex Load Balancer dashboard" width="100%">
+  <img src="https://raw.githubusercontent.com/Soju06/codex-lb/main/docs/screenshots/dashboard.jpeg" alt="Codex Load Balancer dashboard" width="100%">
 </p>
 
 ## Quick Start
@@ -80,7 +80,7 @@ Open [localhost:2455](http://localhost:2455) → Add account → Done.
 
 ## Accounts view
 
-![Accounts list and details](docs/screenshots/accounts.jpeg)
+![Accounts list and details](https://raw.githubusercontent.com/Soju06/codex-lb/main/docs/screenshots/accounts.jpeg)
 
 ## Codex CLI & Extension Setup
 

{codex_lb-0.1.2 → codex_lb-0.1.4}/README.md

@@ -3,7 +3,7 @@
 Load balancer for ChatGPT accounts. Pool multiple accounts, track usage, view everything in a dashboard.
 
 <p align="center">
-  <img src="docs/screenshots/dashboard.jpeg" alt="Codex Load Balancer dashboard" width="100%">
+  <img src="https://raw.githubusercontent.com/Soju06/codex-lb/main/docs/screenshots/dashboard.jpeg" alt="Codex Load Balancer dashboard" width="100%">
 </p>
 
 ## Quick Start
@@ -27,7 +27,7 @@ Open [localhost:2455](http://localhost:2455) → Add account → Done.
 
 ## Accounts view
 
-![Accounts list and details](docs/screenshots/accounts.jpeg)
+![Accounts list and details](https://raw.githubusercontent.com/Soju06/codex-lb/main/docs/screenshots/accounts.jpeg)
 
 ## Codex CLI & Extension Setup
 

{codex_lb-0.1.2 → codex_lb-0.1.4}/app/db/session.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import asyncio
 from pathlib import Path
 from typing import AsyncIterator
 
@@ -23,16 +24,33 @@ def _ensure_sqlite_dir(url: str) -> None:
     Path(path).expanduser().parent.mkdir(parents=True, exist_ok=True)
 
 
+async def _safe_rollback(session: AsyncSession) -> None:
+    if not session.in_transaction():
+        return
+    try:
+        await asyncio.shield(session.rollback())
+    except Exception:
+        return
+
+
+async def _safe_close(session: AsyncSession) -> None:
+    try:
+        await asyncio.shield(session.close())
+    except Exception:
+        return
+
+
 async def get_session() -> AsyncIterator[AsyncSession]:
-    async with SessionLocal() as session:
-        try:
-            yield session
-        except Exception:
-            await session.rollback()
-            raise
-        finally:
-            if session.in_transaction():
-                await session.rollback()
+    session = SessionLocal()
+    try:
+        yield session
+    except BaseException:
+        await _safe_rollback(session)
+        raise
+    finally:
+        if session.in_transaction():
+            await _safe_rollback(session)
+        await _safe_close(session)
 
 
 async def init_db() -> None:

{codex_lb-0.1.2 → codex_lb-0.1.4}/app/dependencies.py

@@ -7,7 +7,7 @@ from dataclasses import dataclass
 from fastapi import Depends
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from app.db.session import SessionLocal, get_session
+from app.db.session import SessionLocal, _safe_close, _safe_rollback, get_session
 from app.modules.accounts.repository import AccountsRepository
 from app.modules.accounts.service import AccountsService
 from app.modules.oauth.service import OauthService
@@ -87,15 +87,16 @@ def get_usage_context(
 
 @asynccontextmanager
 async def _accounts_repo_context() -> AsyncIterator[AccountsRepository]:
-    async with SessionLocal() as session:
-        try:
-            yield AccountsRepository(session)
-        except Exception:
-            await session.rollback()
-            raise
-        finally:
-            if session.in_transaction():
-                await session.rollback()
+    session = SessionLocal()
+    try:
+        yield AccountsRepository(session)
+    except BaseException:
+        await _safe_rollback(session)
+        raise
+    finally:
+        if session.in_transaction():
+            await _safe_rollback(session)
+        await _safe_close(session)
 
 
 def get_oauth_context(

{codex_lb-0.1.2 → codex_lb-0.1.4}/app/modules/proxy/service.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import logging
 import time
 from datetime import timedelta
 from typing import AsyncIterator, Iterable, Mapping
@@ -36,6 +37,8 @@ from app.modules.proxy.usage_updater import UsageUpdater
 from app.modules.request_logs.repository import RequestLogsRepository
 from app.modules.usage.repository import UsageRepository
 
+logger = logging.getLogger(__name__)
+
 
 class ProxyService:
     def __init__(
@@ -191,6 +194,7 @@ class ProxyService:
                 yield format_sse_event(event)
                 return
 
+            account_id_value = account.id
             try:
                 account = await self._ensure_fresh(account)
                 async for line in self._stream_once(
@@ -243,7 +247,15 @@ class ProxyService:
                     await self._load_balancer.mark_permanent_failure(account, exc.code)
                 continue
             except Exception:
-                await self._load_balancer.record_error(account)
+                try:
+                    await self._load_balancer.record_error(account)
+                except Exception:
+                    logger.warning(
+                        "Failed to record proxy error account_id=%s request_id=%s",
+                        account_id_value,
+                        request_id,
+                        exc_info=True,
+                    )
                 if attempt == max_attempts - 1:
                     event = response_failed_event(
                         "upstream_error",
@@ -267,8 +279,9 @@ class ProxyService:
         request_id: str,
         allow_retry: bool,
     ) -> AsyncIterator[str]:
+        account_id_value = account.id
         access_token = self._encryptor.decrypt(account.access_token_encrypted)
-        account_id = _header_account_id(account.id)
+        account_id = _header_account_id(account_id_value)
         model = payload.model
         start = time.monotonic()
         status = "success"
@@ -341,19 +354,27 @@ class ProxyService:
             reasoning_tokens = (
                 usage.output_tokens_details.reasoning_tokens if usage and usage.output_tokens_details else None
             )
-            await self._logs_repo.add_log(
-                account_id=account.id,
-                request_id=request_id,
-                model=model,
-                input_tokens=input_tokens,
-                output_tokens=output_tokens,
-                cached_input_tokens=cached_input_tokens,
-                reasoning_tokens=reasoning_tokens,
-                latency_ms=latency_ms,
-                status=status,
-                error_code=error_code,
-                error_message=error_message,
-            )
+            try:
+                await self._logs_repo.add_log(
+                    account_id=account_id_value,
+                    request_id=request_id,
+                    model=model,
+                    input_tokens=input_tokens,
+                    output_tokens=output_tokens,
+                    cached_input_tokens=cached_input_tokens,
+                    reasoning_tokens=reasoning_tokens,
+                    latency_ms=latency_ms,
+                    status=status,
+                    error_code=error_code,
+                    error_message=error_message,
+                )
+            except Exception:
+                logger.warning(
+                    "Failed to persist request log account_id=%s request_id=%s",
+                    account_id_value,
+                    request_id,
+                    exc_info=True,
+                )
 
     async def _refresh_usage(self, accounts: list[Account]) -> None:
         latest_usage = await self._usage_repo.latest_by_account(window="primary")

{codex_lb-0.1.2 → codex_lb-0.1.4}/app/modules/request_logs/repository.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import asyncio
 from datetime import datetime
 
 from sqlalchemy import and_, select
@@ -49,9 +50,13 @@ class RequestLogsRepository:
             requested_at=requested_at or utcnow(),
         )
         self._session.add(log)
-        await self._session.commit()
-        await self._session.refresh(log)
-        return log
+        try:
+            await self._session.commit()
+            await self._session.refresh(log)
+            return log
+        except BaseException:
+            await _safe_rollback(self._session)
+            raise
 
     async def list_recent(
         self,
@@ -84,3 +89,12 @@ class RequestLogsRepository:
             stmt = stmt.limit(limit)
         result = await self._session.execute(stmt)
         return list(result.scalars().all())
+
+
+async def _safe_rollback(session: AsyncSession) -> None:
+    if not session.in_transaction():
+        return
+    try:
+        await asyncio.shield(session.rollback())
+    except Exception:
+        return

{codex_lb-0.1.2 → codex_lb-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "codex-lb"
-version = "0.1.2"
+version = "0.1.4"
 description = "Codex load balancer and proxy for ChatGPT accounts with usage dashboard"
 readme = "README.md"
 license = { file = "LICENSE" }

codex_lb-0.1.2/.github/release-please-manifest.json

@@ -1,3 +0,0 @@
-{
-  ".": "0.1.2"
-}