wehub-resource-sync
c3bf08ac8d
K8s Workspace Integration Tests / k8s-workspace-tests (push) Waiting to run
Pre-commit / run (ubuntu-latest) (push) Waiting to run
Python Unittest Coverage / test (macos-15, 3.11) (push) Waiting to run
Python Unittest Coverage / test (ubuntu-latest, 3.11) (push) Waiting to run
Python Unittest Coverage / test (windows-latest, 3.11) (push) Waiting to run
Web UI / check (push) Waiting to run
106 lines
2.9 KiB
Python
106 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Example of Anthropic Claude model calls with AnthropicMultiAgentFormatter.
|
|
|
|
The multi-agent formatter wraps prior conversation history in
|
|
<history></history> tags, enabling the model to handle multi-agent
|
|
conversations where more than one non-user agent is involved.
|
|
"""
|
|
import asyncio
|
|
import os
|
|
|
|
from _utils import stream_and_collect
|
|
from agentscope.formatter import AnthropicMultiAgentFormatter
|
|
from agentscope.message import Msg, TextBlock
|
|
from agentscope.model import AnthropicChatModel
|
|
from agentscope.credential import AnthropicCredential
|
|
|
|
|
|
async def example_multiagent() -> None:
|
|
"""Simulate a multi-agent conversation and let claude-opus-4-5
|
|
summarize it.
|
|
|
|
Alice and Bob discuss the weather, then a moderator (the model) is asked
|
|
to summarize the conversation.
|
|
"""
|
|
formatter = AnthropicMultiAgentFormatter()
|
|
|
|
model = AnthropicChatModel(
|
|
credential=AnthropicCredential(
|
|
api_key=os.environ["ANTHROPIC_API_KEY"],
|
|
),
|
|
model="claude-opus-4-5",
|
|
stream=True,
|
|
context_size=1_000_000,
|
|
parameters=AnthropicChatModel.Parameters(
|
|
thinking_enable=True,
|
|
thinking_budget=1024,
|
|
),
|
|
formatter=formatter,
|
|
)
|
|
|
|
# Multi-agent conversation history between Alice and Bob
|
|
msgs = [
|
|
Msg(
|
|
name="system",
|
|
content=[
|
|
TextBlock(
|
|
text="You are a helpful moderator. Summarize the "
|
|
"conversation.",
|
|
),
|
|
],
|
|
role="system",
|
|
),
|
|
Msg(
|
|
name="alice",
|
|
content=[
|
|
TextBlock(
|
|
text="Hi Bob! What do you think about the weather today?",
|
|
),
|
|
],
|
|
role="user",
|
|
),
|
|
Msg(
|
|
name="bob",
|
|
content=[
|
|
TextBlock(
|
|
text="It's quite sunny and warm, Alice. Perfect for a "
|
|
"walk!",
|
|
),
|
|
],
|
|
role="assistant",
|
|
),
|
|
Msg(
|
|
name="alice",
|
|
content=[
|
|
TextBlock(text="Agreed! I might head to the park later."),
|
|
],
|
|
role="user",
|
|
),
|
|
Msg(
|
|
name="bob",
|
|
content=[
|
|
TextBlock(
|
|
text="Great idea. I'll join you if I finish work early.",
|
|
),
|
|
],
|
|
role="assistant",
|
|
),
|
|
Msg(
|
|
name="moderator",
|
|
content=[
|
|
TextBlock(
|
|
text="Please summarize the conversation above in one "
|
|
"sentence.",
|
|
),
|
|
],
|
|
role="user",
|
|
),
|
|
]
|
|
|
|
print("=== Multi-Agent Formatter Call ===")
|
|
await stream_and_collect(await model(msgs))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(example_multiagent())
|