add AI Agent example

Author: ks6088ts

apps/1_call_azure_openai_chat/agents_stream_iteration.py

@@ -0,0 +1,58 @@
+# https://github.com/Azure/azure-sdk-for-python/blob/main/sdk/ai/azure-ai-projects/samples/agents/sample_agents_stream_iteration.py
+import os
+
+from azure.ai.projects import AIProjectClient
+from azure.ai.projects.models import AgentStreamEvent, MessageDeltaChunk, RunStep, ThreadMessage, ThreadRun
+from azure.identity import DefaultAzureCredential
+
+project_client = AIProjectClient.from_connection_string(
+    credential=DefaultAzureCredential(),
+    conn_str=os.environ["PROJECT_CONNECTION_STRING"],
+)
+
+with project_client:
+    # Create an agent and run stream with iteration
+    agent = project_client.agents.create_agent(
+        model=os.environ["MODEL_DEPLOYMENT_NAME"],
+        name="my-assistant",
+        instructions="You are a helpful assistant",
+    )
+    print(f"Created agent, ID {agent.id}")
+
+    thread = project_client.agents.create_thread()
+    print(f"Created thread, thread ID {thread.id}")
+
+    message = project_client.agents.create_message(thread_id=thread.id, role="user", content="Hello, tell me a joke")
+    print(f"Created message, message ID {message.id}")
+
+    # [START iterate_stream]
+    with project_client.agents.create_stream(thread_id=thread.id, agent_id=agent.id) as stream:
+        for event_type, event_data, _ in stream:
+            if isinstance(event_data, MessageDeltaChunk):
+                print(f"Text delta received: {event_data.text}")
+
+            elif isinstance(event_data, ThreadMessage):
+                print(f"ThreadMessage created. ID: {event_data.id}, Status: {event_data.status}")
+
+            elif isinstance(event_data, ThreadRun):
+                print(f"ThreadRun status: {event_data.status}")
+
+            elif isinstance(event_data, RunStep):
+                print(f"RunStep type: {event_data.type}, Status: {event_data.status}")
+
+            elif event_type == AgentStreamEvent.ERROR:
+                print(f"An error occurred. Data: {event_data}")
+
+            elif event_type == AgentStreamEvent.DONE:
+                print("Stream completed.")
+                break
+
+            else:
+                print(f"Unhandled Event Type: {event_type}, Data: {event_data}")
+    # [END iterate_stream]
+
+    project_client.agents.delete_agent(agent.id)
+    print("Deleted agent")
+
+    messages = project_client.agents.list_messages(thread_id=thread.id)
+    print(f"Messages: {messages}")