alita-sdk 0.3.359__py3-none-any.whl → 0.3.360__py3-none-any.whl

alita_sdk/runtime/clients/client.py

@@ -231,25 +231,40 @@ class AlitaClient:
         if not self.model_image_generation:
             raise ValueError("Image generation model is not configured for this client")
 
-        # Prepare the request data
+        # Prepare the request data - only include non-"auto" parameters
+        # This prevents LiteLLM from adding unsupported parameters to extra_body
         image_generation_data = {
             "prompt": prompt,
             "model": self.model_image_generation,
             "n": n,
-            "size": size,
-            "quality": quality,
             "response_format": response_format,
+            # Enable drop_params to prevent LiteLLM from adding unsupported params to extra_body
+            "drop_params": True
         }
 
+        # Only add optional parameters if they have meaningful values
+        if size and size.lower() != "auto":
+            image_generation_data["size"] = size
+            
+        if quality and quality.lower() != "auto":
+            image_generation_data["quality"] = quality
+
         if style:
             image_generation_data["style"] = style
 
+        # Standard headers for image generation
+        image_headers = self.headers.copy()
+        image_headers.update({
+            "Content-Type": "application/json",
+        })
+
         logger.info(f"Generating image with model: {self.model_image_generation}, prompt: {prompt[:50]}...")
+        logger.debug(f"Image generation request data: {image_generation_data}")
 
         try:
             response = requests.post(
                 self.image_generation_url,
-                headers=self.headers,
+                headers=image_headers,
                 json=image_generation_data,
                 verify=False,
                 timeout=self.model_timeout
@@ -259,6 +274,12 @@ class AlitaClient:
 
         except requests.exceptions.HTTPError as e:
             logger.error(f"Image generation failed: {e.response.status_code} - {e.response.text}")
+            # Try to parse the error response for more details
+            try:
+                error_details = e.response.json()
+                logger.error(f"Error details: {error_details}")
+            except:
+                pass
             raise
         except requests.exceptions.RequestException as e:
             logger.error(f"Image generation request failed: {e}")

{alita_sdk-0.3.359.dist-info → alita_sdk-0.3.360.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alita_sdk
-Version: 0.3.359
+Version: 0.3.360
 Summary: SDK for building langchain agents using resources from Alita
 Author-email: Artem Rozumenko <artyom.rozumenko@gmail.com>, Mikalai Biazruchka <mikalai_biazruchka@epam.com>, Roman Mitusov <roman_mitusov@epam.com>, Ivan Krakhmaliuk <lifedj27@gmail.com>, Artem Dubrovskiy <ad13box@gmail.com>
 License-Expression: Apache-2.0

{alita_sdk-0.3.359.dist-info → alita_sdk-0.3.360.dist-info}/RECORD

@@ -36,7 +36,7 @@ alita_sdk/configurations/zephyr_essential.py,sha256=tUIrh-PRNvdrLBj6rJXqlF-h6oaM
 alita_sdk/runtime/__init__.py,sha256=4W0UF-nl3QF2bvET5lnah4o24CoTwSoKXhuN0YnwvEE,828
 alita_sdk/runtime/clients/__init__.py,sha256=BdehU5GBztN1Qi1Wul0cqlU46FxUfMnI6Vq2Zd_oq1M,296
 alita_sdk/runtime/clients/artifact.py,sha256=Tt3aWcxu20bVW6EX7s_iX5CTmcItKhUnkk8Q2gv2vw0,4036
-alita_sdk/runtime/clients/client.py,sha256=sNa_2Ep1ugfuMQS27LxkPnw05EChpNOW3MdsdhL8bSA,45421
+alita_sdk/runtime/clients/client.py,sha256=dBk8HpvV2sdQnUg0fAYK8wEFSrOybBxqPFs9FmQILCA,46391
 alita_sdk/runtime/clients/datasource.py,sha256=HAZovoQN9jBg0_-lIlGBQzb4FJdczPhkHehAiVG3Wx0,1020
 alita_sdk/runtime/clients/prompt.py,sha256=li1RG9eBwgNK_Qf0qUaZ8QNTmsncFrAL2pv3kbxZRZg,1447
 alita_sdk/runtime/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -352,8 +352,8 @@ alita_sdk/tools/zephyr_scale/api_wrapper.py,sha256=kT0TbmMvuKhDUZc0i7KO18O38JM9S
 alita_sdk/tools/zephyr_squad/__init__.py,sha256=0ne8XLJEQSLOWfzd2HdnqOYmQlUliKHbBED5kW_Vias,2895
 alita_sdk/tools/zephyr_squad/api_wrapper.py,sha256=kmw_xol8YIYFplBLWTqP_VKPRhL_1ItDD0_vXTe_UuI,14906
 alita_sdk/tools/zephyr_squad/zephyr_squad_cloud_client.py,sha256=R371waHsms4sllHCbijKYs90C-9Yu0sSR3N4SUfQOgU,5066
-alita_sdk-0.3.359.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-alita_sdk-0.3.359.dist-info/METADATA,sha256=dG_WTlIsQxUJe23SVu_qFSJZM_5VXyW-xfG7UrcTSEE,19071
-alita_sdk-0.3.359.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-alita_sdk-0.3.359.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
-alita_sdk-0.3.359.dist-info/RECORD,,
+alita_sdk-0.3.360.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+alita_sdk-0.3.360.dist-info/METADATA,sha256=Y-VuZWNlusthSXN5C6oGtaH619yAib7Xil-fAVAq0MU,19071
+alita_sdk-0.3.360.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+alita_sdk-0.3.360.dist-info/top_level.txt,sha256=0vJYy5p_jK6AwVb1aqXr7Kgqgk3WDtQ6t5C-XI9zkmg,10
+alita_sdk-0.3.360.dist-info/RECORD,,