type fix graphrag sample

microsoft · Jan 9, 2025 · b9b3ec0 · b9b3ec0
1 parent dff93ea
commit b9b3ec0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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}")