Praison AI

edit • AI Agent SDK

edit: Edit an existing image with a prompt.

edit

Method

</div>

This is a method of the ImageAgent class in the image_agent module.

Edit an existing image with a prompt.

Signature

python
def edit(image: str, prompt: str, mask: Optional[str], n: int, size: Optional[str]) -> Dict[str, Any]

Parameters

ParamField str

Path or URL to the image to edit

ParamField str

Description of the desired edits

ParamField Optional[str]

Optional mask image (transparent areas will be edited)

ParamField int

Number of images to generate

ParamField Optional[str]

Output image size **kwargs: Additional provider-specific parameters

Returns

ResponseField Dict[str, Any]

ImageResponse with edited image(s)

Usage

python
agent = ImageAgent(llm="openai/dall-e-2")
    result = agent.edit("photo.png", "Add a sunset in the background")

Uses

  • litellm.image_edit

Source

View on GitHub

praisonaiagents/agent/image_agent.py at line 247