chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,116 @@
|
||||
# Copyright (c) Microsoft. All rights reserved.
|
||||
|
||||
import asyncio
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from azure.core.credentials import TokenCredential
|
||||
from azure.identity import AzureCliCredential
|
||||
|
||||
from semantic_kernel.agents import AgentGroupChat, AzureAssistantAgent, ChatCompletionAgent
|
||||
from semantic_kernel.connectors.ai.open_ai import AzureChatCompletion, AzureOpenAISettings
|
||||
from semantic_kernel.contents import AuthorRole
|
||||
from semantic_kernel.kernel import Kernel
|
||||
|
||||
if TYPE_CHECKING:
|
||||
pass
|
||||
|
||||
"""
|
||||
The following sample demonstrates how to create an OpenAI
|
||||
assistant using either Azure OpenAI or OpenAI, a chat completion
|
||||
agent and have them participate in a group chat to work towards
|
||||
the user's requirement. It also demonstrates how the underlying
|
||||
agent reset method is used to clear the current state of the chat
|
||||
|
||||
Note: This sample use the `AgentGroupChat` feature of Semantic Kernel, which is
|
||||
no longer maintained. For a replacement, consider using the `GroupChatOrchestration`.
|
||||
|
||||
Read more about the `GroupChatOrchestration` here:
|
||||
https://learn.microsoft.com/semantic-kernel/frameworks/agent/agent-orchestration/group-chat?pivots=programming-language-python
|
||||
|
||||
Here is a migration guide from `AgentGroupChat` to `GroupChatOrchestration`:
|
||||
https://learn.microsoft.com/semantic-kernel/support/migration/group-chat-orchestration-migration-guide?pivots=programming-language-python
|
||||
"""
|
||||
|
||||
|
||||
def _create_kernel_with_chat_completion(service_id: str, credential: TokenCredential) -> Kernel:
|
||||
kernel = Kernel()
|
||||
kernel.add_service(AzureChatCompletion(service_id=service_id, credential=credential))
|
||||
return kernel
|
||||
|
||||
|
||||
async def main():
|
||||
credential = AzureCliCredential()
|
||||
|
||||
# First create the ChatCompletionAgent
|
||||
chat_agent = ChatCompletionAgent(
|
||||
kernel=_create_kernel_with_chat_completion("chat", credential),
|
||||
name="chat_agent",
|
||||
instructions="""
|
||||
The user may either provide information or query on information previously provided.
|
||||
If the query does not correspond with information provided, inform the user that their query
|
||||
cannot be answered.
|
||||
""",
|
||||
)
|
||||
|
||||
# Next, we will create the AzureAssistantAgent
|
||||
|
||||
# Create the client using Azure OpenAI resources and configuration
|
||||
client = AzureAssistantAgent.create_client(credential=credential)
|
||||
|
||||
# Create the assistant definition
|
||||
definition = await client.beta.assistants.create(
|
||||
model=AzureOpenAISettings().chat_deployment_name,
|
||||
name="copywriter",
|
||||
instructions="""
|
||||
The user may either provide information or query on information previously provided.
|
||||
If the query does not correspond with information provided, inform the user that their query
|
||||
cannot be answered.
|
||||
""",
|
||||
)
|
||||
|
||||
# Create the AzureAssistantAgent instance using the client and the assistant definition
|
||||
assistant_agent = AzureAssistantAgent(
|
||||
client=client,
|
||||
definition=definition,
|
||||
)
|
||||
|
||||
# Create the AgentGroupChat object, which will manage the chat between the agents
|
||||
# We don't always need to specify the agents in the chat up front
|
||||
# As shown below, calling `chat.invoke(agent=<agent>)` will automatically add the
|
||||
# agent to the chat
|
||||
chat = AgentGroupChat()
|
||||
|
||||
try:
|
||||
user_inputs = [
|
||||
"What is my favorite color?",
|
||||
"I like green.",
|
||||
"What is my favorite color?",
|
||||
"[RESET]",
|
||||
"What is my favorite color?",
|
||||
]
|
||||
|
||||
for user_input in user_inputs:
|
||||
# Check for reset indicator
|
||||
if user_input == "[RESET]":
|
||||
print("\nResetting chat...")
|
||||
await chat.reset()
|
||||
continue
|
||||
|
||||
# First agent (assistant_agent) receives the user input
|
||||
await chat.add_chat_message(user_input)
|
||||
print(f"\n{AuthorRole.USER}: '{user_input}'")
|
||||
async for message in chat.invoke(agent=assistant_agent):
|
||||
if message.content is not None:
|
||||
print(f"\n# {message.role} - {message.name or '*'}: '{message.content}'")
|
||||
|
||||
# Second agent (chat_agent) just responds without new user input
|
||||
async for message in chat.invoke(agent=chat_agent):
|
||||
if message.content is not None:
|
||||
print(f"\n# {message.role} - {message.name or '*'}: '{message.content}'")
|
||||
finally:
|
||||
await chat.reset()
|
||||
await assistant_agent.client.beta.assistants.delete(assistant_agent.id)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
Reference in New Issue
Block a user