mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-21 07:51:55 +00:00
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
|
import os
|
||
|
|
|
||
|
|
from agnext.components.models import AzureOpenAIChatCompletionClient
|
||
|
|
from agnext.core import AgentRuntime
|
||
|
|
from auditor import AuditAgent
|
||
|
|
from graphic_designer import GraphicDesignerAgent
|
||
|
|
from openai import AsyncAzureOpenAI
|
||
|
|
|
||
|
|
|
||
|
|
async def build_app(runtime: AgentRuntime) -> None:
|
||
|
|
chat_client = AzureOpenAIChatCompletionClient(
|
||
|
|
model="gpt-4-32",
|
||
|
|
azure_endpoint=os.environ["CHAT_ENDPOINT"],
|
||
|
|
api_version="2024-02-01",
|
||
|
|
model_capabilities={
|
||
|
|
"vision": True,
|
||
|
|
"function_calling": True,
|
||
|
|
"json_output": True,
|
||
|
|
},
|
||
|
|
api_key=os.environ["CHAT_ENDPOINT_KEY"],
|
||
|
|
)
|
||
|
|
|
||
|
|
image_client = AsyncAzureOpenAI(
|
||
|
|
azure_endpoint=os.environ["IMAGE_ENDPOINT"],
|
||
|
|
azure_deployment="dall-e-3",
|
||
|
|
api_key=os.environ["IMAGE_ENDPOINT_KEY"],
|
||
|
|
api_version="2024-02-01",
|
||
|
|
)
|
||
|
|
|
||
|
|
runtime.register("GraphicDesigner", lambda: GraphicDesignerAgent(client=image_client))
|
||
|
|
runtime.register("Auditor", lambda: AuditAgent(model_client=chat_client))
|