update

app.py

@@ -40,11 +40,20 @@ def stream_words_mcp(query: str):
         tokens = loop.run_until_complete(collect_tokens())
         loop.close()
         
-        # 返回结果给MCP客户端
-        return f"成功生成 {len(tokens)} 个英文单词: {', '.join(tokens[:10])}{'...' if len(tokens) > 10 else ''}"
+        # 返回结果给MCP客户端 - 返回全部单词
+        return f"成功生成 {len(tokens)} 个英文单词: {tokens}"
     except Exception as e:
         return f"生成单词时出错: {str(e)}"
 
+async def stream_words_mcp_stream(query: str):
+    """MCP工具：流式返回英文单词"""
+    try:
+        async for token in stream_words(query):
+            if token.strip():
+                yield token
+    except Exception as e:
+        yield f"生成单词时出错: {str(e)}"
+
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🧠 English Words Generator")
     gr.Markdown("输入你的查询，AI 将为你生成 200 个英文单词（流式输出）")
@@ -112,6 +121,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     # 注册MCP工具 - 使用包装函数并正确配置输出
     demo.load(stream_words_mcp, inputs=None, outputs=gr.Textbox(label="MCP结果", visible=False))
+    # 注册流式MCP工具
+    demo.load(stream_words_mcp_stream, inputs=None, outputs=gr.Textbox(label="流式MCP结果", visible=False))
 
 # 开启 MCP server
 demo.launch(mcp_server=True, server_port=7860)