Skip to content

Commit 6025506

Browse files
authored
feat(store) : add new model and prompt in image generation (#9099)
Update Marketplace Image generation Prompt and Model **Changes:** - Updated the image generation prompt for Marketplace to better highlight agent functionality: ``` Create a visually engaging app store thumbnail for the AI agent that highlights what it does in a clear and captivating way: - **Name**: {agent.name} - **Description**: {agent.description} Focus on showcasing its core functionality with an appealing design. ``` - Changed the model to `black-forest-labs/flux-1.1-pro` for improved results.
1 parent 54f8d3b commit 6025506

File tree

1 file changed

+2
-2
lines changed
  • autogpt_platform/backend/backend/server/v2/store

1 file changed

+2
-2
lines changed

autogpt_platform/backend/backend/server/v2/store/image_gen.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ async def generate_agent_image(agent: Graph) -> io.BytesIO:
3838
raise ValueError("Missing Replicate API key in settings")
3939

4040
# Construct prompt from agent details
41-
prompt = f"App store image for AI agent that gives a cool visual representation of what the agent does: - {agent.name} - {agent.description}"
41+
prompt = f"Create a visually engaging app store thumbnail for the AI agent that highlights what it does in a clear and captivating way:\n- **Name**: {agent.name}\n- **Description**: {agent.description}\nFocus on showcasing its core functionality with an appealing design."
4242

4343
# Set up Replicate client
4444
client = replicate.Client(api_token=settings.secrets.replicate_api_key)
@@ -59,7 +59,7 @@ async def generate_agent_image(agent: Graph) -> io.BytesIO:
5959

6060
try:
6161
# Run model
62-
output = client.run("black-forest-labs/flux-pro", input=input_data)
62+
output = client.run("black-forest-labs/flux-1.1-pro", input=input_data)
6363

6464
# Depending on the model output, extract the image URL or bytes
6565
# If the output is a list of FileOutput or URLs

0 commit comments

Comments
 (0)