前段执行query时报错

[YF-learn]

{'type': 'query', 'query': '请分析位于E:\\Desktop\\github_pg\\youtu-agent\\examples\\data_analysis\\demo_data_cat_breeds_clean.csv的猫品种数据，提取有价值的信息。'}
ERROR:root:Error processing query: 'NoneType' object has no attribute 'query'
ERROR:root:Traceback (most recent call last):
File "E:\Desktop\github_pg\youtu-agent\utu\ui\webui_agents.py", line 275, in _handle_query
await self._handle_query_noexcept(query)
File "E:\Desktop\github_pg\youtu-agent\utu\ui\webui_agents.py", line 165, in _handle_query_noexcept
if query.query.strip() == "":
^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'query'

[lllljie]

{'type': 'query', 'query': '请分析位于E:\\Desktop\\github_pg\\youtu-agent\\examples\\data_analysis\\demo_data_cat_breeds_clean.csv的猫品种数据，提取有价值的信息。'} ERROR:root:Error processing query: 'NoneType' object has no attribute 'query' ERROR:root:Traceback (most recent call last): File "E:\Desktop\github_pg\youtu-agent\utu\ui\webui_agents.py", line 275, in _handle_query await self._handle_query_noexcept(query) File "E:\Desktop\github_pg\youtu-agent\utu\ui\webui_agents.py", line 165, in _handle_query_noexcept if query.query.strip() == "": ^^^^^^^^^^^ AttributeError: 'NoneType' object has no attribute 'query'

emm，试着改一下？
async def on_message(self, message: str):
try:
data = json.loads(message)
print(f"DEBUG on_message: Raw data: {data}")

        # 手动处理，不依赖 UserRequest 模型
        if data.get('type') == 'query':
            # 直接从 data 中获取 query
            query_str = data.get('query', '')
            if not query_str and isinstance(data.get('content'), dict):
                query_str = data.get('content', {}).get('query', '')
            
            print(f"DEBUG on_message: Extracted query string: '{query_str}'")
            
            if not query_str.strip():
                await self._handle_error("Query cannot be empty")
                return
            
            # 直接创建 UserQuery 对象放入队列
            user_query = UserQuery(query=query_str)
            print(f"DEBUG on_message: Created UserQuery: {user_query}")
            await self.query_queue.put(user_query)