Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion packages/capabilities/image-generation/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ classifiers = [
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Typing :: Typed",
]
keywords = ["ai", "image-generation", "dall-e", "imagen", "openai", "google", "bytedance"]
keywords = ["ai", "image-generation", "dall-e", "imagen", "openai", "google", "byteplus"]

[project.urls]
Homepage = "https://withceleste.ai"
Expand All @@ -28,6 +28,7 @@ Issues = "https://github.com/withceleste/celeste-python/issues"
[tool.uv.sources]
celeste-ai = { workspace = true }
celeste-bfl = { workspace = true }
celeste-byteplus = { workspace = true }
celeste-google = { workspace = true }
celeste-openai = { workspace = true }

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,15 +2,15 @@

from celeste import Model
from celeste_image_generation.providers.bfl.models import MODELS as BFL_MODELS
from celeste_image_generation.providers.bytedance.models import (
MODELS as BYTEDANCE_MODELS,
from celeste_image_generation.providers.byteplus.models import (
MODELS as BYTEPLUS_MODELS,
)
from celeste_image_generation.providers.google.models import MODELS as GOOGLE_MODELS
from celeste_image_generation.providers.openai.models import MODELS as OPENAI_MODELS

MODELS: list[Model] = [
*BFL_MODELS,
*BYTEDANCE_MODELS,
*BYTEPLUS_MODELS,
*GOOGLE_MODELS,
*OPENAI_MODELS,
]
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,8 @@ def _get_providers() -> list[tuple[Provider, type[Client]]]:
from celeste_image_generation.providers.bfl.client import (
BFLImageGenerationClient,
)
from celeste_image_generation.providers.bytedance.client import (
ByteDanceImageGenerationClient,
from celeste_image_generation.providers.byteplus.client import (
BytePlusImageGenerationClient,
)
from celeste_image_generation.providers.google.client import (
GoogleImageGenerationClient,
Expand All @@ -23,7 +23,7 @@ def _get_providers() -> list[tuple[Provider, type[Client]]]:

return [
(Provider.BFL, BFLImageGenerationClient),
(Provider.BYTEDANCE, ByteDanceImageGenerationClient),
(Provider.BYTEPLUS, BytePlusImageGenerationClient),
(Provider.GOOGLE, GoogleImageGenerationClient),
(Provider.OPENAI, OpenAIImageGenerationClient),
]
Expand Down

This file was deleted.

This file was deleted.

This file was deleted.

This file was deleted.

Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
# ByteDance Image Generation Provider
# BytePlus Image Generation Provider

## Credentials

**Environment variable:** `BYTEDANCE_API_KEY`
**Environment variable:** `BYTEPLUS_API_KEY`

**Setup:**
1. Register at [console.byteplus.com](https://console.byteplus.com)
2. Activate model in ModelArk section
3. Generate API key with image generation permissions
4. Set environment variable: `export BYTEDANCE_API_KEY="your-key"`
4. Set environment variable: `export BYTEPLUS_API_KEY="your-key"`

**Note:** Models must be activated in BytePlus console before use. If you get a 404 error, activate the model or use an Endpoint ID (`ep-xxx`) instead of Model ID.
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
"""BytePlus provider for image generation."""

from .client import BytePlusImageGenerationClient
from .models import MODELS

__all__ = ["MODELS", "BytePlusImageGenerationClient"]
Loading
Loading