refactor: 重构目录结构 - 简化层级

backend/app/subgraphs/news_analysis/nodes.py

@@ -0,0 +1,185 @@
+"""
+资讯子图节点 - 使用公共工具版本
+News Analysis Subgraph Nodes - Using Common Tools
+"""
+
+from typing import Dict, Any
+from datetime import datetime
+
+# 公共工具
+from ..common import MarkdownFormatter
+
+from .state import (
+    NewsAnalysisState,
+    NewsAction,
+    NewsItem,
+    NewsSource
+)
+from .api_client import news_api
+
+
+def parse_intent(state: NewsAnalysisState) -> NewsAnalysisState:
+    """
+    解析用户意图节点
+    确定用户想做什么操作
+    """
+    query_lower = state.user_query.lower()
+    
+    if any(keyword in query_lower for keyword in ["资讯", "新闻", "news", "report"]):
+        state.action = NewsAction.QUERY_NEWS
+    elif any(keyword in query_lower for keyword in ["分析", "analyze", "url", "链接"]):
+        state.action = NewsAction.ANALYZE_URL
+    elif any(keyword in query_lower for keyword in ["关键词", "keyword", "提取"]):
+        state.action = NewsAction.EXTRACT_KEYWORDS
+    elif any(keyword in query_lower for keyword in ["报告", "生成", "generate"]):
+        state.action = NewsAction.GENERATE_REPORT
+    else:
+        state.action = NewsAction.QUERY_NEWS
+    
+    return state
+
+
+def query_news(state: NewsAnalysisState) -> NewsAnalysisState:
+    """
+    查询资讯节点
+    """
+    state.current_phase = "executing"
+    
+    # 使用 API 客户端
+    news_items = news_api.query_news(state.user_query)
+    state.news_items = news_items
+    
+    return state
+
+
+def analyze_url(state: NewsAnalysisState) -> NewsAnalysisState:
+    """
+    分析 URL 节点
+    """
+    state.current_phase = "executing"
+    
+    # 从用户输入中提取 URL（简单处理）
+    query = state.user_query
+    url = query
+    for keyword in ["分析", "analyze", "url", "链接"]:
+        url = url.replace(keyword, "").strip()
+    
+    if not url:
+        url = "https://example.com/news/article"
+    
+    state.custom_urls = [url]
+    
+    # 使用 API 客户端
+    analysis = news_api.analyze_url(url)
+    state.analysis = analysis
+    
+    return state
+
+
+def extract_keywords(state: NewsAnalysisState) -> NewsAnalysisState:
+    """
+    提取关键词节点
+    """
+    state.current_phase = "executing"
+    
+    # 使用 API 客户端
+    keywords = news_api.extract_keywords(state.user_query)
+    state.extracted_keywords = keywords
+    
+    return state
+
+
+def generate_report(state: NewsAnalysisState) -> NewsAnalysisState:
+    """
+    生成报告节点
+    """
+    state.current_phase = "executing"
+    
+    # 使用 API 客户端
+    report = news_api.generate_report(state.user_query)
+    state.report_content = report
+    
+    return state
+
+
+def format_result(state: NewsAnalysisState) -> NewsAnalysisState:
+    """
+    格式化结果节点（使用公共工具）
+    """
+    state.current_phase = "formatting"
+    
+    md = MarkdownFormatter()
+    output_lines = []
+    
+    output_lines.append("┌───────────────────────────────────┐")
+    output_lines.append("│          📰 资讯助手               │")
+    output_lines.append("└───────────────────────────────────┘")
+    output_lines.append("")
+    
+    if state.action == NewsAction.QUERY_NEWS and state.news_items:
+        output_lines.append(md.heading("📰 最新资讯", 2))
+        output_lines.append("")
+        
+        for item in state.news_items:
+            output_lines.append(md.heading(item.title, 3))
+            output_lines.append(f"> 来源: {item.source.value}")
+            output_lines.append(f"> 时间: {item.published_at.strftime('%Y-%m-%d %H:%M')}")
+            if item.summary:
+                output_lines.append("")
+                output_lines.append(item.summary)
+            if item.url:
+                output_lines.append(f"🔗 链接: {md.link(item.title, item.url)}")
+            output_lines.append("")
+    
+    elif state.action == NewsAction.EXTRACT_KEYWORDS and state.extracted_keywords:
+        output_lines.append(md.heading("🏷️ 提取的关键词", 2))
+        output_lines.append("")
+        keywords_data = [
+            {"关键词": k, "权重": f"{w:.2f}"}
+            for k, w in state.extracted_keywords.items()
+        ]
+        output_lines.append(md.table(keywords_data))
+    
+    elif state.action == NewsAction.GENERATE_REPORT and state.report_content:
+        output_lines.append(md.heading("📊 分析报告", 2))
+        output_lines.append("")
+        output_lines.append(state.report_content)
+    
+    elif state.action == NewsAction.ANALYZE_URL and state.analysis:
+        output_lines.append(md.heading("🔍 URL 分析", 2))
+        output_lines.append("")
+        output_lines.append(f"> URL: {state.custom_urls[0]}")
+        output_lines.append("")
+        output_lines.append(state.analysis)
+    
+    else:
+        output_lines.append(md.heading("✨ 操作完成", 2))
+        output_lines.append("您的请求已处理。")
+    
+    # 页脚提示
+    output_lines.append("")
+    output_lines.append("---")
+    output_lines.append("💡 提示：您可以继续查询资讯、提取关键词或者生成报告！")
+    
+    state.final_result = "\n".join(output_lines)
+    state.success = True
+    state.current_phase = "completed"
+
+    return state
+
+
+def should_continue(state: NewsAnalysisState) -> str:
+    """
+    条件路由函数：根据 action 决定下一个节点
+    """
+    action = state.action
+    if action == NewsAction.QUERY:
+        return "query_news"
+    elif action == NewsAction.ANALYZE_URL:
+        return "analyze_url"
+    elif action == NewsAction.EXTRACT_KEYWORDS:
+        return "extract_keywords"
+    elif action == NewsAction.GENERATE_REPORT:
+        return "generate_report"
+    else:
+        return "format_result"