Skip to main content

ChatImg image generation package

Project description

ChatImg

ChatImg 是 ChatArch 的图片生成包,承接原 chattool image 中已经解耦的 provider 实现。

当前支持:

  • openai / crs:OpenAI-compatible Images API,走 OPENAI_API_KEY,请求 /v1/images/generations
  • codex / openai-codex:ChatGPT/Codex OAuth image bridge,走 CODEX_ACCESS_TOKEN,支持 gpt-image-2-* preset。
  • pollinations:Pollinations.ai image URL generation and model listing。
  • siliconflow:SiliconFlow OpenAI-compatible image generation。
  • huggingface:Hugging Face Inference image generation。
  • liblib:LiblibAI signed API generation。
  • tongyi:通义万相 / DashScope image generation。

安装

开发安装:

pip install -e ".[dev]"

通义万相 SDK 可选依赖:

pip install -e ".[images]"

CLI

chatimg --help
chatimg --version
chatimg openai generate "a small red apple icon" -o apple.png
chatimg codex list-models
chatimg pollinations list-models

生成示例:

chatimg openai generate "a watercolor fox in the snow" --model gpt-image-2-medium --size 1024x1024 -o fox.png
chatimg codex generate "a watercolor fox in the snow" --aspect-ratio square -o fox-codex.png
chatimg pollinations generate "a cyberpunk cat" --model flux --width 512 --height 512 -o cat.png
chatimg siliconflow generate "a cute dog" --size 1024x1024 -o dog.png
chatimg huggingface generate "A futuristic city at night" -o city.png
chatimg liblib generate "A cute dog" --model-id liblib-sdxl-model -o dog.png
chatimg tongyi generate "一只赛博朋克猫" --size "1024*1024" -o cat.png

OpenAI-compatible / CRS Images API

openai provider 对齐 OpenAI 官方 Images API:POST {OPENAI_API_BASE}/images/generations。官方 base 是 https://api.openai.com/v1,因此完整官方接口是 https://api.openai.com/v1/images/generations。CRS 兼容服务可用自己的带 /v1 base,例如 https://crs.example/openai/v1

常用配置字段:

  • OPENAI_API_BASE:OpenAI-compatible API base,必须包含 /v1
  • OPENAI_API_KEY:API key;不会 fallback 到 access token。
  • OPENAI_IMAGE_MODEL:默认 image preset,例如 gpt-image-2-medium,会拆成 model=gpt-image-2quality=medium
chatenv use -t oai apple
chatimg openai generate "a small red apple icon" -o apple.png

Codex / GPT Image2 实测示例

codex provider 走 ChatGPT/Codex OAuth-backed Responses API,请求里的 image tool model 是 gpt-image-2。它不是外部 Codex CLI。

常用配置字段:

  • CODEX_ACCESS_TOKEN:Codex OAuth access token。
  • CODEX_REFRESH_TOKEN:Codex OAuth refresh token,用于刷新 access token。
  • CODEX_ACCESS_TOKEN_EXPIRES_AT:access token 的 UTC ISO 过期时间。
  • CODEX_OAUTH_BASE_URL:Codex OAuth auth server base URL,默认 https://auth.openai.com
  • CODEX_API_BASE:Codex backend base URL,默认 https://chatgpt.com/backend-api/codex
  • CODEX_HOST_MODEL:承载 image_generation tool 的 host model,默认 gpt-5.4
  • CODEX_IMAGE_MODEL:默认 image preset,默认 gpt-image-2-medium

codex provider 不再读取 ~/.hermes/auth.json,也不再维护 OPENAI_CODEX_* 变量。--timeout--aspect-ratio 是命令级参数,不写入长期 env。

基础验收图:

chatimg codex generate \
  "A clean minimal ChatImg acceptance test illustration: a friendly robot holding a small picture frame, modern flat design, white background, no text" \
  --image-model gpt-image-2-low \
  --aspect-ratio square \
  -o generated/chatimg-gpt-image2-basic.png \
  --timeout 300

快速排序流程图:

chatimg codex generate \
  "Create a clean landscape technical design flowchart explaining quicksort. Use this exact example array: [6, 3, 8, 5, 1, 10, 2]. Pick pivot = 5. Partition correctly: left part [3, 1, 2] labeled < pivot, right part [6, 8, 10] labeled > pivot. Include steps: START, base case length <= 1?, pick pivot, partition array, recursively sort left part, recursively sort right part, concatenate sorted-left + pivot + sorted-right, END. Modern vector style, white background, blue/orange accents, arrows, decision diamond, readable simple English labels. Avoid mathematical mistakes." \
  --image-model gpt-image-2-medium \
  --aspect-ratio landscape \
  -o generated/chatimg-gpt-image2-quicksort-flowchart.png \
  --timeout 300

Python API

from chatimg.image import create_generator

generator = create_generator("codex")
result = generator.generate("A cute cat astronaut")

配置

ChatImg 注册了 chatenv.configs entry point:

chatimg = "chatimg.config"

支持的主要环境变量:

  • OPENAI_API_BASE, OPENAI_API_KEY, OPENAI_IMAGE_MODEL
  • CODEX_ACCESS_TOKEN, CODEX_REFRESH_TOKEN, CODEX_ACCESS_TOKEN_EXPIRES_AT, CODEX_OAUTH_BASE_URL, CODEX_API_BASE, CODEX_HOST_MODEL, CODEX_IMAGE_MODEL
  • POLLINATIONS_API_KEY, POLLINATIONS_MODEL_ID
  • SILICONFLOW_API_KEY, SILICONFLOW_MODEL_ID
  • HUGGINGFACE_HUB_TOKEN
  • LIBLIB_ACCESS_KEY, LIBLIB_SECRET_KEY, LIBLIB_MODEL_ID
  • DASHSCOPE_API_KEY

开发验证

PYTHONPATH=src python -m pytest -q
PYTHONPATH=src python -m chatimg.cli --help
PYTHONPATH=src python -m chatimg.cli codex list-models
python -m build
python -m twine check dist/*

发布状态

PyPI ChatImg0.1.x 开始发布功能版本;0.1.3 是 Codex 配置 alias 修正 patch 版本。

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

chatimg-0.1.3.tar.gz (23.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

chatimg-0.1.3-py3-none-any.whl (26.3 kB view details)

Uploaded Python 3

File details

Details for the file chatimg-0.1.3.tar.gz.

File metadata

  • Download URL: chatimg-0.1.3.tar.gz
  • Upload date:
  • Size: 23.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for chatimg-0.1.3.tar.gz
Algorithm Hash digest
SHA256 005fa150d88aed4cd2b3419912152b883422f79300ceb1ce8629be999ed8e27b
MD5 d5f9bb65fb3b5e9db50be26441be7c22
BLAKE2b-256 19e2b007d9718ca7d2397bb9719f9430d48cbebd10c5ee823e434c0e6e08d393

See more details on using hashes here.

Provenance

The following attestation bundles were made for chatimg-0.1.3.tar.gz:

Publisher: publish.yml on ChatArch/ChatImg

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file chatimg-0.1.3-py3-none-any.whl.

File metadata

  • Download URL: chatimg-0.1.3-py3-none-any.whl
  • Upload date:
  • Size: 26.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for chatimg-0.1.3-py3-none-any.whl
Algorithm Hash digest
SHA256 b2ed161b264fce83c810d99679b6984fce0a332cde5e2593013e6dbc689fe4f4
MD5 deeb3b9b7bd009bfddc289d4caff0ea5
BLAKE2b-256 7ca5a5b7889ffe5f75d588935db8dde557881b75e97ac507bed307d88a6ae1b1

See more details on using hashes here.

Provenance

The following attestation bundles were made for chatimg-0.1.3-py3-none-any.whl:

Publisher: publish.yml on ChatArch/ChatImg

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page