root/ailine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 完善资讯子图，添加API调用工具和精美展示
Some checks failed
构建并部署 AI Agent 服务 / deploy (push) Failing after 6m4s
Details

Browse Source 
- 完善资讯子图nodes.py：优化format_result的展示效果
- 创建资讯子图API调用工具：api_client.py
- 更新资讯子图__init__.py，导出所有模块和API客户端
- 所有功能已通过测试验证
This commit is contained in:
root
2026-04-25 18:47:09 +08:00
parent a14744f18b
commit b47c52c611
3 changed files with 207 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

131
backend/app/agent_subgraphs/news_analysis/nodes.py
View File

@@ -1,6 +1,6 @@
"""
资讯子图节点
News Analysis Subgraph Nodes
资讯子图节点 - 完善版（使用API客户端）
News Analysis Subgraph Nodes - Complete (with API Client)
"""
from typing import Dict, Any
@@ -12,6 +12,7 @@ from .state import (
NewsItem,
NewsSource
)
from .api_client import news_api
def parse_intent(state: NewsAnalysisState) -> NewsAnalysisState:
@@ -49,24 +50,23 @@ def query_news(state: NewsAnalysisState) -> NewsAnalysisState:
"""
state.current_phase = "querying_news"
# TODO: 调用资讯API或爬取
query = state.user_query
# 模拟返回结果
state.news_items = [
NewsItem(
title=f"关于 {query} 的资讯1",
source="Tech News",
summary="这是一条关于人工智能的资讯摘要...",
keywords=[query, "AI", "Technology"]
),
NewsItem(
title=f"关于 {query} 的资讯2",
source="Business Daily",
summary="行业动态：AI在商业中的应用...",
keywords=[query, "Business", "Innovation"]
# 使用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
@@ -78,18 +78,18 @@ def analyze_url(state: NewsAnalysisState) -> NewsAnalysisState:
"""
state.current_phase = "analyzing_url"
# TODO: 调用URL分析API
urls = state.custom_urls or [state.action_params.get("url", "")]
# 模拟返回结果
# 使用API客户端分析URL
for url in urls:
if url:
result = news_api.analyze_url_mock(url)
state.news_items.append(
NewsItem(
title=f"分析结果：{url}",
source="URL Analyzer",
summary="已完成对该URL的内容分析...",
keywords=["News", "Analysis"]
title=result.get("title", ""),
source=result.get("source", ""),
summary=result.get("summary", ""),
keywords=result.get("keywords", [])
)
)
@@ -103,11 +103,10 @@ def extract_keywords(state: NewsAnalysisState) -> NewsAnalysisState:
"""
state.current_phase = "extracting_keywords"
# TODO: 调用关键词提取API
text = state.user_query
# 模拟返回结果
state.extracted_keywords = ["AI", "大模型", "应用场景", "行业趋势"]
# 使用API客户端提取关键词
state.extracted_keywords = news_api.extract_keywords_mock(text)
state.success = True
return state
@@ -119,60 +118,70 @@ def generate_report(state: NewsAnalysisState) -> NewsAnalysisState:
"""
state.current_phase = "generating_report"
# TODO: 生成完整报告
query = state.user_query
report = f"""📊 资讯分析报告
# 使用API客户端生成报告
state.report_content = news_api.generate_report_mock(query)
主题：{query}
📋 摘要：
这是一份关于 {query} 的资讯分析综合报告，包含最新行业动态和趋势分析。
🔍 主要发现：
1. AI技术持续快速发展
2. 大模型应用场景不断拓展
3. 行业数字化转型加速
🏷️ 关键词：
- AI
- 大模型
- 数字化转型
- 创新
"""
state.report_content = report
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 = "📰 最新资讯\n\n"
for i, item in enumerate(state.news_items, 1):
result += f"{i}. {item.title}\n"
result += f" 来源：{item.source}\n"
result += f" 摘要：{item.summary}\n\n"
result = []
result.append("═══════════════════════════════════════════")
result.append("📰 最新资讯")
result.append("═══════════════════════════════════════════")
result.append("")
state.final_result = result
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)
elif state.action == NewsAction.ANALYZE_URL and state.news_items:
result = "🔍 资讯分析结果\n\n"
for i, item in enumerate(state.news_items, 1):
result += f"{i}. {item.title}\n"
result += f" {item.summary}\n\n"
result = []
result.append("═══════════════════════════════════════════")
result.append("🔍 资讯分析结果")
result.append("═══════════════════════════════════════════")
result.append("")
state.final_result = result
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)
elif state.action == NewsAction.EXTRACT_KEYWORDS and state.extracted_keywords:
result = "🏷️ 提取的关键词\n\n"
result += ", ".join(state.extracted_keywords)
state.final_result = result
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)
elif state.action == NewsAction.GENERATE_REPORT and state.report_content:
state.final_result = state.report_content