timbal 2.2.0__tar.gz → 2.2.2__tar.gz

{timbal-2.2.0 → timbal-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: timbal
-Version: 2.2.0
+Version: 2.2.2
 Summary: Simple, performant, battle-tested framework for building reliable AI applications
 Project-URL: Source, https://github.com/timbal-ai/timbal
 Author-email: Timbal <team@timbal.ai>

{timbal-2.2.0 → timbal-2.2.2}/python/timbal/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '2.2.0'
-__version_tuple__ = version_tuple = (2, 2, 0)
+__version__ = version = '2.2.2'
+__version_tuple__ = version_tuple = (2, 2, 2)
 
 __commit_id__ = commit_id = None

{timbal-2.2.0 → timbal-2.2.2}/python/timbal/codegen/flow.py

@@ -71,8 +71,20 @@ def _extract_key_from_lambda(fn: Any, source_step: str) -> str | None:
 
 
 def _is_model_enum(values: list) -> bool:
-    """Return True if a list of enum values looks like LLM model IDs (provider/name)."""
-    return len(values) > 5 and all(isinstance(v, str) and "/" in v for v in values[:5])
+    """Return True if an enum is the canonical LLM model ID set (provider/name).
+
+    Matches against ``timbal.core.models.Model`` rather than guessing from shape:
+    other registries (e.g. Krea's image/video IDs) share the ``provider/name``
+    form but must keep their enum inline so editors resolve the right options.
+    """
+    if len(values) <= 5 or not all(isinstance(v, str) and "/" in v for v in values[:5]):
+        return False
+    import typing
+
+    from timbal.core.models import Model
+
+    known = set(typing.get_args(Model))
+    return all(v in known for v in values)
 
 
 def _compact_field(field: dict) -> dict:

{timbal-2.2.0 → timbal-2.2.2}/python/timbal/platform/types.py

@@ -34,7 +34,6 @@ class UploadFileResponse(BaseModel):
     created_at: datetime
     expires_at: datetime | None
     name: str
-    # Optional: POST /files (no org scope) returns a temporary upload without an id.
-    # POST /orgs/{org_id}/files returns a permanent upload with an id.
+    # POST /files (no org scope) returns a temporary upload without an id.
     id: str | int | None = None
     url: str

{timbal-2.2.0 → timbal-2.2.2}/python/timbal/tools/__init__.py

@@ -238,6 +238,22 @@ if TYPE_CHECKING:
         HappyScribeListTranscriptions,
         HappyScribeUploadFile,
     )
+    from .higgsfield import (
+        HiggsfieldCancelRequest,
+        HiggsfieldCheckStatus,
+        HiggsfieldExpandImage,
+        HiggsfieldGetResult,
+        HiggsfieldImageToImage,
+        HiggsfieldImageToVideo,
+        HiggsfieldLipsync,
+        HiggsfieldRemoveBackground,
+        HiggsfieldSoulGenerate,
+        HiggsfieldSoulTrain,
+        HiggsfieldSubmit,
+        HiggsfieldTextToImage,
+        HiggsfieldTextToVideo,
+        HiggsfieldUploadFile,
+    )
     from .hubspot import (
         HubSpotCreateAssociation,
         HubSpotCreateCompany,
@@ -1969,6 +1985,20 @@ __all__ = [
     "HappyScribeListStyleGuides",
     "HappyScribeListTranscriptions",
     "HappyScribeUploadFile",
+    "HiggsfieldCancelRequest",
+    "HiggsfieldCheckStatus",
+    "HiggsfieldExpandImage",
+    "HiggsfieldGetResult",
+    "HiggsfieldImageToImage",
+    "HiggsfieldImageToVideo",
+    "HiggsfieldLipsync",
+    "HiggsfieldRemoveBackground",
+    "HiggsfieldSoulGenerate",
+    "HiggsfieldSoulTrain",
+    "HiggsfieldSubmit",
+    "HiggsfieldTextToImage",
+    "HiggsfieldTextToVideo",
+    "HiggsfieldUploadFile",
     "ElasticsearchIngestAttachment",
     "ElasticsearchBulkOperations",
     "ElasticsearchDeleteDocument",
@@ -3014,6 +3044,20 @@ _LAZY_IMPORTS = {
     "HappyScribeListStyleGuides": ".happy_scribe",
     "HappyScribeListTranscriptions": ".happy_scribe",
     "HappyScribeUploadFile": ".happy_scribe",
+    "HiggsfieldCancelRequest": ".higgsfield",
+    "HiggsfieldCheckStatus": ".higgsfield",
+    "HiggsfieldExpandImage": ".higgsfield",
+    "HiggsfieldGetResult": ".higgsfield",
+    "HiggsfieldImageToImage": ".higgsfield",
+    "HiggsfieldImageToVideo": ".higgsfield",
+    "HiggsfieldLipsync": ".higgsfield",
+    "HiggsfieldRemoveBackground": ".higgsfield",
+    "HiggsfieldSoulGenerate": ".higgsfield",
+    "HiggsfieldSoulTrain": ".higgsfield",
+    "HiggsfieldSubmit": ".higgsfield",
+    "HiggsfieldTextToImage": ".higgsfield",
+    "HiggsfieldTextToVideo": ".higgsfield",
+    "HiggsfieldUploadFile": ".higgsfield",
     "ElasticsearchIngestAttachment": ".elasticsearch",
     "ElasticsearchBulkOperations": ".elasticsearch",
     "ElasticsearchDeleteDocument": ".elasticsearch",

{timbal-2.2.0 → timbal-2.2.2}/python/timbal/tools/gemini_images.py

@@ -8,9 +8,15 @@ from ._creds import resolve_api_key
 
 _GEMINI_BASE = "https://generativelanguage.googleapis.com/v1beta/models"
 
-_DEFAULT_EDIT_MODEL = "gemini-3.1-flash-image-preview"
-_DEFAULT_GENERATE_MODEL = "gemini-3.1-flash-image-preview"
-_DEFAULT_ANALYZE_MODEL = "gemini-3.1-flash-image-preview"
+# Image generation/editing: stable GA model (listed as "Nano Banana 2" in the Gemini API).
+# Preview sibling gemini-3.1-flash-image-preview still works but is not the stable default.
+_DEFAULT_IMAGE_MODEL = "gemini-3.1-flash-image"
+# Vision analysis: general multimodal flash — no image output modality needed.
+_DEFAULT_ANALYZE_MODEL = "gemini-3.5-flash"
+
+
+def _gemini_headers(api_key: str) -> dict[str, str]:
+    return {"x-goog-api-key": api_key, "Content-Type": "application/json"}
 
 
 def _parse_gemini_response(data: dict[str, Any]) -> dict[str, Any]:
@@ -46,10 +52,17 @@ class GeminiImagesEditImage(Tool):
 
     def __init__(self, **kwargs: Any) -> None:
         async def _edit_image(
-            prompt: str = Field(..., description="Instruction describing the desired edit, e.g. 'remove the background'."),
+            prompt: str = Field(
+                ..., description="Instruction describing the desired edit, e.g. 'remove the background'."
+            ),
             image_base64: str = Field(..., description="The source image encoded as a base64 string."),
-            image_mime_type: str = Field("image/png", description="MIME type of the input image, e.g. 'image/png', 'image/jpeg'."),
-            model: str = Field(_DEFAULT_EDIT_MODEL, description="Gemini model to use. Defaults to gemini-2.0-flash-exp-image-generation."),
+            image_mime_type: str = Field(
+                "image/png", description="MIME type of the input image, e.g. 'image/png', 'image/jpeg'."
+            ),
+            model: str = Field(
+                _DEFAULT_IMAGE_MODEL,
+                description=f"Gemini image model. Defaults to {_DEFAULT_IMAGE_MODEL}.",
+            ),
         ) -> Any:
             api_key = await resolve_api_key(tool=self, provider_name="Gemini", env_var="GEMINI_API_KEY")
             import httpx
@@ -74,7 +87,7 @@ class GeminiImagesEditImage(Tool):
             async with httpx.AsyncClient(timeout=120.0) as client:
                 response = await client.post(
                     f"{_GEMINI_BASE}/{model}:generateContent",
-                    headers={"Authorization": f"Bearer {api_key}"},
+                    headers=_gemini_headers(api_key),
                     json=payload,
                 )
                 response.raise_for_status()
@@ -86,8 +99,7 @@ class GeminiImagesEditImage(Tool):
 class GeminiImagesGenerateImage(Tool):
     name: str = "gemini_generate_image"
     description: str | None = (
-        "Generate a new image from a text prompt using Gemini. "
-        "Returns the generated image as a base64 string."
+        "Generate a new image from a text prompt using Gemini. Returns the generated image as a base64 string."
     )
     integration: Annotated[str, Integration("gemini")] | None = None
     api_key: SecretStr | None = None
@@ -101,8 +113,13 @@ class GeminiImagesGenerateImage(Tool):
     def __init__(self, **kwargs: Any) -> None:
         async def _generate_image(
             prompt: str = Field(..., description="Text description of the image to generate."),
-            negative_prompt: str | None = Field(None, description="Optional description of what to avoid in the image."),
-            model: str = Field(_DEFAULT_GENERATE_MODEL, description="Gemini model to use. Defaults to gemini-2.0-flash-exp-image-generation."),
+            negative_prompt: str | None = Field(
+                None, description="Optional description of what to avoid in the image."
+            ),
+            model: str = Field(
+                _DEFAULT_IMAGE_MODEL,
+                description=f"Gemini image model. Defaults to {_DEFAULT_IMAGE_MODEL}.",
+            ),
         ) -> Any:
             api_key = await resolve_api_key(tool=self, provider_name="Gemini", env_var="GEMINI_API_KEY")
             import httpx
@@ -112,18 +129,14 @@ class GeminiImagesGenerateImage(Tool):
                 text = f"{prompt}\n\nAvoid: {negative_prompt}"
 
             payload = {
-                "contents": [
-                    {
-                        "parts": [{"text": text}]
-                    }
-                ],
+                "contents": [{"parts": [{"text": text}]}],
                 "generationConfig": {"responseModalities": ["IMAGE", "TEXT"]},
             }
 
             async with httpx.AsyncClient(timeout=120.0) as client:
                 response = await client.post(
                     f"{_GEMINI_BASE}/{model}:generateContent",
-                    headers={"Authorization": f"Bearer {api_key}"},
+                    headers=_gemini_headers(api_key),
                     json=payload,
                 )
                 response.raise_for_status()
@@ -149,10 +162,18 @@ class GeminiImagesAnalyzeImage(Tool):
 
     def __init__(self, **kwargs: Any) -> None:
         async def _analyze_image(
-            prompt: str = Field(..., description="Question or instruction about the image, e.g. 'Describe this image', 'What text is visible?', 'List all objects'."),
+            prompt: str = Field(
+                ...,
+                description="Question or instruction about the image, e.g. 'Describe this image', 'What text is visible?', 'List all objects'.",
+            ),
             image_base64: str = Field(..., description="The image encoded as a base64 string."),
-            image_mime_type: str = Field("image/png", description="MIME type of the image, e.g. 'image/png', 'image/jpeg'."),
-            model: str = Field(_DEFAULT_ANALYZE_MODEL, description="Gemini model to use. Defaults to gemini-2.0-flash."),
+            image_mime_type: str = Field(
+                "image/png", description="MIME type of the image, e.g. 'image/png', 'image/jpeg'."
+            ),
+            model: str = Field(
+                _DEFAULT_ANALYZE_MODEL,
+                description=f"Gemini vision model. Defaults to {_DEFAULT_ANALYZE_MODEL}.",
+            ),
         ) -> Any:
             api_key = await resolve_api_key(tool=self, provider_name="Gemini", env_var="GEMINI_API_KEY")
             import httpx
@@ -177,7 +198,7 @@ class GeminiImagesAnalyzeImage(Tool):
             async with httpx.AsyncClient(timeout=60.0) as client:
                 response = await client.post(
                     f"{_GEMINI_BASE}/{model}:generateContent",
-                    headers={"Authorization": f"Bearer {api_key}"},
+                    headers=_gemini_headers(api_key),
                     json=payload,
                 )
                 response.raise_for_status()
@@ -210,8 +231,13 @@ class GeminiImagesGenerateImageWithReference(Tool):
         async def _generate_image_with_reference(
             prompt: str = Field(..., description="Text instruction describing the desired output image."),
             reference_images_base64: list[str] = Field(..., description="List of reference images as base64 strings."),
-            reference_mime_types: list[str] | None = Field(None, description="MIME type per reference image. Defaults to 'image/png' for all."),
-            model: str = Field(_DEFAULT_GENERATE_MODEL, description="Gemini model to use. Defaults to gemini-2.0-flash-exp-image-generation."),
+            reference_mime_types: list[str] | None = Field(
+                None, description="MIME type per reference image. Defaults to 'image/png' for all."
+            ),
+            model: str = Field(
+                _DEFAULT_IMAGE_MODEL,
+                description=f"Gemini image model. Defaults to {_DEFAULT_IMAGE_MODEL}.",
+            ),
         ) -> Any:
             api_key = await resolve_api_key(tool=self, provider_name="Gemini", env_var="GEMINI_API_KEY")
             import httpx
@@ -230,7 +256,7 @@ class GeminiImagesGenerateImageWithReference(Tool):
             async with httpx.AsyncClient(timeout=120.0) as client:
                 response = await client.post(
                     f"{_GEMINI_BASE}/{model}:generateContent",
-                    headers={"Authorization": f"Bearer {api_key}"},
+                    headers=_gemini_headers(api_key),
                     json=payload,
                 )
                 response.raise_for_status()
@@ -253,6 +279,7 @@ class GeminiImagesImageToBase64(Tool):
             import base64
 
             import httpx
+
             async with httpx.AsyncClient(timeout=30.0) as client:
                 response = await client.get(url)
                 response.raise_for_status()