chore: import upstream snapshot with attribution

2026-07-13 13:21:23 +08:00
commit b957a53def
5423 changed files with 863745 additions and 0 deletions
@@ -0,0 +1,85 @@
+# 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())