wehub-resource-sync
86 lines
3.1 KiB
Python
86 lines
3.1 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
import asyncio
|
|
from typing import Annotated
|
|
|
|
from azure.identity import AzureCliCredential
|
|
|
|
from semantic_kernel.agents import AssistantAgentThread, AzureAssistantAgent
|
|
from semantic_kernel.connectors.ai.open_ai import AzureOpenAISettings
|
|
from semantic_kernel.contents import AuthorRole
|
|
from semantic_kernel.functions import kernel_function
|
|
|
|
"""
|
|
The following sample demonstrates how to create an OpenAI
|
|
assistant using either Azure OpenAI or OpenAI. OpenAI Assistants
|
|
allow for function calling, the use of file search and a
|
|
code interpreter. Assistant Threads are used to manage the
|
|
conversation state, similar to a Semantic Kernel Chat History.
|
|
This sample also demonstrates the Assistants Streaming
|
|
capability and how to manage an Assistants chat history.
|
|
"""
|
|
|
|
|
|
# Define a sample plugin for the sample
|
|
class MenuPlugin:
|
|
"""A sample Menu Plugin used for the concept sample."""
|
|
|
|
@kernel_function(description="Provides a list of specials from the menu.")
|
|
def get_specials(self) -> Annotated[str, "Returns the specials from the menu."]:
|
|
return """
|
|
Special Soup: Clam Chowder
|
|
Special Salad: Cobb Salad
|
|
Special Drink: Chai Tea
|
|
"""
|
|
|
|
@kernel_function(description="Provides the price of the requested menu item.")
|
|
def get_item_price(
|
|
self, menu_item: Annotated[str, "The name of the menu item."]
|
|
) -> Annotated[str, "Returns the price of the menu item."]:
|
|
return "$9.99"
|
|
|
|
|
|
async def main():
|
|
# Create the client using Azure OpenAI resources and configuration
|
|
client = AzureAssistantAgent.create_client(credential=AzureCliCredential())
|
|
|
|
# Define the assistant definition
|
|
definition = await client.beta.assistants.create(
|
|
model=AzureOpenAISettings().chat_deployment_name,
|
|
name="Host",
|
|
instructions="Answer questions about the menu.",
|
|
)
|
|
|
|
# Create the AzureAssistantAgent instance using the client and the assistant definition and the defined plugin
|
|
agent = AzureAssistantAgent(
|
|
client=client,
|
|
definition=definition,
|
|
plugins=[MenuPlugin()],
|
|
)
|
|
|
|
# Create a new thread for use with the assistant
|
|
# If no thread is provided, a new thread will be
|
|
# created and returned with the initial response
|
|
thread: AssistantAgentThread = None
|
|
|
|
user_inputs = ["Hello", "What is the special soup?", "What is the special drink?", "How much is that?", "Thank you"]
|
|
|
|
try:
|
|
for user_input in user_inputs:
|
|
print(f"# {AuthorRole.USER}: '{user_input}'")
|
|
|
|
first_chunk = True
|
|
async for response in agent.invoke_stream(messages=user_input, thread=thread):
|
|
thread = response.thread
|
|
if first_chunk:
|
|
print(f"# {response.role}: ", end="", flush=True)
|
|
first_chunk = False
|
|
print(response.content, end="", flush=True)
|
|
print()
|
|
finally:
|
|
await thread.delete() if thread else None
|
|
await client.beta.assistants.delete(assistant_id=agent.id)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|