From 738740684f08d26cd2a1766aeb09eb2628bd70c7 Mon Sep 17 00:00:00 2001 From: kamilbenkirane Date: Sun, 14 Dec 2025 19:30:49 +0100 Subject: [PATCH 1/4] refactor: reorganize packages into capabilities directory Move all capability packages from packages/ to packages/capabilities/ to prepare for v0.3 architecture which separates capabilities from providers. - Move text-generation, image-generation, video-generation, speech-generation - Update pyproject.toml workspace members and pythonpath - Update Makefile paths for typecheck, test, integration-test --- Makefile | 6 ++--- .../image-generation/README.md | 4 ++-- .../image-generation/pyproject.toml | 0 .../src/celeste_image_generation/__init__.py | 0 .../src/celeste_image_generation/client.py | 0 .../celeste_image_generation/constraints.py | 0 .../src/celeste_image_generation/io.py | 0 .../src/celeste_image_generation/models.py | 0 .../celeste_image_generation/parameters.py | 0 .../providers/__init__.py | 0 .../providers/bfl/__init__.py | 0 .../providers/bfl/client.py | 0 .../providers/bfl/config.py | 0 .../providers/bfl/models.py | 0 .../providers/bfl/parameters.py | 0 .../providers/bytedance/README.md | 0 .../providers/bytedance/__init__.py | 0 .../providers/bytedance/client.py | 0 .../providers/bytedance/config.py | 0 .../providers/bytedance/models.py | 0 .../providers/bytedance/parameters.py | 0 .../providers/bytedance/streaming.py | 0 .../providers/google/__init__.py | 0 .../providers/google/client.py | 0 .../providers/google/config.py | 0 .../providers/google/gemini_api.py | 0 .../providers/google/imagen_api.py | 0 .../providers/google/models.py | 0 .../providers/google/parameters.py | 0 .../providers/openai/__init__.py | 0 .../providers/openai/client.py | 0 .../providers/openai/config.py | 0 .../providers/openai/models.py | 0 .../providers/openai/parameters.py | 0 .../providers/openai/streaming.py | 0 .../src/celeste_image_generation/py.typed | 0 .../src/celeste_image_generation/streaming.py | 0 .../test_image_generation/__init__.py | 0 .../test_image_generation/test_generate.py | 0 .../unit_tests/providers/google}/__init__.py | 0 .../providers/google/test_finish_reason.py | 0 .../speech-generation/README.md | 4 ++-- .../speech-generation/pyproject.toml | 0 .../src/celeste_speech_generation/__init__.py | 0 .../src/celeste_speech_generation/client.py | 0 .../celeste_speech_generation/constraints.py | 0 .../src/celeste_speech_generation/io.py | 0 .../src/celeste_speech_generation/models.py | 0 .../celeste_speech_generation/parameters.py | 0 .../providers/__init__.py | 0 .../providers/elevenlabs/__init__.py | 0 .../providers/elevenlabs/client.py | 0 .../providers/elevenlabs/config.py | 0 .../providers/elevenlabs/models.py | 0 .../providers/elevenlabs/parameters.py | 0 .../providers/elevenlabs/streaming.py | 0 .../providers/elevenlabs/voices.py | 0 .../providers/google/__init__.py | 0 .../providers/google/client.py | 0 .../providers/google/config.py | 0 .../providers/google/models.py | 0 .../providers/google/parameters.py | 0 .../providers/google/voices.py | 0 .../providers/openai/__init__.py | 0 .../providers/openai/client.py | 0 .../providers/openai/config.py | 0 .../providers/openai/models.py | 0 .../providers/openai/parameters.py | 0 .../providers/openai/voices.py | 0 .../src/celeste_speech_generation/py.typed | 0 .../celeste_speech_generation/streaming.py | 0 .../src/celeste_speech_generation/voices.py | 0 .../speech-generation/tests}/conftest.py | 0 .../tests/integration_tests}/__init__.py | 0 .../tests/integration_tests/conftest.py | 24 +++++++++++++++++++ .../test_speech_generation}/__init__.py | 0 .../test_speech_generation/test_generate.py | 0 .../test_speech_generation/test_stream.py | 0 .../text-generation/README.md | 4 ++-- .../text-generation/pyproject.toml | 0 .../src/celeste_text_generation/__init__.py | 0 .../src/celeste_text_generation/client.py | 0 .../src/celeste_text_generation/io.py | 0 .../src/celeste_text_generation/models.py | 0 .../src/celeste_text_generation/parameters.py | 0 .../providers/__init__.py | 0 .../providers/anthropic/__init__.py | 0 .../providers/anthropic/client.py | 0 .../providers/anthropic/config.py | 0 .../providers/anthropic/models.py | 0 .../providers/anthropic/parameters.py | 0 .../providers/anthropic/streaming.py | 0 .../providers/cohere/__init__.py | 0 .../providers/cohere/client.py | 0 .../providers/cohere/config.py | 0 .../providers/cohere/models.py | 0 .../providers/cohere/parameters.py | 0 .../providers/cohere/streaming.py | 0 .../providers/google/__init__.py | 0 .../providers/google/client.py | 0 .../providers/google/config.py | 0 .../providers/google/models.py | 0 .../providers/google/parameters.py | 0 .../providers/google/streaming.py | 0 .../providers/mistral/__init__.py | 0 .../providers/mistral/client.py | 0 .../providers/mistral/config.py | 0 .../providers/mistral/models.py | 0 .../providers/mistral/parameters.py | 0 .../providers/mistral/streaming.py | 0 .../providers/openai/__init__.py | 0 .../providers/openai/client.py | 0 .../providers/openai/config.py | 0 .../providers/openai/models.py | 0 .../providers/openai/parameters.py | 0 .../providers/openai/streaming.py | 0 .../providers/xai/__init__.py | 0 .../providers/xai/client.py | 0 .../providers/xai/config.py | 0 .../providers/xai/models.py | 0 .../providers/xai/parameters.py | 0 .../providers/xai/streaming.py | 0 .../src/celeste_text_generation/py.typed | 0 .../src/celeste_text_generation/streaming.py | 0 .../tests/integration_tests/conftest.py | 0 .../test_text_generation/__init__.py | 0 .../test_text_generation/test_generate.py | 0 .../test_text_generation/test_stream.py | 0 .../video-generation/README.md | 4 ++-- .../video-generation/pyproject.toml | 0 .../src/celeste_video_generation/__init__.py | 0 .../src/celeste_video_generation/client.py | 0 .../src/celeste_video_generation/io.py | 0 .../src/celeste_video_generation/models.py | 0 .../celeste_video_generation/parameters.py | 0 .../providers/__init__.py | 0 .../providers/bytedance/README.md | 0 .../providers/bytedance/__init__.py | 0 .../providers/bytedance/client.py | 0 .../providers/bytedance/config.py | 0 .../providers/bytedance/models.py | 0 .../providers/bytedance/parameters.py | 0 .../providers/google/__init__.py | 0 .../providers/google/client.py | 0 .../providers/google/config.py | 0 .../providers/google/models.py | 0 .../providers/google/parameters.py | 0 .../providers/openai/__init__.py | 0 .../providers/openai/client.py | 0 .../providers/openai/config.py | 0 .../providers/openai/models.py | 0 .../providers/openai/parameters.py | 0 .../src/celeste_video_generation/py.typed | 0 .../test_video_generation/__init__.py | 0 .../test_video_generation/test_generate.py | 0 .../test_speech_generation/__init__.py | 0 pyproject.toml | 10 ++++---- 157 files changed, 40 insertions(+), 16 deletions(-) rename packages/{ => capabilities}/image-generation/README.md (95%) rename packages/{ => capabilities}/image-generation/pyproject.toml (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/__init__.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/client.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/constraints.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/io.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/models.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/parameters.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/__init__.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bfl/__init__.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bfl/client.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bfl/config.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bfl/models.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bfl/parameters.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bytedance/README.md (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bytedance/__init__.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bytedance/client.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bytedance/config.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bytedance/models.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bytedance/parameters.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/bytedance/streaming.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/google/__init__.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/google/client.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/google/config.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/google/gemini_api.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/google/imagen_api.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/google/models.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/google/parameters.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/openai/__init__.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/openai/client.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/openai/config.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/openai/models.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/openai/parameters.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/providers/openai/streaming.py (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/py.typed (100%) rename packages/{ => capabilities}/image-generation/src/celeste_image_generation/streaming.py (100%) rename packages/{ => capabilities}/image-generation/tests/integration_tests/test_image_generation/__init__.py (100%) rename packages/{ => capabilities}/image-generation/tests/integration_tests/test_image_generation/test_generate.py (100%) rename packages/{image-generation/tests/unit_tests => capabilities/image-generation/tests/unit_tests/providers/google}/__init__.py (100%) rename packages/{ => capabilities}/image-generation/tests/unit_tests/providers/google/test_finish_reason.py (100%) rename packages/{ => capabilities}/speech-generation/README.md (95%) rename packages/{ => capabilities}/speech-generation/pyproject.toml (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/__init__.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/client.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/constraints.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/io.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/models.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/parameters.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/__init__.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/elevenlabs/__init__.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/elevenlabs/client.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/elevenlabs/config.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/elevenlabs/models.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/elevenlabs/parameters.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/elevenlabs/streaming.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/elevenlabs/voices.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/google/__init__.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/google/client.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/google/config.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/google/models.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/google/parameters.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/google/voices.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/openai/__init__.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/openai/client.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/openai/config.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/openai/models.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/openai/parameters.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/providers/openai/voices.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/py.typed (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/streaming.py (100%) rename packages/{ => capabilities}/speech-generation/src/celeste_speech_generation/voices.py (100%) rename packages/{speech-generation/tests/integration_tests => capabilities/speech-generation/tests}/conftest.py (100%) rename packages/{image-generation/tests/unit_tests/providers/google => capabilities/speech-generation/tests/integration_tests}/__init__.py (100%) create mode 100644 packages/capabilities/speech-generation/tests/integration_tests/conftest.py rename packages/{speech-generation/tests/integration_tests => capabilities/speech-generation/tests/integration_tests/test_speech_generation}/__init__.py (100%) rename packages/{ => capabilities}/speech-generation/tests/integration_tests/test_speech_generation/test_generate.py (100%) rename packages/{ => capabilities}/speech-generation/tests/integration_tests/test_speech_generation/test_stream.py (100%) rename packages/{ => capabilities}/text-generation/README.md (95%) rename packages/{ => capabilities}/text-generation/pyproject.toml (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/__init__.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/client.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/io.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/models.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/parameters.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/__init__.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/anthropic/__init__.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/anthropic/client.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/anthropic/config.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/anthropic/models.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/anthropic/parameters.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/anthropic/streaming.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/cohere/__init__.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/cohere/client.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/cohere/config.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/cohere/models.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/cohere/parameters.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/cohere/streaming.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/google/__init__.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/google/client.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/google/config.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/google/models.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/google/parameters.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/google/streaming.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/mistral/__init__.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/mistral/client.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/mistral/config.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/mistral/models.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/mistral/parameters.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/mistral/streaming.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/openai/__init__.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/openai/client.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/openai/config.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/openai/models.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/openai/parameters.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/openai/streaming.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/xai/__init__.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/xai/client.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/xai/config.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/xai/models.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/xai/parameters.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/providers/xai/streaming.py (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/py.typed (100%) rename packages/{ => capabilities}/text-generation/src/celeste_text_generation/streaming.py (100%) rename packages/{ => capabilities}/text-generation/tests/integration_tests/conftest.py (100%) rename packages/{ => capabilities}/text-generation/tests/integration_tests/test_text_generation/__init__.py (100%) rename packages/{ => capabilities}/text-generation/tests/integration_tests/test_text_generation/test_generate.py (100%) rename packages/{ => capabilities}/text-generation/tests/integration_tests/test_text_generation/test_stream.py (100%) rename packages/{ => capabilities}/video-generation/README.md (95%) rename packages/{ => capabilities}/video-generation/pyproject.toml (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/__init__.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/client.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/io.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/models.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/parameters.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/__init__.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/bytedance/README.md (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/bytedance/__init__.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/bytedance/client.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/bytedance/config.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/bytedance/models.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/bytedance/parameters.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/google/__init__.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/google/client.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/google/config.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/google/models.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/google/parameters.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/openai/__init__.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/openai/client.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/openai/config.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/openai/models.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/providers/openai/parameters.py (100%) rename packages/{ => capabilities}/video-generation/src/celeste_video_generation/py.typed (100%) rename packages/{ => capabilities}/video-generation/tests/integration_tests/test_video_generation/__init__.py (100%) rename packages/{ => capabilities}/video-generation/tests/integration_tests/test_video_generation/test_generate.py (100%) delete mode 100644 packages/speech-generation/tests/integration_tests/test_speech_generation/__init__.py diff --git a/Makefile b/Makefile index 2fa2a81d..8be072a8 100644 --- a/Makefile +++ b/Makefile @@ -33,15 +33,15 @@ format: # Type checking (fail fast on any error) typecheck: - @uv run mypy -p celeste && uv run mypy tests/ && uv run mypy packages/image-generation packages/text-generation packages/video-generation packages/speech-generation + @uv run mypy -p celeste && uv run mypy tests/ && uv run mypy packages/capabilities/image-generation packages/capabilities/text-generation packages/capabilities/video-generation packages/capabilities/speech-generation # Testing test: - uv run pytest tests/unit_tests packages/*/tests/unit_tests --cov=celeste --cov-report=term-missing --cov-fail-under=80 -v + uv run pytest tests/unit_tests packages/capabilities/*/tests/unit_tests --cov=celeste --cov-report=term-missing --cov-fail-under=80 -v # Integration testing (requires API keys) integration-test: - uv run pytest packages/*/tests/integration_tests/ -m integration -v --dist=worksteal -n auto + uv run pytest packages/capabilities/*/tests/integration_tests/ -m integration -v --dist=worksteal -n auto # Security scanning (config reads from pyproject.toml) security: diff --git a/packages/image-generation/README.md b/packages/capabilities/image-generation/README.md similarity index 95% rename from packages/image-generation/README.md rename to packages/capabilities/image-generation/README.md index 4896830d..c0a6ae62 100644 --- a/packages/image-generation/README.md +++ b/packages/capabilities/image-generation/README.md @@ -5,7 +5,7 @@ **Image Generation capability for Celeste AI** [![Python](https://img.shields.io/badge/Python-3.12+-blue?style=for-the-badge)](https://www.python.org/) -[![License](https://img.shields.io/badge/License-MIT-yellow?style=for-the-badge)](../../LICENSE) +[![License](https://img.shields.io/badge/License-MIT-yellow?style=for-the-badge)](../../../LICENSE) [Quick Start](#-quick-start) • [Documentation](https://withceleste.ai/docs) • [Request Provider](https://github.com/withceleste/celeste-python/issues/new) @@ -66,7 +66,7 @@ See [CONTRIBUTING.md](../../CONTRIBUTING.md) for guidelines. ## 📄 License -MIT License – see [LICENSE](../../LICENSE) for details. +MIT License – see [LICENSE](../../../LICENSE) for details. --- diff --git a/packages/image-generation/pyproject.toml b/packages/capabilities/image-generation/pyproject.toml similarity index 100% rename from packages/image-generation/pyproject.toml rename to packages/capabilities/image-generation/pyproject.toml diff --git a/packages/image-generation/src/celeste_image_generation/__init__.py b/packages/capabilities/image-generation/src/celeste_image_generation/__init__.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/__init__.py rename to packages/capabilities/image-generation/src/celeste_image_generation/__init__.py diff --git a/packages/image-generation/src/celeste_image_generation/client.py b/packages/capabilities/image-generation/src/celeste_image_generation/client.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/client.py rename to packages/capabilities/image-generation/src/celeste_image_generation/client.py diff --git a/packages/image-generation/src/celeste_image_generation/constraints.py b/packages/capabilities/image-generation/src/celeste_image_generation/constraints.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/constraints.py rename to packages/capabilities/image-generation/src/celeste_image_generation/constraints.py diff --git a/packages/image-generation/src/celeste_image_generation/io.py b/packages/capabilities/image-generation/src/celeste_image_generation/io.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/io.py rename to packages/capabilities/image-generation/src/celeste_image_generation/io.py diff --git a/packages/image-generation/src/celeste_image_generation/models.py b/packages/capabilities/image-generation/src/celeste_image_generation/models.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/models.py rename to packages/capabilities/image-generation/src/celeste_image_generation/models.py diff --git a/packages/image-generation/src/celeste_image_generation/parameters.py b/packages/capabilities/image-generation/src/celeste_image_generation/parameters.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/parameters.py rename to packages/capabilities/image-generation/src/celeste_image_generation/parameters.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/__init__.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/__init__.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/__init__.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/__init__.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/bfl/__init__.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/__init__.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bfl/__init__.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/__init__.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/bfl/client.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/client.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bfl/client.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/client.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/bfl/config.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/config.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bfl/config.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/config.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/bfl/models.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/models.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bfl/models.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/models.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/bfl/parameters.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/parameters.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bfl/parameters.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/parameters.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/README.md b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/README.md similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bytedance/README.md rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/README.md diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/__init__.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/__init__.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bytedance/__init__.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/__init__.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/client.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/client.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bytedance/client.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/client.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/config.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/config.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bytedance/config.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/config.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/models.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/models.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bytedance/models.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/models.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/parameters.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/parameters.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bytedance/parameters.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/parameters.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/streaming.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/streaming.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/bytedance/streaming.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/bytedance/streaming.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/__init__.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/google/__init__.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/google/__init__.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/google/__init__.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/client.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/google/client.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/google/client.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/google/client.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/config.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/google/config.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/google/config.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/google/config.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/gemini_api.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/google/gemini_api.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/google/gemini_api.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/google/gemini_api.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/imagen_api.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/google/imagen_api.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/google/imagen_api.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/google/imagen_api.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/models.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/google/models.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/google/models.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/google/models.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/parameters.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/google/parameters.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/google/parameters.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/google/parameters.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/openai/__init__.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/__init__.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/openai/__init__.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/__init__.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/openai/client.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/client.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/openai/client.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/client.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/openai/config.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/config.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/openai/config.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/config.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/openai/models.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/models.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/openai/models.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/models.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/openai/parameters.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/parameters.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/openai/parameters.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/parameters.py diff --git a/packages/image-generation/src/celeste_image_generation/providers/openai/streaming.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/streaming.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/providers/openai/streaming.py rename to packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/streaming.py diff --git a/packages/image-generation/src/celeste_image_generation/py.typed b/packages/capabilities/image-generation/src/celeste_image_generation/py.typed similarity index 100% rename from packages/image-generation/src/celeste_image_generation/py.typed rename to packages/capabilities/image-generation/src/celeste_image_generation/py.typed diff --git a/packages/image-generation/src/celeste_image_generation/streaming.py b/packages/capabilities/image-generation/src/celeste_image_generation/streaming.py similarity index 100% rename from packages/image-generation/src/celeste_image_generation/streaming.py rename to packages/capabilities/image-generation/src/celeste_image_generation/streaming.py diff --git a/packages/image-generation/tests/integration_tests/test_image_generation/__init__.py b/packages/capabilities/image-generation/tests/integration_tests/test_image_generation/__init__.py similarity index 100% rename from packages/image-generation/tests/integration_tests/test_image_generation/__init__.py rename to packages/capabilities/image-generation/tests/integration_tests/test_image_generation/__init__.py diff --git a/packages/image-generation/tests/integration_tests/test_image_generation/test_generate.py b/packages/capabilities/image-generation/tests/integration_tests/test_image_generation/test_generate.py similarity index 100% rename from packages/image-generation/tests/integration_tests/test_image_generation/test_generate.py rename to packages/capabilities/image-generation/tests/integration_tests/test_image_generation/test_generate.py diff --git a/packages/image-generation/tests/unit_tests/__init__.py b/packages/capabilities/image-generation/tests/unit_tests/providers/google/__init__.py similarity index 100% rename from packages/image-generation/tests/unit_tests/__init__.py rename to packages/capabilities/image-generation/tests/unit_tests/providers/google/__init__.py diff --git a/packages/image-generation/tests/unit_tests/providers/google/test_finish_reason.py b/packages/capabilities/image-generation/tests/unit_tests/providers/google/test_finish_reason.py similarity index 100% rename from packages/image-generation/tests/unit_tests/providers/google/test_finish_reason.py rename to packages/capabilities/image-generation/tests/unit_tests/providers/google/test_finish_reason.py diff --git a/packages/speech-generation/README.md b/packages/capabilities/speech-generation/README.md similarity index 95% rename from packages/speech-generation/README.md rename to packages/capabilities/speech-generation/README.md index 42ad8d11..9df87cb0 100644 --- a/packages/speech-generation/README.md +++ b/packages/capabilities/speech-generation/README.md @@ -5,7 +5,7 @@ **Speech Generation capability for Celeste AI** [![Python](https://img.shields.io/badge/Python-3.12+-blue?style=for-the-badge)](https://www.python.org/) -[![License](https://img.shields.io/badge/License-MIT-yellow?style=for-the-badge)](../../LICENSE) +[![License](https://img.shields.io/badge/License-MIT-yellow?style=for-the-badge)](../../../LICENSE) [Quick Start](#-quick-start) • [Documentation](https://withceleste.ai/docs) • [Request Provider](https://github.com/withceleste/celeste-python/issues/new) @@ -66,7 +66,7 @@ See [CONTRIBUTING.md](../../CONTRIBUTING.md) for guidelines. ## 📄 License -MIT License – see [LICENSE](../../LICENSE) for details. +MIT License – see [LICENSE](../../../LICENSE) for details. --- diff --git a/packages/speech-generation/pyproject.toml b/packages/capabilities/speech-generation/pyproject.toml similarity index 100% rename from packages/speech-generation/pyproject.toml rename to packages/capabilities/speech-generation/pyproject.toml diff --git a/packages/speech-generation/src/celeste_speech_generation/__init__.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/__init__.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/__init__.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/__init__.py diff --git a/packages/speech-generation/src/celeste_speech_generation/client.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/client.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/client.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/client.py diff --git a/packages/speech-generation/src/celeste_speech_generation/constraints.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/constraints.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/constraints.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/constraints.py diff --git a/packages/speech-generation/src/celeste_speech_generation/io.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/io.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/io.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/io.py diff --git a/packages/speech-generation/src/celeste_speech_generation/models.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/models.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/models.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/models.py diff --git a/packages/speech-generation/src/celeste_speech_generation/parameters.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/parameters.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/parameters.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/parameters.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/__init__.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/__init__.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/__init__.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/__init__.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/__init__.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/__init__.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/__init__.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/__init__.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/client.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/client.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/client.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/client.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/config.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/config.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/config.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/config.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/models.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/models.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/models.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/models.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/parameters.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/parameters.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/parameters.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/parameters.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/streaming.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/streaming.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/streaming.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/streaming.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/voices.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/voices.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/elevenlabs/voices.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/elevenlabs/voices.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/google/__init__.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/__init__.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/google/__init__.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/__init__.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/google/client.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/client.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/google/client.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/client.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/google/config.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/config.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/google/config.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/config.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/google/models.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/models.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/google/models.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/models.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/google/parameters.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/parameters.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/google/parameters.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/parameters.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/google/voices.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/voices.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/google/voices.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/google/voices.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/openai/__init__.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/__init__.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/openai/__init__.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/__init__.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/openai/client.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/client.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/openai/client.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/client.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/openai/config.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/config.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/openai/config.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/config.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/openai/models.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/models.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/openai/models.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/models.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/openai/parameters.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/parameters.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/openai/parameters.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/parameters.py diff --git a/packages/speech-generation/src/celeste_speech_generation/providers/openai/voices.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/voices.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/providers/openai/voices.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/providers/openai/voices.py diff --git a/packages/speech-generation/src/celeste_speech_generation/py.typed b/packages/capabilities/speech-generation/src/celeste_speech_generation/py.typed similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/py.typed rename to packages/capabilities/speech-generation/src/celeste_speech_generation/py.typed diff --git a/packages/speech-generation/src/celeste_speech_generation/streaming.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/streaming.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/streaming.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/streaming.py diff --git a/packages/speech-generation/src/celeste_speech_generation/voices.py b/packages/capabilities/speech-generation/src/celeste_speech_generation/voices.py similarity index 100% rename from packages/speech-generation/src/celeste_speech_generation/voices.py rename to packages/capabilities/speech-generation/src/celeste_speech_generation/voices.py diff --git a/packages/speech-generation/tests/integration_tests/conftest.py b/packages/capabilities/speech-generation/tests/conftest.py similarity index 100% rename from packages/speech-generation/tests/integration_tests/conftest.py rename to packages/capabilities/speech-generation/tests/conftest.py diff --git a/packages/image-generation/tests/unit_tests/providers/google/__init__.py b/packages/capabilities/speech-generation/tests/integration_tests/__init__.py similarity index 100% rename from packages/image-generation/tests/unit_tests/providers/google/__init__.py rename to packages/capabilities/speech-generation/tests/integration_tests/__init__.py diff --git a/packages/capabilities/speech-generation/tests/integration_tests/conftest.py b/packages/capabilities/speech-generation/tests/integration_tests/conftest.py new file mode 100644 index 00000000..81c92564 --- /dev/null +++ b/packages/capabilities/speech-generation/tests/integration_tests/conftest.py @@ -0,0 +1,24 @@ +"""Pytest configuration and fixtures for integration tests.""" + +from collections.abc import AsyncGenerator +from typing import Any + +import pytest_asyncio + +from celeste.http import close_all_http_clients + + +@pytest_asyncio.fixture(autouse=True) +async def cleanup_http_clients() -> AsyncGenerator[None, None]: + """Ensure HTTP clients are closed after each test. + + This fixture runs automatically after each test to ensure HTTP clients + are properly closed before pytest-asyncio closes the event loop. + This prevents "Event loop is closed" errors when using pytest-xdist. + """ + yield + await close_all_http_clients() + + +def pytest_configure(config: Any) -> None: # noqa: ANN401 + config.addinivalue_line("markers", "integration: mark test as an integration test") diff --git a/packages/speech-generation/tests/integration_tests/__init__.py b/packages/capabilities/speech-generation/tests/integration_tests/test_speech_generation/__init__.py similarity index 100% rename from packages/speech-generation/tests/integration_tests/__init__.py rename to packages/capabilities/speech-generation/tests/integration_tests/test_speech_generation/__init__.py diff --git a/packages/speech-generation/tests/integration_tests/test_speech_generation/test_generate.py b/packages/capabilities/speech-generation/tests/integration_tests/test_speech_generation/test_generate.py similarity index 100% rename from packages/speech-generation/tests/integration_tests/test_speech_generation/test_generate.py rename to packages/capabilities/speech-generation/tests/integration_tests/test_speech_generation/test_generate.py diff --git a/packages/speech-generation/tests/integration_tests/test_speech_generation/test_stream.py b/packages/capabilities/speech-generation/tests/integration_tests/test_speech_generation/test_stream.py similarity index 100% rename from packages/speech-generation/tests/integration_tests/test_speech_generation/test_stream.py rename to packages/capabilities/speech-generation/tests/integration_tests/test_speech_generation/test_stream.py diff --git a/packages/text-generation/README.md b/packages/capabilities/text-generation/README.md similarity index 95% rename from packages/text-generation/README.md rename to packages/capabilities/text-generation/README.md index 64c60c12..cca61ea0 100644 --- a/packages/text-generation/README.md +++ b/packages/capabilities/text-generation/README.md @@ -5,7 +5,7 @@ **Text Generation capability for Celeste AI** [![Python](https://img.shields.io/badge/Python-3.12+-blue?style=for-the-badge)](https://www.python.org/) -[![License](https://img.shields.io/badge/License-MIT-yellow?style=for-the-badge)](../../LICENSE) +[![License](https://img.shields.io/badge/License-MIT-yellow?style=for-the-badge)](../../../LICENSE) [Quick Start](#-quick-start) • [Documentation](https://withceleste.ai/docs) • [Request Provider](https://github.com/withceleste/celeste-python/issues/new) @@ -68,7 +68,7 @@ See [CONTRIBUTING.md](../../CONTRIBUTING.md) for guidelines. ## 📄 License -MIT License – see [LICENSE](../../LICENSE) for details. +MIT License – see [LICENSE](../../../LICENSE) for details. --- diff --git a/packages/text-generation/pyproject.toml b/packages/capabilities/text-generation/pyproject.toml similarity index 100% rename from packages/text-generation/pyproject.toml rename to packages/capabilities/text-generation/pyproject.toml diff --git a/packages/text-generation/src/celeste_text_generation/__init__.py b/packages/capabilities/text-generation/src/celeste_text_generation/__init__.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/__init__.py rename to packages/capabilities/text-generation/src/celeste_text_generation/__init__.py diff --git a/packages/text-generation/src/celeste_text_generation/client.py b/packages/capabilities/text-generation/src/celeste_text_generation/client.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/client.py rename to packages/capabilities/text-generation/src/celeste_text_generation/client.py diff --git a/packages/text-generation/src/celeste_text_generation/io.py b/packages/capabilities/text-generation/src/celeste_text_generation/io.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/io.py rename to packages/capabilities/text-generation/src/celeste_text_generation/io.py diff --git a/packages/text-generation/src/celeste_text_generation/models.py b/packages/capabilities/text-generation/src/celeste_text_generation/models.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/models.py rename to packages/capabilities/text-generation/src/celeste_text_generation/models.py diff --git a/packages/text-generation/src/celeste_text_generation/parameters.py b/packages/capabilities/text-generation/src/celeste_text_generation/parameters.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/parameters.py rename to packages/capabilities/text-generation/src/celeste_text_generation/parameters.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/__init__.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/__init__.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/__init__.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/__init__.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/anthropic/__init__.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/__init__.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/anthropic/__init__.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/__init__.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/anthropic/client.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/client.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/anthropic/client.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/client.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/anthropic/config.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/config.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/anthropic/config.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/config.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/anthropic/models.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/models.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/anthropic/models.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/models.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/anthropic/parameters.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/parameters.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/anthropic/parameters.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/parameters.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/anthropic/streaming.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/streaming.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/anthropic/streaming.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/anthropic/streaming.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/cohere/__init__.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/__init__.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/cohere/__init__.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/__init__.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/cohere/client.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/client.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/cohere/client.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/client.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/cohere/config.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/config.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/cohere/config.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/config.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/cohere/models.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/models.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/cohere/models.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/models.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/cohere/parameters.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/parameters.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/cohere/parameters.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/parameters.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/cohere/streaming.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/streaming.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/cohere/streaming.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/cohere/streaming.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/google/__init__.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/google/__init__.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/google/__init__.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/google/__init__.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/google/client.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/google/client.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/google/client.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/google/client.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/google/config.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/google/config.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/google/config.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/google/config.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/google/models.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/google/models.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/google/models.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/google/models.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/google/parameters.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/google/parameters.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/google/parameters.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/google/parameters.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/google/streaming.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/google/streaming.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/google/streaming.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/google/streaming.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/__init__.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/__init__.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/mistral/__init__.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/__init__.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/client.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/client.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/mistral/client.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/client.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/config.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/config.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/mistral/config.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/config.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/models.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/mistral/models.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/models.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/parameters.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/parameters.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/mistral/parameters.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/parameters.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/streaming.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/streaming.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/mistral/streaming.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/mistral/streaming.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/__init__.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/__init__.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/openai/__init__.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/__init__.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/client.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/client.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/openai/client.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/client.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/config.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/config.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/openai/config.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/config.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/models.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/models.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/openai/models.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/models.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/parameters.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/parameters.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/openai/parameters.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/parameters.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/streaming.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/streaming.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/openai/streaming.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/openai/streaming.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/xai/__init__.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/__init__.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/xai/__init__.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/__init__.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/xai/client.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/client.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/xai/client.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/client.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/xai/config.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/config.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/xai/config.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/config.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/xai/models.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/models.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/xai/models.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/models.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/xai/parameters.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/parameters.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/xai/parameters.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/parameters.py diff --git a/packages/text-generation/src/celeste_text_generation/providers/xai/streaming.py b/packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/streaming.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/providers/xai/streaming.py rename to packages/capabilities/text-generation/src/celeste_text_generation/providers/xai/streaming.py diff --git a/packages/text-generation/src/celeste_text_generation/py.typed b/packages/capabilities/text-generation/src/celeste_text_generation/py.typed similarity index 100% rename from packages/text-generation/src/celeste_text_generation/py.typed rename to packages/capabilities/text-generation/src/celeste_text_generation/py.typed diff --git a/packages/text-generation/src/celeste_text_generation/streaming.py b/packages/capabilities/text-generation/src/celeste_text_generation/streaming.py similarity index 100% rename from packages/text-generation/src/celeste_text_generation/streaming.py rename to packages/capabilities/text-generation/src/celeste_text_generation/streaming.py diff --git a/packages/text-generation/tests/integration_tests/conftest.py b/packages/capabilities/text-generation/tests/integration_tests/conftest.py similarity index 100% rename from packages/text-generation/tests/integration_tests/conftest.py rename to packages/capabilities/text-generation/tests/integration_tests/conftest.py diff --git a/packages/text-generation/tests/integration_tests/test_text_generation/__init__.py b/packages/capabilities/text-generation/tests/integration_tests/test_text_generation/__init__.py similarity index 100% rename from packages/text-generation/tests/integration_tests/test_text_generation/__init__.py rename to packages/capabilities/text-generation/tests/integration_tests/test_text_generation/__init__.py diff --git a/packages/text-generation/tests/integration_tests/test_text_generation/test_generate.py b/packages/capabilities/text-generation/tests/integration_tests/test_text_generation/test_generate.py similarity index 100% rename from packages/text-generation/tests/integration_tests/test_text_generation/test_generate.py rename to packages/capabilities/text-generation/tests/integration_tests/test_text_generation/test_generate.py diff --git a/packages/text-generation/tests/integration_tests/test_text_generation/test_stream.py b/packages/capabilities/text-generation/tests/integration_tests/test_text_generation/test_stream.py similarity index 100% rename from packages/text-generation/tests/integration_tests/test_text_generation/test_stream.py rename to packages/capabilities/text-generation/tests/integration_tests/test_text_generation/test_stream.py diff --git a/packages/video-generation/README.md b/packages/capabilities/video-generation/README.md similarity index 95% rename from packages/video-generation/README.md rename to packages/capabilities/video-generation/README.md index f465c7c6..276d63a9 100644 --- a/packages/video-generation/README.md +++ b/packages/capabilities/video-generation/README.md @@ -5,7 +5,7 @@ **Video Generation capability for Celeste AI** [![Python](https://img.shields.io/badge/Python-3.12+-blue?style=for-the-badge)](https://www.python.org/) -[![License](https://img.shields.io/badge/License-MIT-yellow?style=for-the-badge)](../../LICENSE) +[![License](https://img.shields.io/badge/License-MIT-yellow?style=for-the-badge)](../../../LICENSE) [Quick Start](#-quick-start) • [Documentation](https://withceleste.ai/docs) • [Request Provider](https://github.com/withceleste/celeste-python/issues/new) @@ -66,7 +66,7 @@ See [CONTRIBUTING.md](../../CONTRIBUTING.md) for guidelines. ## 📄 License -MIT License – see [LICENSE](../../LICENSE) for details. +MIT License – see [LICENSE](../../../LICENSE) for details. --- diff --git a/packages/video-generation/pyproject.toml b/packages/capabilities/video-generation/pyproject.toml similarity index 100% rename from packages/video-generation/pyproject.toml rename to packages/capabilities/video-generation/pyproject.toml diff --git a/packages/video-generation/src/celeste_video_generation/__init__.py b/packages/capabilities/video-generation/src/celeste_video_generation/__init__.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/__init__.py rename to packages/capabilities/video-generation/src/celeste_video_generation/__init__.py diff --git a/packages/video-generation/src/celeste_video_generation/client.py b/packages/capabilities/video-generation/src/celeste_video_generation/client.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/client.py rename to packages/capabilities/video-generation/src/celeste_video_generation/client.py diff --git a/packages/video-generation/src/celeste_video_generation/io.py b/packages/capabilities/video-generation/src/celeste_video_generation/io.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/io.py rename to packages/capabilities/video-generation/src/celeste_video_generation/io.py diff --git a/packages/video-generation/src/celeste_video_generation/models.py b/packages/capabilities/video-generation/src/celeste_video_generation/models.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/models.py rename to packages/capabilities/video-generation/src/celeste_video_generation/models.py diff --git a/packages/video-generation/src/celeste_video_generation/parameters.py b/packages/capabilities/video-generation/src/celeste_video_generation/parameters.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/parameters.py rename to packages/capabilities/video-generation/src/celeste_video_generation/parameters.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/__init__.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/__init__.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/__init__.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/__init__.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/bytedance/README.md b/packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/README.md similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/bytedance/README.md rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/README.md diff --git a/packages/video-generation/src/celeste_video_generation/providers/bytedance/__init__.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/__init__.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/bytedance/__init__.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/__init__.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/bytedance/client.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/client.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/bytedance/client.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/client.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/bytedance/config.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/config.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/bytedance/config.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/config.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/bytedance/models.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/models.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/bytedance/models.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/models.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/bytedance/parameters.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/parameters.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/bytedance/parameters.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/bytedance/parameters.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/google/__init__.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/google/__init__.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/google/__init__.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/google/__init__.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/google/client.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/google/client.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/google/client.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/google/client.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/google/config.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/google/config.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/google/config.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/google/config.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/google/models.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/google/models.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/google/models.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/google/models.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/google/parameters.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/google/parameters.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/google/parameters.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/google/parameters.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/openai/__init__.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/openai/__init__.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/openai/__init__.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/openai/__init__.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/openai/client.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/openai/client.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/openai/client.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/openai/client.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/openai/config.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/openai/config.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/openai/config.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/openai/config.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/openai/models.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/openai/models.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/openai/models.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/openai/models.py diff --git a/packages/video-generation/src/celeste_video_generation/providers/openai/parameters.py b/packages/capabilities/video-generation/src/celeste_video_generation/providers/openai/parameters.py similarity index 100% rename from packages/video-generation/src/celeste_video_generation/providers/openai/parameters.py rename to packages/capabilities/video-generation/src/celeste_video_generation/providers/openai/parameters.py diff --git a/packages/video-generation/src/celeste_video_generation/py.typed b/packages/capabilities/video-generation/src/celeste_video_generation/py.typed similarity index 100% rename from packages/video-generation/src/celeste_video_generation/py.typed rename to packages/capabilities/video-generation/src/celeste_video_generation/py.typed diff --git a/packages/video-generation/tests/integration_tests/test_video_generation/__init__.py b/packages/capabilities/video-generation/tests/integration_tests/test_video_generation/__init__.py similarity index 100% rename from packages/video-generation/tests/integration_tests/test_video_generation/__init__.py rename to packages/capabilities/video-generation/tests/integration_tests/test_video_generation/__init__.py diff --git a/packages/video-generation/tests/integration_tests/test_video_generation/test_generate.py b/packages/capabilities/video-generation/tests/integration_tests/test_video_generation/test_generate.py similarity index 100% rename from packages/video-generation/tests/integration_tests/test_video_generation/test_generate.py rename to packages/capabilities/video-generation/tests/integration_tests/test_video_generation/test_generate.py diff --git a/packages/speech-generation/tests/integration_tests/test_speech_generation/__init__.py b/packages/speech-generation/tests/integration_tests/test_speech_generation/__init__.py deleted file mode 100644 index e69de29b..00000000 diff --git a/pyproject.toml b/pyproject.toml index befe304e..6c3a5fef 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -59,7 +59,7 @@ dev = [ ] [tool.uv.workspace] -members = ["packages/*"] +members = ["packages/capabilities/*"] [tool.uv.sources] celeste-text-generation = { workspace = true } @@ -81,10 +81,10 @@ addopts = "-ra --strict-markers --strict-config" asyncio_mode = "auto" pythonpath = [ "src", - "packages/text-generation/src", - "packages/image-generation/src", - "packages/video-generation/src", - "packages/speech-generation/src", + "packages/capabilities/text-generation/src", + "packages/capabilities/image-generation/src", + "packages/capabilities/video-generation/src", + "packages/capabilities/speech-generation/src", ] markers = [ "slow: marks tests as slow (deselect with '-m \"not slow\"')", From eadb7e18a5485bd6936a29976665277ba54ea60f Mon Sep 17 00:00:00 2001 From: kamilbenkirane Date: Sun, 14 Dec 2025 19:36:24 +0100 Subject: [PATCH 2/4] fix: remove duplicate conftest.py in speech-generation tests --- .../speech-generation/tests/conftest.py | 24 ------------------- 1 file changed, 24 deletions(-) delete mode 100644 packages/capabilities/speech-generation/tests/conftest.py diff --git a/packages/capabilities/speech-generation/tests/conftest.py b/packages/capabilities/speech-generation/tests/conftest.py deleted file mode 100644 index 81c92564..00000000 --- a/packages/capabilities/speech-generation/tests/conftest.py +++ /dev/null @@ -1,24 +0,0 @@ -"""Pytest configuration and fixtures for integration tests.""" - -from collections.abc import AsyncGenerator -from typing import Any - -import pytest_asyncio - -from celeste.http import close_all_http_clients - - -@pytest_asyncio.fixture(autouse=True) -async def cleanup_http_clients() -> AsyncGenerator[None, None]: - """Ensure HTTP clients are closed after each test. - - This fixture runs automatically after each test to ensure HTTP clients - are properly closed before pytest-asyncio closes the event loop. - This prevents "Event loop is closed" errors when using pytest-xdist. - """ - yield - await close_all_http_clients() - - -def pytest_configure(config: Any) -> None: # noqa: ANN401 - config.addinivalue_line("markers", "integration: mark test as an integration test") From c5bfb5cbf0f15c1f7d02e67d5272cf2cb90d7104 Mon Sep 17 00:00:00 2001 From: kamilbenkirane Date: Sun, 14 Dec 2025 20:53:06 +0100 Subject: [PATCH 3/4] fix: update CI workflow paths for capabilities directory --- .github/workflows/ci.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index b9be8339..77bb453b 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -64,7 +64,7 @@ jobs: python-version: ${{ inputs.python-version || '3.12' }} - run: | if [ -d "packages" ]; then - uv run mypy -p celeste && uv run mypy tests/ && uv run mypy packages/image-generation packages/text-generation packages/video-generation packages/speech-generation + uv run mypy -p celeste && uv run mypy tests/ && uv run mypy packages/capabilities/image-generation packages/capabilities/text-generation packages/capabilities/video-generation packages/capabilities/speech-generation else uv run mypy -p celeste && uv run mypy tests/ fi From 2ff399ea1519427c8b95e8b31fdf9b8034b72b7a Mon Sep 17 00:00:00 2001 From: kamilbenkirane Date: Sun, 14 Dec 2025 20:56:01 +0100 Subject: [PATCH 4/4] fix: update publish workflow paths for capabilities directory --- .github/workflows/publish.yml | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml index d5b32704..ee21d5e2 100644 --- a/.github/workflows/publish.yml +++ b/.github/workflows/publish.yml @@ -69,13 +69,13 @@ jobs: base: ${{ steps.get-base.outputs.base }} filters: | text-generation: - - 'packages/text-generation/**' + - 'packages/capabilities/text-generation/**' image-generation: - - 'packages/image-generation/**' + - 'packages/capabilities/image-generation/**' video-generation: - - 'packages/video-generation/**' + - 'packages/capabilities/video-generation/**' speech-generation: - - 'packages/speech-generation/**' + - 'packages/capabilities/speech-generation/**' core: - 'src/celeste/**' - id: build-matrix @@ -136,7 +136,7 @@ jobs: BYTEDANCE_API_KEY: ${{ secrets.BYTEDANCE_API_KEY }} XAI_API_KEY: ${{ secrets.XAI_API_KEY }} ELEVENLABS_API_KEY: ${{ secrets.ELEVENLABS_API_KEY }} - run: uv run pytest packages/${{ matrix.package }}/tests/integration_tests/ -m integration -v + run: uv run pytest packages/capabilities/${{ matrix.package }}/tests/integration_tests/ -m integration -v build: needs: [validate-release, run-ci, detect-changes, integration-tests]