refactor: 所有子图使用公共工具,避免重复造轮子
Some checks failed
构建并部署 AI Agent 服务 / deploy (push) Failing after 6m20s
Some checks failed
构建并部署 AI Agent 服务 / deploy (push) Failing after 6m20s
This commit is contained in:
@@ -1,11 +1,14 @@
|
||||
"""
|
||||
资讯子图节点 - 完善版(使用API客户端)
|
||||
News Analysis Subgraph Nodes - Complete (with API Client)
|
||||
资讯子图节点 - 使用公共工具版本
|
||||
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,
|
||||
@@ -20,25 +23,17 @@ def parse_intent(state: NewsAnalysisState) -> NewsAnalysisState:
|
||||
解析用户意图节点
|
||||
确定用户想做什么操作
|
||||
"""
|
||||
state.current_phase = "intent_parsing"
|
||||
|
||||
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
|
||||
@@ -48,52 +43,36 @@ def query_news(state: NewsAnalysisState) -> NewsAnalysisState:
|
||||
"""
|
||||
查询资讯节点
|
||||
"""
|
||||
state.current_phase = "querying_news"
|
||||
state.current_phase = "executing"
|
||||
|
||||
query = state.user_query
|
||||
# 使用 API 客户端
|
||||
news_items = news_api.query_news(state.user_query)
|
||||
state.news_items = news_items
|
||||
|
||||
# 使用API客户端查询资讯
|
||||
news_data = news_api.query_news_mock(query)
|
||||
|
||||
# 转换为NewsItem对象
|
||||
for news in news_data:
|
||||
state.news_items.append(
|
||||
NewsItem(
|
||||
title=news.get("title", ""),
|
||||
source=news.get("source", ""),
|
||||
summary=news.get("summary", ""),
|
||||
keywords=news.get("keywords", []),
|
||||
author=news.get("author", ""),
|
||||
published_at=news.get("published_at", None)
|
||||
)
|
||||
)
|
||||
|
||||
state.success = True
|
||||
return state
|
||||
|
||||
|
||||
def analyze_url(state: NewsAnalysisState) -> NewsAnalysisState:
|
||||
"""
|
||||
分析资讯URL节点
|
||||
分析 URL 节点
|
||||
"""
|
||||
state.current_phase = "analyzing_url"
|
||||
state.current_phase = "executing"
|
||||
|
||||
urls = state.custom_urls or [state.action_params.get("url", "")]
|
||||
# 从用户输入中提取 URL(简单处理)
|
||||
query = state.user_query
|
||||
url = query
|
||||
for keyword in ["分析", "analyze", "url", "链接"]:
|
||||
url = url.replace(keyword, "").strip()
|
||||
|
||||
# 使用API客户端分析URL
|
||||
for url in urls:
|
||||
if url:
|
||||
result = news_api.analyze_url_mock(url)
|
||||
state.news_items.append(
|
||||
NewsItem(
|
||||
title=result.get("title", ""),
|
||||
source=result.get("source", ""),
|
||||
summary=result.get("summary", ""),
|
||||
keywords=result.get("keywords", [])
|
||||
)
|
||||
)
|
||||
if not url:
|
||||
url = "https://example.com/news/article"
|
||||
|
||||
state.custom_urls = [url]
|
||||
|
||||
# 使用 API 客户端
|
||||
analysis = news_api.analyze_url(url)
|
||||
state.analysis = analysis
|
||||
|
||||
state.success = True
|
||||
return state
|
||||
|
||||
|
||||
@@ -101,14 +80,12 @@ def extract_keywords(state: NewsAnalysisState) -> NewsAnalysisState:
|
||||
"""
|
||||
提取关键词节点
|
||||
"""
|
||||
state.current_phase = "extracting_keywords"
|
||||
state.current_phase = "executing"
|
||||
|
||||
text = state.user_query
|
||||
# 使用 API 客户端
|
||||
keywords = news_api.extract_keywords(state.user_query)
|
||||
state.extracted_keywords = keywords
|
||||
|
||||
# 使用API客户端提取关键词
|
||||
state.extracted_keywords = news_api.extract_keywords_mock(text)
|
||||
|
||||
state.success = True
|
||||
return state
|
||||
|
||||
|
||||
@@ -116,101 +93,76 @@ def generate_report(state: NewsAnalysisState) -> NewsAnalysisState:
|
||||
"""
|
||||
生成报告节点
|
||||
"""
|
||||
state.current_phase = "generating_report"
|
||||
state.current_phase = "executing"
|
||||
|
||||
query = state.user_query
|
||||
# 使用 API 客户端
|
||||
report = news_api.generate_report(state.user_query)
|
||||
state.report_content = report
|
||||
|
||||
# 使用API客户端生成报告
|
||||
state.report_content = news_api.generate_report_mock(query)
|
||||
|
||||
state.success = True
|
||||
return state
|
||||
|
||||
|
||||
def format_result(state: NewsAnalysisState) -> NewsAnalysisState:
|
||||
"""
|
||||
格式化结果节点 - 精美展示
|
||||
格式化结果节点(使用公共工具)
|
||||
"""
|
||||
state.current_phase = "formatting"
|
||||
|
||||
if state.action == NewsAction.QUERY_NEWS and state.news_items:
|
||||
result = []
|
||||
result.append("═══════════════════════════════════════════")
|
||||
result.append("📰 最新资讯")
|
||||
result.append("═══════════════════════════════════════════")
|
||||
result.append("")
|
||||
|
||||
for i, item in enumerate(state.news_items, 1):
|
||||
result.append(f"{i}. {item.title}")
|
||||
result.append(f" 来源:{item.source}")
|
||||
result.append(f" 摘要:{item.summary}")
|
||||
if item.keywords:
|
||||
result.append(f" 🏷️ 关键词:{', '.join(item.keywords)}")
|
||||
result.append("")
|
||||
|
||||
result.append("═══════════════════════════════════════════")
|
||||
result.append("💡 提示:点击资讯查看详情,或生成分析报告")
|
||||
|
||||
state.final_result = "\n".join(result)
|
||||
md = MarkdownFormatter()
|
||||
output_lines = []
|
||||
|
||||
elif state.action == NewsAction.ANALYZE_URL and state.news_items:
|
||||
result = []
|
||||
result.append("═══════════════════════════════════════════")
|
||||
result.append("🔍 资讯分析结果")
|
||||
result.append("═══════════════════════════════════════════")
|
||||
result.append("")
|
||||
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 i, item in enumerate(state.news_items, 1):
|
||||
result.append(f"{i}. {item.title}")
|
||||
result.append(f" {item.summary}")
|
||||
if item.keywords:
|
||||
result.append(f" 🏷️ 关键词:{', '.join(item.keywords)}")
|
||||
result.append("")
|
||||
|
||||
result.append("═══════════════════════════════════════════")
|
||||
|
||||
state.final_result = "\n".join(result)
|
||||
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:
|
||||
result = []
|
||||
result.append("═══════════════════════════════════════════")
|
||||
result.append("🏷️ 提取的关键词")
|
||||
result.append("═══════════════════════════════════════════")
|
||||
result.append("")
|
||||
result.append(" " + ", ".join(state.extracted_keywords))
|
||||
result.append("")
|
||||
result.append("═══════════════════════════════════════════")
|
||||
|
||||
state.final_result = "\n".join(result)
|
||||
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:
|
||||
state.final_result = 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:
|
||||
if not state.final_result:
|
||||
state.final_result = "资讯操作完成"
|
||||
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"
|
||||
|
||||
state.current_phase = "done"
|
||||
return state
|
||||
|
||||
|
||||
def should_continue(state: NewsAnalysisState) -> str:
|
||||
"""
|
||||
条件路由:决定下一步该做什么
|
||||
"""
|
||||
if state.error_message:
|
||||
return "format_result"
|
||||
|
||||
# 根据action路由
|
||||
if state.action == NewsAction.NONE:
|
||||
return "parse_intent"
|
||||
elif state.action == NewsAction.QUERY_NEWS:
|
||||
return "query_news"
|
||||
elif state.action == NewsAction.ANALYZE_URL:
|
||||
return "analyze_url"
|
||||
elif state.action == NewsAction.EXTRACT_KEYWORDS:
|
||||
return "extract_keywords"
|
||||
elif state.action == NewsAction.GENERATE_REPORT:
|
||||
return "generate_report"
|
||||
else:
|
||||
return "format_result"
|
||||
|
||||
Reference in New Issue
Block a user