github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix TeamOne examples with subscriptions to topics (#407)

Browse Source 
* logprobs openai support

* remove print statement

* Update _types.py

* Update _openai_client.py

* empty commit to run checks

* fix teamone examples subscription

* update sample usage

---------

Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com>
Co-authored-by: Jack Gerrits <jackgerrits@users.noreply.github.com>
Co-authored-by: Jack Gerrits <jack@jackgerrits.com>
This commit is contained in:
Hussein Mozannar
2024-08-27 12:03:21 -07:00
committed by GitHub
parent 4eadecf4c9
commit c8f6f3bb38
22 changed files with 125 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
python/samples/patterns/coder_executor.py
View File

@@ -21,8 +21,7 @@ from dataclasses import dataclass
from typing import Dict, List
from agnext.application import SingleThreadedAgentRuntime
from agnext.components import DefaultTopicId, RoutedAgent, message_handler
from agnext.components._type_subscription import TypeSubscription
from agnext.components import DefaultSubscription, DefaultTopicId, RoutedAgent, message_handler
from agnext.components.code_executor import CodeBlock, CodeExecutor, LocalCommandLineCodeExecutor
from agnext.components.models import (
AssistantMessage,
@@ -31,7 +30,6 @@ from agnext.components.models import (
SystemMessage,
UserMessage,
)
from agnext.core import TopicId
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
@@ -190,15 +188,19 @@ async def main(task: str, temp_dir: str) -> None:
# Register the agents.
await runtime.register(
"coder", lambda: Coder(model_client=get_chat_completion_client_from_envs(model="gpt-4-turbo"))
"coder",
lambda: Coder(model_client=get_chat_completion_client_from_envs(model="gpt-4-turbo")),
lambda: [DefaultSubscription()],
)
await runtime.register(
"executor",
lambda: Executor(executor=LocalCommandLineCodeExecutor(work_dir=temp_dir)),
lambda: [DefaultSubscription()],
)
await runtime.register("executor", lambda: Executor(executor=LocalCommandLineCodeExecutor(work_dir=temp_dir)))
await runtime.add_subscription(TypeSubscription("default", "coder"))
await runtime.add_subscription(TypeSubscription("default", "executor"))
runtime.start()
# Publish the task message.
await runtime.publish_message(TaskMessage(content=task), topic_id=TopicId("default", "default"))
await runtime.publish_message(TaskMessage(content=task), topic_id=DefaultTopicId())
await runtime.stop_when_idle()