mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-04 07:26:28 +00:00
292964934b
* Add initial file surfer * Fix return type * Add example for running coder * Run hatch check * Add example for file surfer * Add aiofiles to dependencies
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import asyncio
|
|
|
|
from agnext.application import SingleThreadedAgentRuntime
|
|
from agnext.components.models import OpenAIChatCompletionClient
|
|
from team_one.coder import Coder, LLMResponseMessage, TaskMessage
|
|
|
|
|
|
async def main() -> None:
|
|
# Create the runtime.
|
|
runtime = SingleThreadedAgentRuntime()
|
|
|
|
# Register agents.
|
|
coder = runtime.register_and_get(
|
|
"coder",
|
|
lambda: Coder(model_client=OpenAIChatCompletionClient(model="gpt-4o")),
|
|
)
|
|
|
|
task = TaskMessage(input(f"Enter a task for {coder.name}: "))
|
|
|
|
# Send a task to the tool user.
|
|
result = runtime.send_message(task, coder)
|
|
|
|
# Run the runtime until the task is completed.
|
|
while not result.done():
|
|
await runtime.process_next()
|
|
|
|
# Print the result.
|
|
final_response = result.result()
|
|
assert isinstance(final_response, LLMResponseMessage)
|
|
print("--------------------")
|
|
print(final_response.content)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import logging
|
|
|
|
logging.basicConfig(level=logging.WARNING)
|
|
logging.getLogger("agnext").setLevel(logging.DEBUG)
|
|
asyncio.run(main())
|