Spaces:
Paused
Paused
File size: 2,975 Bytes
0b9dc2e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | # -*- coding: utf-8 -*-
"""Example of Moonshot model calls with MoonshotMultiAgentFormatter.
The multi-agent formatter wraps prior conversation history in
<history></history> tags and preserves reasoning_content for Moonshot's
Preserved Thinking feature in multi-turn conversations.
"""
import asyncio
import os
from _utils import stream_and_collect
from agentscope.formatter import MoonshotMultiAgentFormatter
from agentscope.message import Msg, TextBlock
from agentscope.model import MoonshotChatModel
from agentscope.credential import MoonshotCredential
async def example_multiagent() -> None:
"""Simulate a multi-agent conversation and let kimi-k2.6 summarize it.
Alice and Bob discuss the weather, then a moderator (the model) is asked
to summarize the conversation.
"""
formatter = MoonshotMultiAgentFormatter()
model = MoonshotChatModel(
credential=MoonshotCredential(
api_key=os.environ["MOONSHOT_API_KEY"],
),
model="kimi-k2.6",
stream=True,
context_size=262_144,
parameters=MoonshotChatModel.Parameters(thinking_enable=True),
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())
|