pygeai 0.6.0b10__py3-none-any.whl → 0.6.0b11__py3-none-any.whl

pygeai/chat/clients.py

@@ -4,7 +4,7 @@ from pathlib import Path
 from typing import List, Dict, Optional, Union, Generator
 
 from pygeai import logger
-from pygeai.chat.endpoints import CHAT_V1, GENERATE_IMAGE_V1, RESPONSES_V1
+from pygeai.chat.endpoints import CHAT_V1, GENERATE_IMAGE_V1, EDIT_IMAGE_V1, RESPONSES_V1
 from pygeai.core.base.clients import BaseClient
 from pygeai.core.common.exceptions import InvalidAPIResponseException
 from pygeai.core.utils.validators import validate_status_code
@@ -275,6 +275,51 @@ class ChatClient(BaseClient):
         logger.debug(f"Image generation result: {result}")
         return result
 
+    def edit_image(
+            self,
+            model: str,
+            prompt: str,
+            image: str,
+            size: str,
+            n: int = 1,
+            quality: Optional[str] = None
+    ) -> dict:
+        """
+        Edits an existing image based on the provided parameters.
+
+        :param model: str - The model specification for image editing, e.g., "openai/gpt-image-1". (Required)
+        :param prompt: str - Description of the desired edit, e.g., "remove background people". (Required)
+        :param image: str - URL of the image to be edited, e.g., "https://example.com/image.jpg". (Required)
+        :param size: str - Desired dimensions of the output image in pixels, e.g., "1024x1024". (Required)
+        :param n: int - Number of edited images to generate (1-10, depending on the model). Default is 1. (Optional)
+        :param quality: Optional[str] - Rendering quality, e.g., "high", "medium", "low". (Optional)
+        :return: dict - The API response containing the edited image data.
+        :raises InvalidAPIResponseException: If the API response cannot be processed.
+        """
+        data = {
+            'model': model,
+            'prompt': prompt,
+            'image': image,
+            'size': size,
+            'n': n
+        }
+
+        if quality:
+            data['quality'] = quality
+
+        logger.debug(f"Editing image with data: {data}")
+
+        response = self.api_service.post(
+            endpoint=EDIT_IMAGE_V1,
+            data=data
+        )
+
+        validate_status_code(response)
+
+        result = parse_json_response(response, "edit image")
+        logger.debug(f"Image editing result: {result}")
+        return result
+
     def get_response(
             self,
             model: str,

pygeai/chat/endpoints.py

@@ -1,4 +1,5 @@
 CHAT_V1 = "/chat"                           # POST Chat endpoint
 CHAT_COMPLETION_V1 = "/chat/completion"     # POST Chat completion endpoint
 GENERATE_IMAGE_V1 = "/images"               # POST Generate image
+EDIT_IMAGE_V1 = "/images/edits"             # POST Edit image
 RESPONSES_V1 = "/responses"                 # POST Responses API endpoint

pygeai/cli/commands/chat.py

@@ -661,6 +661,92 @@ generate_image_options = [
 ]
 
 
+def get_edit_image(option_list: list):
+    model = None
+    prompt = None
+    image = None
+    size = None
+    n = 1
+    quality = None
+
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "model":
+            model = option_arg
+        if option_flag.name == "prompt":
+            prompt = option_arg
+        if option_flag.name == "image":
+            image = option_arg
+        if option_flag.name == "size":
+            size = option_arg
+        if option_flag.name == "n":
+            try:
+                n = int(option_arg)
+                if n < 1 or n > 10:
+                    raise WrongArgumentError("n must be an integer between 1 and 10.")
+            except ValueError:
+                raise WrongArgumentError("n must be a valid integer.")
+        if option_flag.name == "quality":
+            quality = option_arg
+
+    if not (model and prompt and image and size):
+        raise MissingRequirementException("Cannot edit image without specifying model, prompt, image, and size.")
+
+    client = ChatClient()
+    try:
+        result = client.edit_image(
+            model=model,
+            prompt=prompt,
+            image=image,
+            size=size,
+            n=n,
+            quality=quality
+        )
+        Console.write_stdout(f"Image editing result: \n{result}\n")
+    except Exception as e:
+        logger.error(f"Error editing image: {e}")
+        Console.write_stderr(f"Failed to edit image: {e}")
+
+
+edit_image_options = [
+    Option(
+        "model",
+        ["--model", "-m"],
+        "The model specification for image editing, e.g., 'openai/gpt-image-1'.",
+        True
+    ),
+    Option(
+        "prompt",
+        ["--prompt", "-p"],
+        "Description of the desired edit, e.g., 'remove the ball'.",
+        True
+    ),
+    Option(
+        "image",
+        ["--image", "-img"],
+        "URL of the image to be edited, e.g., 'https://example.com/image.jpg'.",
+        True
+    ),
+    Option(
+        "size",
+        ["--size", "-s"],
+        "Desired dimensions of the output image in pixels, e.g., '1024x1024'.",
+        True
+    ),
+    Option(
+        "n",
+        ["--n"],
+        "Number of edited images to generate (1-10, depending on the model). Default is 1.",
+        True
+    ),
+    Option(
+        "quality",
+        ["--quality", "-q"],
+        "Rendering quality, e.g., 'high', 'medium', 'low'.",
+        True
+    ),
+]
+
+
 def get_response(option_list: list):
     model = None
     input_text = None
@@ -912,6 +998,15 @@ chat_commands = [
         [],
         generate_image_options
     ),
+    Command(
+        "edit_image",
+        ["edit-image", "edit-img"],
+        "Edit an existing image using the specified model and parameters",
+        get_edit_image,
+        ArgumentsEnum.REQUIRED,
+        [],
+        edit_image_options
+    ),
     Command(
         "response",
         ["response", "resp"],

{pygeai-0.6.0b10.dist-info → pygeai-0.6.0b11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygeai
-Version: 0.6.0b10
+Version: 0.6.0b11
 Summary: Software Development Kit to interact with Globant Enterprise AI.
 Author-email: Globant <geai-sdk@globant.com>
 License-Expression: MIT

{pygeai-0.6.0b10.dist-info → pygeai-0.6.0b11.dist-info}/RECORD

@@ -184,8 +184,8 @@ pygeai/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/auth/clients.py,sha256=3Eg9d9XJlPCIXz2N90LalchahV2gPaD0DWj8iBcw41Y,5422
 pygeai/auth/endpoints.py,sha256=KOZ5gS8I8MIamlfbj4bN8w6kMmIdAKr6puKJrw0cEtw,589
 pygeai/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/chat/clients.py,sha256=ANxYMn6vslPW-J4vjVl09CM4TOdSferrDIEcsE_7Fg0,18129
-pygeai/chat/endpoints.py,sha256=4X00h4RPyCo-KNuNca51K78dBtRzp75kDYTzAR1QpNI,281
+pygeai/chat/clients.py,sha256=-PovdRQnwpsHtqoH0UNyAvx8XhDm8xLcZyvTBmsqhNc,19829
+pygeai/chat/endpoints.py,sha256=r_zvsL5UIqnqhz9I0CG0eQSKV-egwx6EKdaZXj-HndA,343
 pygeai/chat/iris.py,sha256=-9pDHQpWhR_PvbZ6rD8eMPFk46PI9sCdPQ9aAyvSexs,413
 pygeai/chat/managers.py,sha256=f0BGfu9EF0G8rUyARslZi0pyDTL2yQadav0taCljI_I,3114
 pygeai/chat/session.py,sha256=k7Y6rr9x7CfAGDI-Vt3c6eGLQX57YZ74lEVJGzwwdzw,1193
@@ -204,7 +204,7 @@ pygeai/cli/commands/assistant.py,sha256=fQ_El6_BmFDpFjm_gPxzWk7bOzhimhiTwG8K0Upc
 pygeai/cli/commands/auth.py,sha256=w1XIcTYh7SQ4keOhQiJiUJ2NFMY349K53P2BoGe2Rgc,8318
 pygeai/cli/commands/base.py,sha256=OKlYNpQvTCxOOLDSR-KkmoUvKs4OGYerwo_escQaDwY,7406
 pygeai/cli/commands/builders.py,sha256=xXk1F4phSQxHN3NiQltl_KEZdCwwJiKLmVqQsft2OC4,1130
-pygeai/cli/commands/chat.py,sha256=A2wGOE7fmqNjOdOVcs_jrSHhma6cqDzY_O3xVwpiPoY,32557
+pygeai/cli/commands/chat.py,sha256=i70GQA21RVMjMf5dsWXthUvLFjjXPb8G54AYGB_pz-s,35202
 pygeai/cli/commands/common.py,sha256=zL1cWKMTqzqMsHBDFfVzbZe0i2l0hgJNpzjSRgvloY8,16568
 pygeai/cli/commands/configuration.py,sha256=J1Y8AH1xYWvcY4bzqKvF-_ggEDJ22Dv4iEcCBi2pq_8,4140
 pygeai/cli/commands/docs.py,sha256=8JEgKAHFL19g1rmgQzFwdzqLBP8ySvnpQogXwJPWfS4,3444
@@ -777,9 +777,9 @@ pygeai/vendor/a2a/utils/helpers.py,sha256=6Tbd8SVfXvdNEk6WYmLOjrAxkzFf1aIg8dkFfB
 pygeai/vendor/a2a/utils/message.py,sha256=gc_EKO69CJ4HkR76IFgsy-kENJz1dn7CfSgWJWvt-gs,2197
 pygeai/vendor/a2a/utils/task.py,sha256=BYRA_L1HpoUGJAVlyHML0lCM9Awhf2Ovjj7oPFXKbh0,1647
 pygeai/vendor/a2a/utils/telemetry.py,sha256=VvSp1Ztqaobkmq9-3sNhhPEilJS32-JTSfKzegkj6FU,10861
-pygeai-0.6.0b10.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
-pygeai-0.6.0b10.dist-info/METADATA,sha256=qJnwOAK8YfAZS4xF4Po-PbUzcziLyOrIx2VP2ODfTU8,7978
-pygeai-0.6.0b10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pygeai-0.6.0b10.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
-pygeai-0.6.0b10.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
-pygeai-0.6.0b10.dist-info/RECORD,,
+pygeai-0.6.0b11.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
+pygeai-0.6.0b11.dist-info/METADATA,sha256=HX6wRMsmyVkQxxEI7ykag0wIgJ9aiBWkZoEFR1DFjCI,7978
+pygeai-0.6.0b11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygeai-0.6.0b11.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
+pygeai-0.6.0b11.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
+pygeai-0.6.0b11.dist-info/RECORD,,