mirror of
https://github.com/microsoft/autogen.git
synced 2025-08-01 21:32:38 +00:00
e1a823fb6d
* initial impl of topics and subscriptions * Update python/src/agnext/core/_agent_runtime.py Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com> * add topic in context * migrate * migrate code for topics * migrate team one * edit notebooks * formatting * fix imports * Build proto * Fix circular import --------- Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com>
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
import logging
|
|
from typing import Literal
|
|
|
|
import openai
|
|
from agnext.components import (
|
|
TypeRoutedAgent,
|
|
message_handler,
|
|
)
|
|
from agnext.core import MessageContext
|
|
from messages import ArticleCreated, GraphicDesignCreated
|
|
|
|
|
|
class GraphicDesignerAgent(TypeRoutedAgent):
|
|
def __init__(
|
|
self,
|
|
client: openai.AsyncClient,
|
|
model: Literal["dall-e-2", "dall-e-3"] = "dall-e-3",
|
|
):
|
|
super().__init__("")
|
|
self._client = client
|
|
self._model = model
|
|
|
|
@message_handler
|
|
async def handle_user_chat_input(self, message: ArticleCreated, ctx: MessageContext) -> None:
|
|
logger = logging.getLogger("graphic_designer")
|
|
try:
|
|
logger.info(f"Asking model to generate an image for the article '{message.article}'.")
|
|
response = await self._client.images.generate(
|
|
model=self._model, prompt=message.article, response_format="url"
|
|
)
|
|
logger.info(f"Image response: '{response.data[0]}'")
|
|
assert len(response.data) > 0 and response.data[0].url is not None
|
|
image_uri = response.data[0].url
|
|
logger.info(f"Generated image for article. Got response: '{image_uri}'")
|
|
|
|
assert ctx.topic_id is not None
|
|
await self.publish_message(
|
|
GraphicDesignCreated(UserId=message.UserId, imageUri=image_uri), topic_id=ctx.topic_id
|
|
)
|
|
except Exception as e:
|
|
logger.error(f"Failed to generate image for article. Error: {e}")
|