From b9b3ec0f5218bfc139d9de9771d410e979f4b009 Mon Sep 17 00:00:00 2001 From: Leonardo Pinheiro Date: Thu, 9 Jan 2025 20:51:28 +1000 Subject: [PATCH] type fix graphrag sample --- python/samples/agentchat_graphrag/app.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/samples/agentchat_graphrag/app.py b/python/samples/agentchat_graphrag/app.py index bd43126744f0..736aed159d47 100644 --- a/python/samples/agentchat_graphrag/app.py +++ b/python/samples/agentchat_graphrag/app.py @@ -1,4 +1,5 @@ import asyncio +from autogen_agentchat.messages import TextMessage from autogen_ext.models.openai import AzureOpenAIChatCompletionClient from autogen_ext.tools.graphrag import ( GlobalSearchTool, @@ -57,7 +58,7 @@ async def main(): response_stream = team.run_stream(task=query) async for msg in response_stream: - if hasattr(msg, "content"): + if isinstance(msg, TextMessage) and hasattr(msg, "content"): print(f"\nAgent response: {msg.content}")