From fedd80b826516e67eabc621eabc6926bc0729a21 Mon Sep 17 00:00:00 2001 From: kamilbenkirane Date: Mon, 17 Nov 2025 17:54:00 +0100 Subject: [PATCH] Standardize provider docstrings with capability suffixes - Update all provider __init__.py files: '{Provider} provider for {capability}.' - Update all provider client.py files: '{Provider} client implementation for {capability}.' - Update all provider config.py files: '{Provider} provider configuration for {capability}.' - Update all provider parameters.py files: '{Provider} parameter mappers for {capability}.' - Update all provider models.py files: '{Provider} models for {capability}.' - Remove 'Gemini' from Google docstrings for consistency This ensures consistent docstring patterns across all provider files to avoid ambiguity when the same provider appears in multiple packages. --- .../providers/bytedance/__init__.py | 2 +- .../providers/bytedance/client.py | 2 +- .../providers/bytedance/config.py | 2 +- .../providers/bytedance/models.py | 2 +- .../providers/bytedance/parameters.py | 2 +- .../providers/google/__init__.py | 2 +- .../providers/google/client.py | 2 +- .../providers/google/config.py | 2 +- .../providers/google/models.py | 2 +- .../providers/google/parameters.py | 2 +- .../providers/openai/__init__.py | 2 +- .../providers/openai/client.py | 2 +- .../providers/openai/config.py | 2 +- .../providers/openai/models.py | 2 +- .../providers/openai/parameters.py | 2 +- .../providers/anthropic/__init__.py | 2 +- .../providers/anthropic/client.py | 2 +- .../providers/anthropic/config.py | 2 +- .../providers/anthropic/models.py | 2 +- .../providers/anthropic/parameters.py | 2 +- .../providers/cohere/__init__.py | 2 +- .../providers/cohere/client.py | 2 +- .../providers/cohere/config.py | 2 +- .../providers/cohere/models.py | 2 +- .../providers/cohere/parameters.py | 2 +- .../providers/google/__init__.py | 2 +- .../providers/google/client.py | 2 +- .../providers/google/config.py | 2 +- .../providers/google/models.py | 2 +- .../providers/google/parameters.py | 2 +- .../providers/mistral/__init__.py | 2 +- .../providers/mistral/client.py | 2 +- .../providers/mistral/config.py | 2 +- .../providers/mistral/models.py | 2 +- .../providers/mistral/parameters.py | 2 +- .../providers/openai/__init__.py | 2 +- .../providers/openai/client.py | 2 +- .../providers/openai/config.py | 2 +- .../providers/openai/models.py | 2 +- .../providers/openai/parameters.py | 2 +- .../src/celeste_video_generation/client.py | 13 ++++++++++--- .../providers/bytedance/config.py | 2 +- .../providers/bytedance/parameters.py | 2 +- .../providers/google/config.py | 2 +- .../providers/google/models.py | 2 +- .../providers/google/parameters.py | 2 +- .../providers/openai/client.py | 2 +- .../providers/openai/config.py | 2 +- .../providers/openai/models.py | 2 +- .../providers/openai/parameters.py | 2 +- 50 files changed, 59 insertions(+), 52 deletions(-) diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/__init__.py b/packages/image-generation/src/celeste_image_generation/providers/bytedance/__init__.py index 10769780..623b1947 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/bytedance/__init__.py +++ b/packages/image-generation/src/celeste_image_generation/providers/bytedance/__init__.py @@ -1,4 +1,4 @@ -"""ByteDance provider.""" +"""ByteDance provider for image generation.""" from .client import ByteDanceImageGenerationClient from .models import MODELS diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/client.py b/packages/image-generation/src/celeste_image_generation/providers/bytedance/client.py index dc85857e..f86ced40 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/bytedance/client.py +++ b/packages/image-generation/src/celeste_image_generation/providers/bytedance/client.py @@ -1,4 +1,4 @@ -"""ByteDance client implementation.""" +"""ByteDance client implementation for image generation.""" import base64 from collections.abc import AsyncIterator diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/config.py b/packages/image-generation/src/celeste_image_generation/providers/bytedance/config.py index 7554ddda..f7a40b82 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/bytedance/config.py +++ b/packages/image-generation/src/celeste_image_generation/providers/bytedance/config.py @@ -1,4 +1,4 @@ -"""ByteDance provider configuration.""" +"""ByteDance provider configuration for image generation.""" # HTTP Configuration BASE_URL = "https://ark.ap-southeast.bytepluses.com/api/v3" diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/models.py b/packages/image-generation/src/celeste_image_generation/providers/bytedance/models.py index 52495259..6273ccba 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/bytedance/models.py +++ b/packages/image-generation/src/celeste_image_generation/providers/bytedance/models.py @@ -1,4 +1,4 @@ -"""ByteDance models.""" +"""ByteDance models for image generation.""" from celeste import Model, Provider from celeste.constraints import Bool, Choice diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/parameters.py b/packages/image-generation/src/celeste_image_generation/providers/bytedance/parameters.py index f6e145fc..310e24ce 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/bytedance/parameters.py +++ b/packages/image-generation/src/celeste_image_generation/providers/bytedance/parameters.py @@ -1,4 +1,4 @@ -"""ByteDance parameter mappers.""" +"""ByteDance parameter mappers for image generation.""" from typing import Any diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/__init__.py b/packages/image-generation/src/celeste_image_generation/providers/google/__init__.py index 39c5b9be..992b1691 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/google/__init__.py +++ b/packages/image-generation/src/celeste_image_generation/providers/google/__init__.py @@ -1,4 +1,4 @@ -"""Google provider.""" +"""Google provider for image generation.""" from .client import GoogleImageGenerationClient from .models import MODELS diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/client.py b/packages/image-generation/src/celeste_image_generation/providers/google/client.py index 2440c0e9..ab5a40a0 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/google/client.py +++ b/packages/image-generation/src/celeste_image_generation/providers/google/client.py @@ -1,4 +1,4 @@ -"""Google client implementation.""" +"""Google client implementation for image generation.""" import base64 from typing import Any, Unpack diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/config.py b/packages/image-generation/src/celeste_image_generation/providers/google/config.py index 83901dfd..c1cb4d28 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/google/config.py +++ b/packages/image-generation/src/celeste_image_generation/providers/google/config.py @@ -1,4 +1,4 @@ -"""Google provider configuration.""" +"""Google provider configuration for image generation.""" # HTTP Configuration BASE_URL = "https://generativelanguage.googleapis.com/v1beta/models" diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/models.py b/packages/image-generation/src/celeste_image_generation/providers/google/models.py index 17305299..5cf45b13 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/google/models.py +++ b/packages/image-generation/src/celeste_image_generation/providers/google/models.py @@ -1,4 +1,4 @@ -"""Google models.""" +"""Google models for image generation.""" from celeste import Model, Provider from celeste.constraints import Choice diff --git a/packages/image-generation/src/celeste_image_generation/providers/google/parameters.py b/packages/image-generation/src/celeste_image_generation/providers/google/parameters.py index a7730857..b8201192 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/google/parameters.py +++ b/packages/image-generation/src/celeste_image_generation/providers/google/parameters.py @@ -1,4 +1,4 @@ -"""Google parameter mappers.""" +"""Google parameter mappers for image generation.""" from typing import Any diff --git a/packages/image-generation/src/celeste_image_generation/providers/openai/__init__.py b/packages/image-generation/src/celeste_image_generation/providers/openai/__init__.py index 29b11cb0..96e62b5a 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/openai/__init__.py +++ b/packages/image-generation/src/celeste_image_generation/providers/openai/__init__.py @@ -1,4 +1,4 @@ -"""OpenAI provider.""" +"""OpenAI provider for image generation.""" from .client import OpenAIImageGenerationClient from .models import MODELS diff --git a/packages/image-generation/src/celeste_image_generation/providers/openai/client.py b/packages/image-generation/src/celeste_image_generation/providers/openai/client.py index 73571eff..ea204c9c 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/openai/client.py +++ b/packages/image-generation/src/celeste_image_generation/providers/openai/client.py @@ -1,4 +1,4 @@ -"""OpenAI client implementation.""" +"""OpenAI client implementation for image generation.""" import base64 from collections.abc import AsyncIterator diff --git a/packages/image-generation/src/celeste_image_generation/providers/openai/config.py b/packages/image-generation/src/celeste_image_generation/providers/openai/config.py index 0af0f9eb..195012eb 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/openai/config.py +++ b/packages/image-generation/src/celeste_image_generation/providers/openai/config.py @@ -1,4 +1,4 @@ -"""OpenAI provider configuration.""" +"""OpenAI provider configuration for image generation.""" # HTTP Configuration BASE_URL = "https://api.openai.com" diff --git a/packages/image-generation/src/celeste_image_generation/providers/openai/models.py b/packages/image-generation/src/celeste_image_generation/providers/openai/models.py index 90e9ddd5..6285b3c8 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/openai/models.py +++ b/packages/image-generation/src/celeste_image_generation/providers/openai/models.py @@ -1,4 +1,4 @@ -"""OpenAI models.""" +"""OpenAI models for image generation.""" from celeste import Model, Provider from celeste.constraints import Choice, Range diff --git a/packages/image-generation/src/celeste_image_generation/providers/openai/parameters.py b/packages/image-generation/src/celeste_image_generation/providers/openai/parameters.py index 1db72024..0bfa602e 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/openai/parameters.py +++ b/packages/image-generation/src/celeste_image_generation/providers/openai/parameters.py @@ -1,4 +1,4 @@ -"""OpenAI parameter mappers.""" +"""OpenAI parameter mappers for image generation.""" from typing import Any diff --git a/packages/text-generation/src/celeste_text_generation/providers/anthropic/__init__.py b/packages/text-generation/src/celeste_text_generation/providers/anthropic/__init__.py index 62028733..f1c0bd84 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/anthropic/__init__.py +++ b/packages/text-generation/src/celeste_text_generation/providers/anthropic/__init__.py @@ -1,4 +1,4 @@ -"""Anthropic provider.""" +"""Anthropic provider for text generation.""" from .client import AnthropicTextGenerationClient from .models import MODELS diff --git a/packages/text-generation/src/celeste_text_generation/providers/anthropic/client.py b/packages/text-generation/src/celeste_text_generation/providers/anthropic/client.py index 153e5ef0..45d957db 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/anthropic/client.py +++ b/packages/text-generation/src/celeste_text_generation/providers/anthropic/client.py @@ -1,4 +1,4 @@ -"""Anthropic client implementation.""" +"""Anthropic client implementation for text generation.""" from collections.abc import AsyncIterator from typing import Any, Unpack diff --git a/packages/text-generation/src/celeste_text_generation/providers/anthropic/config.py b/packages/text-generation/src/celeste_text_generation/providers/anthropic/config.py index a46aff12..a4437062 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/anthropic/config.py +++ b/packages/text-generation/src/celeste_text_generation/providers/anthropic/config.py @@ -1,4 +1,4 @@ -"""Anthropic provider configuration.""" +"""Anthropic provider configuration for text generation.""" # HTTP Configuration BASE_URL = "https://api.anthropic.com" diff --git a/packages/text-generation/src/celeste_text_generation/providers/anthropic/models.py b/packages/text-generation/src/celeste_text_generation/providers/anthropic/models.py index ba7ed7fe..76d2690e 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/anthropic/models.py +++ b/packages/text-generation/src/celeste_text_generation/providers/anthropic/models.py @@ -1,4 +1,4 @@ -"""Anthropic models.""" +"""Anthropic models for text generation.""" from celeste import Model, Provider from celeste.constraints import Range, Schema diff --git a/packages/text-generation/src/celeste_text_generation/providers/anthropic/parameters.py b/packages/text-generation/src/celeste_text_generation/providers/anthropic/parameters.py index 25bc1f7a..80d3437f 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/anthropic/parameters.py +++ b/packages/text-generation/src/celeste_text_generation/providers/anthropic/parameters.py @@ -1,4 +1,4 @@ -"""Anthropic parameter mappers.""" +"""Anthropic parameter mappers for text generation.""" import json from typing import Any, get_args, get_origin diff --git a/packages/text-generation/src/celeste_text_generation/providers/cohere/__init__.py b/packages/text-generation/src/celeste_text_generation/providers/cohere/__init__.py index 1711928e..9d5a91c2 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/cohere/__init__.py +++ b/packages/text-generation/src/celeste_text_generation/providers/cohere/__init__.py @@ -1,4 +1,4 @@ -"""Cohere provider.""" +"""Cohere provider for text generation.""" from .client import CohereTextGenerationClient from .models import MODELS diff --git a/packages/text-generation/src/celeste_text_generation/providers/cohere/client.py b/packages/text-generation/src/celeste_text_generation/providers/cohere/client.py index c623755a..5c7cf88a 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/cohere/client.py +++ b/packages/text-generation/src/celeste_text_generation/providers/cohere/client.py @@ -1,4 +1,4 @@ -"""Cohere client implementation.""" +"""Cohere client implementation for text generation.""" from collections.abc import AsyncIterator from typing import Any, Unpack diff --git a/packages/text-generation/src/celeste_text_generation/providers/cohere/config.py b/packages/text-generation/src/celeste_text_generation/providers/cohere/config.py index ea0000f0..7e2e967d 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/cohere/config.py +++ b/packages/text-generation/src/celeste_text_generation/providers/cohere/config.py @@ -1,4 +1,4 @@ -"""Cohere provider configuration.""" +"""Cohere provider configuration for text generation.""" # HTTP Configuration BASE_URL = "https://api.cohere.com" diff --git a/packages/text-generation/src/celeste_text_generation/providers/cohere/models.py b/packages/text-generation/src/celeste_text_generation/providers/cohere/models.py index f3f9cba9..38f25cbc 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/cohere/models.py +++ b/packages/text-generation/src/celeste_text_generation/providers/cohere/models.py @@ -1,4 +1,4 @@ -"""Cohere models.""" +"""Cohere models for text generation.""" from celeste import Model, Provider from celeste.constraints import Range, Schema diff --git a/packages/text-generation/src/celeste_text_generation/providers/cohere/parameters.py b/packages/text-generation/src/celeste_text_generation/providers/cohere/parameters.py index 58b64885..ddab9333 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/cohere/parameters.py +++ b/packages/text-generation/src/celeste_text_generation/providers/cohere/parameters.py @@ -1,4 +1,4 @@ -"""Cohere parameter mappers.""" +"""Cohere parameter mappers for text generation.""" import json from typing import Any, get_args, get_origin diff --git a/packages/text-generation/src/celeste_text_generation/providers/google/__init__.py b/packages/text-generation/src/celeste_text_generation/providers/google/__init__.py index 1b760d52..c6db4d61 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/google/__init__.py +++ b/packages/text-generation/src/celeste_text_generation/providers/google/__init__.py @@ -1,4 +1,4 @@ -"""Google provider.""" +"""Google provider for text generation.""" from .client import GoogleTextGenerationClient from .models import MODELS diff --git a/packages/text-generation/src/celeste_text_generation/providers/google/client.py b/packages/text-generation/src/celeste_text_generation/providers/google/client.py index 45548189..044dc073 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/google/client.py +++ b/packages/text-generation/src/celeste_text_generation/providers/google/client.py @@ -1,4 +1,4 @@ -"""Google client implementation.""" +"""Google client implementation for text generation.""" from collections.abc import AsyncIterator from typing import Any, Unpack diff --git a/packages/text-generation/src/celeste_text_generation/providers/google/config.py b/packages/text-generation/src/celeste_text_generation/providers/google/config.py index 86057e2b..cd20a29e 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/google/config.py +++ b/packages/text-generation/src/celeste_text_generation/providers/google/config.py @@ -1,4 +1,4 @@ -"""Google provider configuration.""" +"""Google provider configuration for text generation.""" # HTTP Configuration BASE_URL = "https://generativelanguage.googleapis.com" diff --git a/packages/text-generation/src/celeste_text_generation/providers/google/models.py b/packages/text-generation/src/celeste_text_generation/providers/google/models.py index 89f8c0de..ce338b7b 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/google/models.py +++ b/packages/text-generation/src/celeste_text_generation/providers/google/models.py @@ -1,4 +1,4 @@ -"""Google models.""" +"""Google models for text generation.""" from celeste import Model, Provider from celeste.constraints import Range, Schema diff --git a/packages/text-generation/src/celeste_text_generation/providers/google/parameters.py b/packages/text-generation/src/celeste_text_generation/providers/google/parameters.py index 8c71424a..63c5112f 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/google/parameters.py +++ b/packages/text-generation/src/celeste_text_generation/providers/google/parameters.py @@ -1,4 +1,4 @@ -"""Google parameter mappers.""" +"""Google parameter mappers for text generation.""" from typing import Any, get_args, get_origin diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/__init__.py b/packages/text-generation/src/celeste_text_generation/providers/mistral/__init__.py index d77eaf55..2f8e8893 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/mistral/__init__.py +++ b/packages/text-generation/src/celeste_text_generation/providers/mistral/__init__.py @@ -1,4 +1,4 @@ -"""Mistral provider.""" +"""Mistral provider for text generation.""" from .client import MistralTextGenerationClient from .models import MODELS diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/client.py b/packages/text-generation/src/celeste_text_generation/providers/mistral/client.py index 2ed37f46..bd2bae54 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/mistral/client.py +++ b/packages/text-generation/src/celeste_text_generation/providers/mistral/client.py @@ -1,4 +1,4 @@ -"""Mistral client implementation.""" +"""Mistral client implementation for text generation.""" from collections.abc import AsyncIterator from typing import Any, Unpack diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/config.py b/packages/text-generation/src/celeste_text_generation/providers/mistral/config.py index aa9d24d5..e889c537 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/mistral/config.py +++ b/packages/text-generation/src/celeste_text_generation/providers/mistral/config.py @@ -1,4 +1,4 @@ -"""Mistral provider configuration.""" +"""Mistral provider configuration for text generation.""" # HTTP Configuration BASE_URL = "https://api.mistral.ai" diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py b/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py index 10fffc73..57ad1447 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py +++ b/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py @@ -1,4 +1,4 @@ -"""Mistral models.""" +"""Mistral models for text generation.""" from celeste import Model, Provider from celeste.constraints import Range, Schema diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/parameters.py b/packages/text-generation/src/celeste_text_generation/providers/mistral/parameters.py index e69b3fa2..918f024b 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/mistral/parameters.py +++ b/packages/text-generation/src/celeste_text_generation/providers/mistral/parameters.py @@ -1,4 +1,4 @@ -"""Mistral parameter mappers.""" +"""Mistral parameter mappers for text generation.""" from typing import Any, get_args, get_origin diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/__init__.py b/packages/text-generation/src/celeste_text_generation/providers/openai/__init__.py index be7016aa..6e0b6d99 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/openai/__init__.py +++ b/packages/text-generation/src/celeste_text_generation/providers/openai/__init__.py @@ -1,4 +1,4 @@ -"""OpenAI provider.""" +"""OpenAI provider for text generation.""" from .client import OpenAITextGenerationClient from .models import MODELS diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/client.py b/packages/text-generation/src/celeste_text_generation/providers/openai/client.py index 2ed0c997..81dba849 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/openai/client.py +++ b/packages/text-generation/src/celeste_text_generation/providers/openai/client.py @@ -1,4 +1,4 @@ -"""OpenAI client implementation.""" +"""OpenAI client implementation for text generation.""" from collections.abc import AsyncIterator from typing import Any, Unpack diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/config.py b/packages/text-generation/src/celeste_text_generation/providers/openai/config.py index f6f2a625..8c72f1d4 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/openai/config.py +++ b/packages/text-generation/src/celeste_text_generation/providers/openai/config.py @@ -1,4 +1,4 @@ -"""OpenAI provider configuration.""" +"""OpenAI provider configuration for text generation.""" # HTTP Configuration BASE_URL = "https://api.openai.com" diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/models.py b/packages/text-generation/src/celeste_text_generation/providers/openai/models.py index ed0b5247..76111a5b 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/openai/models.py +++ b/packages/text-generation/src/celeste_text_generation/providers/openai/models.py @@ -1,4 +1,4 @@ -"""OpenAI models.""" +"""OpenAI models for text generation.""" from celeste import Model, Provider from celeste.constraints import Choice, Range, Schema diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/parameters.py b/packages/text-generation/src/celeste_text_generation/providers/openai/parameters.py index b7361a7a..06a32549 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/openai/parameters.py +++ b/packages/text-generation/src/celeste_text_generation/providers/openai/parameters.py @@ -1,4 +1,4 @@ -"""OpenAI parameter mappers.""" +"""OpenAI parameter mappers for text generation.""" import json from typing import Any, get_args, get_origin diff --git a/packages/video-generation/src/celeste_video_generation/client.py b/packages/video-generation/src/celeste_video_generation/client.py index 89fc8bb9..680c2329 100644 --- a/packages/video-generation/src/celeste_video_generation/client.py +++ b/packages/video-generation/src/celeste_video_generation/client.py @@ -7,6 +7,7 @@ from celeste.artifacts import VideoArtifact from celeste.client import Client +from celeste.exceptions import ValidationError from celeste_video_generation.io import ( VideoGenerationInput, VideoGenerationOutput, @@ -37,11 +38,17 @@ def _parse_content( """Parse content from provider response.""" def _create_inputs( - self, - prompt: str, - **parameters: Unpack[VideoGenerationParameters], + self, *args: str, **parameters: Unpack[VideoGenerationParameters] ) -> VideoGenerationInput: """Map positional arguments to Input type.""" + if args: + return VideoGenerationInput(prompt=args[0]) + prompt: str | None = parameters.get("prompt") + if prompt is None: + msg = ( + "prompt is required (either as positional argument or keyword argument)" + ) + raise ValidationError(msg) return VideoGenerationInput(prompt=prompt) @classmethod diff --git a/packages/video-generation/src/celeste_video_generation/providers/bytedance/config.py b/packages/video-generation/src/celeste_video_generation/providers/bytedance/config.py index 2fabcca9..099d7e00 100644 --- a/packages/video-generation/src/celeste_video_generation/providers/bytedance/config.py +++ b/packages/video-generation/src/celeste_video_generation/providers/bytedance/config.py @@ -1,4 +1,4 @@ -"""ByteDance provider configuration.""" +"""ByteDance provider configuration for video generation.""" # HTTP Configuration BASE_URL = "https://ark.ap-southeast.bytepluses.com/api/v3" diff --git a/packages/video-generation/src/celeste_video_generation/providers/bytedance/parameters.py b/packages/video-generation/src/celeste_video_generation/providers/bytedance/parameters.py index 5ee19a8f..e84e3b07 100644 --- a/packages/video-generation/src/celeste_video_generation/providers/bytedance/parameters.py +++ b/packages/video-generation/src/celeste_video_generation/providers/bytedance/parameters.py @@ -1,4 +1,4 @@ -"""ByteDance parameter mappers.""" +"""ByteDance parameter mappers for video generation.""" from typing import Any diff --git a/packages/video-generation/src/celeste_video_generation/providers/google/config.py b/packages/video-generation/src/celeste_video_generation/providers/google/config.py index a6828d9d..29f53b61 100644 --- a/packages/video-generation/src/celeste_video_generation/providers/google/config.py +++ b/packages/video-generation/src/celeste_video_generation/providers/google/config.py @@ -1,4 +1,4 @@ -"""Google provider configuration.""" +"""Google provider configuration for video generation.""" BASE_URL = "https://generativelanguage.googleapis.com/v1beta" GENERATE_ENDPOINT = "/models/{model_id}:predictLongRunning" diff --git a/packages/video-generation/src/celeste_video_generation/providers/google/models.py b/packages/video-generation/src/celeste_video_generation/providers/google/models.py index c2023ae9..857ebb2d 100644 --- a/packages/video-generation/src/celeste_video_generation/providers/google/models.py +++ b/packages/video-generation/src/celeste_video_generation/providers/google/models.py @@ -1,4 +1,4 @@ -"""Google models.""" +"""Google models for video generation.""" from celeste import Model, Provider from celeste.constraints import Choice, ImageConstraint, ImagesConstraint diff --git a/packages/video-generation/src/celeste_video_generation/providers/google/parameters.py b/packages/video-generation/src/celeste_video_generation/providers/google/parameters.py index 48c17d77..7fb63a6d 100644 --- a/packages/video-generation/src/celeste_video_generation/providers/google/parameters.py +++ b/packages/video-generation/src/celeste_video_generation/providers/google/parameters.py @@ -1,4 +1,4 @@ -"""Google parameter mappers.""" +"""Google parameter mappers for video generation.""" from typing import Any diff --git a/packages/video-generation/src/celeste_video_generation/providers/openai/client.py b/packages/video-generation/src/celeste_video_generation/providers/openai/client.py index 48700b67..b9208533 100644 --- a/packages/video-generation/src/celeste_video_generation/providers/openai/client.py +++ b/packages/video-generation/src/celeste_video_generation/providers/openai/client.py @@ -1,4 +1,4 @@ -"""OpenAI client implementation.""" +"""OpenAI client implementation for video generation.""" import asyncio import base64 diff --git a/packages/video-generation/src/celeste_video_generation/providers/openai/config.py b/packages/video-generation/src/celeste_video_generation/providers/openai/config.py index eaed3566..e45fe93e 100644 --- a/packages/video-generation/src/celeste_video_generation/providers/openai/config.py +++ b/packages/video-generation/src/celeste_video_generation/providers/openai/config.py @@ -1,4 +1,4 @@ -"""OpenAI provider configuration.""" +"""OpenAI provider configuration for video generation.""" # HTTP Configuration BASE_URL = "https://api.openai.com" diff --git a/packages/video-generation/src/celeste_video_generation/providers/openai/models.py b/packages/video-generation/src/celeste_video_generation/providers/openai/models.py index 714899ab..f1063e8e 100644 --- a/packages/video-generation/src/celeste_video_generation/providers/openai/models.py +++ b/packages/video-generation/src/celeste_video_generation/providers/openai/models.py @@ -1,4 +1,4 @@ -"""OpenAI models.""" +"""OpenAI models for video generation.""" from celeste import Model, Provider from celeste.constraints import Choice, ImageConstraint diff --git a/packages/video-generation/src/celeste_video_generation/providers/openai/parameters.py b/packages/video-generation/src/celeste_video_generation/providers/openai/parameters.py index f9578192..83a8a678 100644 --- a/packages/video-generation/src/celeste_video_generation/providers/openai/parameters.py +++ b/packages/video-generation/src/celeste_video_generation/providers/openai/parameters.py @@ -1,4 +1,4 @@ -"""OpenAI parameter mappers.""" +"""OpenAI parameter mappers for video generation.""" from typing import Any