root
a14744f18b
Some checks failed
构建并部署 AI Agent 服务 / deploy (push) Failing after 6m5s
- 完善词典子图：添加生词本功能 - 创建API调用工具：dictionary_api - 添加前端格式化展示工具：result_formatter.py - 创建通讯录和资讯子图的基本结构 - 更新主图状态结构，添加MainGraphState - 添加subgraph_builder.py用于子图集成
90 lines
2.3 KiB
Python
90 lines
2.3 KiB
Python
"""
|
|
资讯子图状态定义
|
|
News Analysis Subgraph State Definition
|
|
"""
|
|
|
|
from enum import Enum, auto
|
|
from typing import Optional, Dict, List, Any
|
|
from dataclasses import dataclass, field
|
|
|
|
|
|
class NewsAction(Enum):
|
|
"""资讯操作类型"""
|
|
NONE = auto()
|
|
QUERY_NEWS = auto() # 查询资讯
|
|
ANALYZE_URL = auto() # 分析资讯
|
|
GENERATE_REPORT = auto() # 生成报告
|
|
FETCH_FROM_SOURCES = auto() # 从指定源获取
|
|
EXTRACT_KEYWORDS = auto() # 提取关键词
|
|
|
|
|
|
@dataclass
|
|
class NewsItem:
|
|
"""资讯条目"""
|
|
title: str = ""
|
|
url: str = ""
|
|
source: str = ""
|
|
content: str = ""
|
|
author: str = ""
|
|
published_at: Optional[str] = None
|
|
summary: str = ""
|
|
keywords: List[str] = field(default_factory=list)
|
|
sentiment: float = 0.0 # 情感分析得分
|
|
metadata: Dict[str, Any] = field(default_factory=dict)
|
|
|
|
|
|
@dataclass
|
|
class NewsSource:
|
|
"""资讯源"""
|
|
name: str = ""
|
|
url: str = ""
|
|
type: str = "" # rss, website, api
|
|
enabled: bool = True
|
|
last_fetched_at: Optional[str] = None
|
|
metadata: Dict[str, Any] = field(default_factory=dict)
|
|
|
|
|
|
@dataclass
|
|
class NewsAnalysisState:
|
|
"""资讯子图状态"""
|
|
# ========== 输入 ==========
|
|
user_query: str = "" # 用户查询
|
|
user_id: str = "" # 用户ID
|
|
|
|
# 操作控制
|
|
action: NewsAction = NewsAction.NONE
|
|
action_params: Dict[str, Any] = field(default_factory=dict)
|
|
|
|
# 源配置
|
|
use_follow_list: bool = False
|
|
custom_urls: List[str] = field(default_factory=list)
|
|
|
|
# ========== 执行过程 ==========
|
|
current_phase: str = "init" # init, fetching, analyzing, done
|
|
current_source_index: int = 0
|
|
primary_fetched: bool = False
|
|
|
|
# 源列表
|
|
sources: List[NewsSource] = field(default_factory=list)
|
|
|
|
# 资讯条目
|
|
news_items: List[NewsItem] = field(default_factory=list)
|
|
|
|
# 关键词
|
|
extracted_keywords: List[str] = field(default_factory=list)
|
|
|
|
# 报告
|
|
report_content: str = ""
|
|
|
|
# ========== 结果 ==========
|
|
success: bool = False
|
|
error_message: str = ""
|
|
final_result: str = ""
|
|
result_data: Dict[str, Any] = field(default_factory=dict)
|
|
|
|
# ========== 元数据 ==========
|
|
start_time: Optional[str] = None
|
|
end_time: Optional[str] = None
|
|
duration: float = 0.0
|
|
debug_info: Dict[str, Any] = field(default_factory=dict)
|