|
1 | 1 | from azure.ai.projects import AIProjectClient |
2 | 2 | from azure.identity import DefaultAzureCredential |
| 3 | +from azure.ai.agents.models import ListSortOrder |
3 | 4 |
|
4 | 5 | project_client = AIProjectClient.from_connection_string( |
5 | 6 | credential=DefaultAzureCredential(), |
6 | 7 | conn_str="<%= connectionString %>") |
7 | 8 |
|
8 | 9 | agent = project_client.agents.get_agent("<%= agentId %>") |
9 | 10 |
|
10 | | -thread = project_client.agents.create_thread() |
| 11 | +thread = project_client.agents.threads.create() |
11 | 12 | print(f"Created thread, ID: {thread.id}") |
12 | 13 |
|
13 | | -message = project_client.agents.create_message( |
| 14 | +message = project_client.agents.messages.create( |
14 | 15 | thread_id=thread.id, |
15 | 16 | role="user", |
16 | 17 | content="<%= userMessage %>" |
17 | 18 | ) |
18 | 19 |
|
19 | | -run = project_client.agents.create_and_process_run( |
| 20 | +run = project_client.agents.runs.create_and_process( |
20 | 21 | thread_id=thread.id, |
21 | 22 | agent_id=agent.id) |
22 | | -messages = project_client.agents.list_messages(thread_id=thread.id) |
23 | 23 |
|
24 | | -for text_message in messages.text_messages: |
25 | | - print(text_message.as_dict()) |
| 24 | +if run.status == "failed": |
| 25 | + print(f"Run failed: {run.last_error}") |
| 26 | +else: |
| 27 | + messages = project_client.agents.messages.list( |
| 28 | + thread_id=thread.id, order=ListSortOrder.ASCENDING) |
| 29 | + |
| 30 | + for message in messages: |
| 31 | + if message.text_messages: |
| 32 | + print(f"{message.role}: {message.text_messages[-1].text.value}") |
0 commit comments