ai_api_unified 2.5.2__tar.gz → 2.5.4__tar.gz

{ai_api_unified-2.5.2 → ai_api_unified-2.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ai_api_unified
-Version: 2.5.2
+Version: 2.5.4
 Summary: Unified access layer for AI completions (LLM), embedding (semantic), image, video, and voice services
 License: MIT
 Keywords: llm,openai,bedrock,gemini,vertex,embeddings,completion,video,voice,api

{ai_api_unified-2.5.2 → ai_api_unified-2.5.4}/pyproject.toml

@@ -36,7 +36,7 @@
 [project]
 name = "ai_api_unified"
 
-version = "2.5.2" # keep in sync with src/ai_api_unified/__version__.py
+version = "2.5.4" # keep in sync with src/ai_api_unified/__version__.py
 description = "Unified access layer for AI completions (LLM), embedding (semantic), image, video, and voice services"
 authors = [{ name = "Dave Thomas", email = "davidcthomas@gmail.com" }]
 license = "MIT"

{ai_api_unified-2.5.2 → ai_api_unified-2.5.4}/src/ai_api_unified/__version__.py

@@ -9,4 +9,4 @@ from __future__ import annotations
 
 __all__: list[str] = ["__version__"]
 
-__version__: str = "2.5.2"
+__version__: str = "2.5.4"

{ai_api_unified-2.5.2 → ai_api_unified-2.5.4}/src/ai_api_unified/videos/ai_google_gemini_videos.py

@@ -40,42 +40,66 @@ class AIGoogleGeminiVideoProperties(AIBaseVideoProperties):
     last_frame_image: AIMediaReference | None = None
     person_generation: str | None = None
     generate_audio: bool | None = None
+    negative_prompt: str | None = None
+    enhance_prompt: bool | None = None
+    output_gcs_uri: str | None = None
+    compression_quality: str | None = None
 
     _ALLOWED_ASPECT_RATIOS: ClassVar[set[str]] = {"16:9", "9:16"}
-    _ALLOWED_RESOLUTIONS: ClassVar[set[str]] = {"720p", "1080p", "4k"}
+    _ALLOWED_RESOLUTIONS: ClassVar[set[str]] = {"720p", "1080p"}
     _ALLOWED_PERSON_GENERATION: ClassVar[set[str]] = {
         "dont_allow",
         "allow_adult",
-        "allow_all",
     }
+    _ALLOWED_COMPRESSION_QUALITY: ClassVar[set[str]] = {"OPTIMIZED", "LOSSLESS"}
 
     @model_validator(mode="after")
     def _validate_google_video_properties(self) -> "AIGoogleGeminiVideoProperties":
         if self.fps is not None:
             raise ValueError(
-                "Google Gemini (Veo) does not support the 'fps' parameter. "
-                "Remove fps from your video properties for this provider."
+                f"Google Gemini (Veo) does not support the 'fps' parameter "
+                f"(received fps={self.fps!r}). Remove fps from your video "
+                f"properties for this provider."
             )
         if (
             self.aspect_ratio is not None
             and self.aspect_ratio not in self._ALLOWED_ASPECT_RATIOS
         ):
             raise ValueError(
-                "Google Gemini video aspect_ratio must be '16:9' or '9:16'."
+                f"Google Gemini video aspect_ratio={self.aspect_ratio!r} is "
+                f"not supported. Must be one of '16:9' or '9:16'."
             )
         if (
             self.resolution is not None
             and self.resolution not in self._ALLOWED_RESOLUTIONS
         ):
             raise ValueError(
-                "Google Gemini video resolution must be one of 720p, 1080p, or 4k."
+                f"Google Gemini video resolution={self.resolution!r} is not "
+                f"supported. Must be one of 720p or 1080p."
             )
         if (
             self.person_generation is not None
             and self.person_generation not in self._ALLOWED_PERSON_GENERATION
         ):
             raise ValueError(
-                "Google Gemini person_generation must be one of dont_allow, allow_adult, or allow_all."
+                f"Google Gemini person_generation={self.person_generation!r} "
+                f"is not supported. Must be one of dont_allow or allow_adult."
+            )
+        if self.compression_quality is not None:
+            normalized: str = self.compression_quality.upper()
+            if normalized not in self._ALLOWED_COMPRESSION_QUALITY:
+                raise ValueError(
+                    f"Google Gemini compression_quality="
+                    f"{self.compression_quality!r} is not supported. "
+                    f"Must be 'OPTIMIZED' or 'LOSSLESS'."
+                )
+            self.compression_quality = normalized
+        if self.output_gcs_uri is not None and not self.output_gcs_uri.startswith(
+            "gs://"
+        ):
+            raise ValueError(
+                f"Google Gemini output_gcs_uri={self.output_gcs_uri!r} is "
+                f"not supported. Must be a gs:// URI."
             )
         return self
 
@@ -372,6 +396,16 @@ class AIGoogleGeminiVideos(AIGoogleBase, AIBaseVideos):
             config_kwargs["generate_audio"] = google_properties.generate_audio
         if google_properties.person_generation is not None:
             config_kwargs["person_generation"] = google_properties.person_generation
+        if google_properties.negative_prompt is not None:
+            config_kwargs["negative_prompt"] = google_properties.negative_prompt
+        if google_properties.enhance_prompt is not None:
+            config_kwargs["enhance_prompt"] = google_properties.enhance_prompt
+        if google_properties.output_gcs_uri is not None:
+            config_kwargs["output_gcs_uri"] = google_properties.output_gcs_uri
+        if google_properties.compression_quality is not None:
+            config_kwargs["compression_quality"] = types.VideoCompressionQuality(
+                google_properties.compression_quality
+            )
         if google_properties.last_frame_image is not None:
             config_kwargs["last_frame"] = self._to_google_image(
                 google_properties.last_frame_image