Skip to content

Commit 03d979a

Browse files
authored
Fix samples (Azure#38884)
1 parent 5b7f412 commit 03d979a

File tree

3 files changed

+13
-11
lines changed

3 files changed

+13
-11
lines changed

sdk/ai/azure-ai-projects/samples/agents/async_samples/sample_agents_stream_eventhandler_async.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -77,17 +77,17 @@ async def main() -> None:
7777
)
7878
print(f"Created message, message ID {message.id}")
7979

80-
async with await project_client.agents.create_stream(
81-
thread_id=thread.id, assistant_id=agent.id, event_handler=MyEventHandler()
82-
) as stream:
83-
async for _, _, func_return in stream:
84-
print(func_return)
85-
86-
await project_client.agents.delete_agent(agent.id)
87-
print("Deleted agent")
88-
89-
messages = await project_client.agents.list_messages(thread_id=thread.id)
90-
print(f"Messages: {messages}")
80+
async with await project_client.agents.create_stream(
81+
thread_id=thread.id, assistant_id=agent.id, event_handler=MyEventHandler()
82+
) as stream:
83+
async for _, _, func_return in stream:
84+
print(func_return)
85+
86+
await project_client.agents.delete_agent(agent.id)
87+
print("Deleted agent")
88+
89+
messages = await project_client.agents.list_messages(thread_id=thread.id)
90+
print(f"Messages: {messages}")
9191

9292

9393
if __name__ == "__main__":

sdk/ai/azure-ai-projects/samples/agents/async_samples/sample_agents_stream_eventhandler_with_functions_async.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ class MyEventHandler(AsyncAgentEventHandler[str]):
4343
def __init__(self, functions: AsyncFunctionTool, project_client: AIProjectClient) -> None:
4444
self.functions = functions
4545
self.project_client = project_client
46+
super().__init__()
4647

4748
async def on_message_delta(self, delta: "MessageDeltaChunk") -> None:
4849
print(f"Text delta received: {delta.text}")

sdk/ai/azure-ai-projects/samples/agents/sample_agents_stream_eventhandler_with_functions.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ class MyEventHandler(AgentEventHandler):
4545

4646
def __init__(self, functions: FunctionTool) -> None:
4747
self.functions = functions
48+
super().__init__()
4849

4950
def on_message_delta(self, delta: "MessageDeltaChunk") -> None:
5051
print(f"Text delta received: {delta.text}")

0 commit comments

Comments
 (0)