1491 lines
50 KiB
Python
1491 lines
50 KiB
Python
from PWF.Convention.Runtime.Config import *
|
||
from PWF.CoreModules.plugin_interface import PluginInterface
|
||
from PWF.CoreModules.flags import *
|
||
from PWF.Convention.Runtime.Architecture import Architecture
|
||
from PWF.Convention.Runtime.GlobalConfig import ProjectConfig
|
||
from PWF.Convention.Runtime.Web import ToolURL
|
||
from PWF.Convention.Runtime.String import LimitStringLength
|
||
from fastapi.responses import HTMLResponse
|
||
from dataclasses import dataclass, field
|
||
from typing import Any, Dict, Iterable, List, Optional, Sequence, TypedDict, override, Union
|
||
import httpx
|
||
import re
|
||
from datetime import datetime, timedelta
|
||
from llama_index.core.tools import FunctionTool
|
||
from llama_index.llms.ollama import Ollama
|
||
from llama_index.core.agent.workflow import AgentWorkflow
|
||
from llama_index.core.agent.workflow.react_agent import ReActAgent
|
||
from bs4 import BeautifulSoup
|
||
|
||
logger: ProjectConfig = Architecture.Get(ProjectConfig)
|
||
MAIN_WEBHOOK_URL = logger.FindItem("main_webhook_url", "")
|
||
logger.SaveProperties()
|
||
|
||
def get_target_web_page_url(year:str, month:str, day:str) -> str:
|
||
return fr"http://mrxwlb.com/{year}/{month}/{day}/{year}年{month}月{day}日新闻联播文字版/"
|
||
|
||
ollama_url = "http://www.liubai.site:11434"
|
||
|
||
class NewsAIAgent:
|
||
"""新闻AI智能体 - 基于LlamaIndex和Ollama的工具调用智能体"""
|
||
|
||
def __init__(self, ollama_url: str = "http://www.liubai.site:11434"):
|
||
self.ollama_url = ollama_url
|
||
self.client: Optional[httpx.AsyncClient] = None
|
||
self.llm = Ollama(model="qwen2.5:7b", base_url=ollama_url, request_timeout=600.0)
|
||
self.workflow: Optional[AgentWorkflow] = None
|
||
self._initialize_agent()
|
||
|
||
def _initialize_agent(self):
|
||
"""初始化智能体和工具"""
|
||
# 创建工具函数
|
||
tools = [
|
||
FunctionTool.from_defaults(
|
||
fn=self._get_current_date,
|
||
name="get_current_date",
|
||
description="获取当前日期，返回格式为'年-月-日'，例如'2025-11-19'"
|
||
),
|
||
FunctionTool.from_defaults(
|
||
fn=self._get_yesterday_date,
|
||
name="get_yesterday_date",
|
||
description="获取昨天的日期，返回格式为'年-月-日'，例如'2025-11-18'"
|
||
),
|
||
FunctionTool.from_defaults(
|
||
fn=self._get_news_content,
|
||
name="get_news_content",
|
||
description="获取指定日期的新闻联播文字内容。参数date格式为'年-月-日'，例如'2025-11-19'。返回该日期的新闻内容文本。"
|
||
),
|
||
FunctionTool.from_defaults(
|
||
fn=self._parse_date_from_text,
|
||
name="parse_date_from_text",
|
||
description="从文本中解析日期。支持格式：'2025年11月19日'、'2025-11-19'、'2025/11/19'、'今天'、'昨天'等。返回格式为'年-月-日'"
|
||
),
|
||
]
|
||
|
||
# 系统提示词 - 更清晰明确的指令
|
||
system_prompt = """你是一个新闻分析助手，专门回答关于新闻联播的问题。
|
||
|
||
【重要】你必须按照以下步骤使用工具：
|
||
|
||
步骤1: 确定日期
|
||
- 如果用户问"今天"或"今日"，调用 get_current_date
|
||
- 如果用户问"昨天"或"昨日"，调用 get_yesterday_date
|
||
- 如果用户提到具体日期（如"2025年11月17日"），调用 parse_date_from_text
|
||
|
||
步骤2: 获取新闻内容
|
||
- 拿到日期后，必须调用 get_news_content(date="YYYY-MM-DD") 获取新闻
|
||
- 注意：date参数格式必须是 "年-月-日"，例如 "2025-11-19"
|
||
|
||
步骤3: 回答问题
|
||
- 仔细阅读获取到的新闻内容
|
||
- 基于新闻内容准确回答用户的问题
|
||
- 如果新闻中没有相关信息，明确告知用户
|
||
|
||
【禁止】直接回答问题而不调用工具！你必须先获取新闻内容才能回答。"""
|
||
|
||
# 创建ReActAgent
|
||
agent = ReActAgent(
|
||
llm=self.llm,
|
||
tools=tools,
|
||
verbose=True,
|
||
system_prompt=system_prompt,
|
||
)
|
||
|
||
# 创建AgentWorkflow
|
||
self.workflow = AgentWorkflow(
|
||
agents=[agent],
|
||
timeout=600.0,
|
||
)
|
||
|
||
async def _get_client(self) -> httpx.AsyncClient:
|
||
"""获取HTTP客户端（懒加载）"""
|
||
if self.client is None:
|
||
self.client = httpx.AsyncClient(timeout=120.0)
|
||
return self.client
|
||
|
||
def _get_current_date(self) -> str:
|
||
"""获取当前日期
|
||
|
||
Returns:
|
||
当前日期字符串，格式：年-月-日
|
||
"""
|
||
now = datetime.now()
|
||
return f"{now.year}-{str(now.month).zfill(2)}-{str(now.day).zfill(2)}"
|
||
|
||
def _get_yesterday_date(self) -> str:
|
||
"""获取昨天日期
|
||
|
||
Returns:
|
||
昨天日期字符串，格式：年-月-日
|
||
"""
|
||
yesterday = datetime.now() - timedelta(days=1)
|
||
return f"{yesterday.year}-{str(yesterday.month).zfill(2)}-{str(yesterday.day).zfill(2)}"
|
||
|
||
def _parse_date_from_text(self, text: str) -> str:
|
||
"""从文本中解析日期
|
||
|
||
Args:
|
||
text: 包含日期信息的文本
|
||
|
||
Returns:
|
||
日期字符串，格式：年-月-日
|
||
"""
|
||
# 尝试匹配日期格式
|
||
date_patterns = [
|
||
r'(\d{4})年(\d{1,2})月(\d{1,2})日',
|
||
r'(\d{4})-(\d{1,2})-(\d{1,2})',
|
||
r'(\d{4})/(\d{1,2})/(\d{1,2})',
|
||
]
|
||
|
||
for pattern in date_patterns:
|
||
match = re.search(pattern, text)
|
||
if match:
|
||
year, month, day = match.groups()
|
||
return f"{year}-{month.zfill(2)}-{day.zfill(2)}"
|
||
|
||
# 检查相对日期
|
||
if "今天" in text or "今日" in text:
|
||
return self._get_current_date()
|
||
elif "昨天" in text or "昨日" in text:
|
||
return self._get_yesterday_date()
|
||
|
||
# 默认返回今天
|
||
return self._get_current_date()
|
||
|
||
def _get_news_content(self, date: str) -> str:
|
||
"""获取指定日期的新闻内容（带缓存）
|
||
|
||
Args:
|
||
date: 日期字符串，格式：年-月-日，例如'2025-11-19'
|
||
|
||
Returns:
|
||
新闻文字内容
|
||
"""
|
||
try:
|
||
# 解析日期
|
||
parts = date.split('-')
|
||
if len(parts) != 3:
|
||
return f"日期格式错误，请使用'年-月-日'格式，例如'2025-11-19'"
|
||
|
||
year, month, day = parts
|
||
# 去掉前导零（某些网站URL格式要求）
|
||
month = str(int(month))
|
||
day = str(int(day))
|
||
|
||
# 检查缓存
|
||
cache_key = f"news_cache/{year}/{month}/{day}"
|
||
cached_file = ProjectConfig().GetFile(cache_key, False)
|
||
|
||
if cached_file.Exists():
|
||
cached_content = cached_file.LoadAsText()
|
||
logger.Log("Info", f"从缓存加载新闻: {date}")
|
||
return cached_content
|
||
|
||
# 如果没有缓存，则抓取网页
|
||
logger.Log("Info", f"从网页抓取新闻: {date}")
|
||
url = get_target_web_page_url(year, month, day)
|
||
|
||
# 使用同步方式获取（因为工具函数需要同步）
|
||
import requests
|
||
|
||
# 直接使用中文URL，让requests自动处理编码
|
||
url = f"http://mrxwlb.com/{year}/{month}/{day}/{year}年{month}月{day}日新闻联播文字版/"
|
||
|
||
# 添加浏览器请求头，模拟真实浏览器访问
|
||
headers = {
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/142.0.0.0 Safari/537.36',
|
||
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8',
|
||
'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
|
||
'Accept-Encoding': 'gzip, deflate',
|
||
'Connection': 'keep-alive',
|
||
'Upgrade-Insecure-Requests': '1',
|
||
}
|
||
|
||
logger.Log("Info", f"请求URL: {url}")
|
||
response = requests.get(url, headers=headers, timeout=30, allow_redirects=True)
|
||
response.raise_for_status()
|
||
|
||
# 使用BeautifulSoup解析HTML
|
||
soup = BeautifulSoup(response.text, 'html.parser')
|
||
|
||
# 移除script和style标签
|
||
for script in soup(["script", "style"]):
|
||
script.decompose()
|
||
|
||
# 获取文本内容
|
||
content = soup.get_text()
|
||
# 清理空白字符
|
||
lines = (line.strip() for line in content.splitlines())
|
||
chunks = (phrase.strip() for line in lines for phrase in line.split(" "))
|
||
content = ' '.join(chunk for chunk in chunks if chunk)
|
||
|
||
if not content or len(content) < 100:
|
||
return f"未能获取到{year}年{month}月{day}日的新闻内容"
|
||
|
||
# 保存到缓存
|
||
ProjectConfig().GetFile(cache_key, True).SaveAsText(content)
|
||
logger.Log("Info", f"新闻内容已缓存: {date}")
|
||
|
||
return content
|
||
|
||
except Exception as e:
|
||
logger.Log("Error", f"获取新闻失败: {e}")
|
||
return f"获取新闻时出错: {str(e)}"
|
||
|
||
async def answer_question(self, query: str) -> str:
|
||
"""根据问题回答新闻内容
|
||
|
||
Args:
|
||
query: 用户问题
|
||
|
||
Returns:
|
||
AI生成的回答
|
||
"""
|
||
try:
|
||
logger.Log("Info", f"="*50)
|
||
logger.Log("Info", f"用户提问: {query}")
|
||
logger.Log("Info", f"使用模型: {self.llm.model}")
|
||
logger.Log("Info", f"="*50)
|
||
|
||
# 使用workflow运行agent（注意参数是user_msg）
|
||
result = await self.workflow.run(user_msg=query)
|
||
|
||
logger.Log("Info", f"Agent执行完成，结果类型: {type(result)}")
|
||
|
||
# 提取回答内容
|
||
if hasattr(result, 'response'):
|
||
answer = str(result.response)
|
||
logger.Log("Info", f"从result.response提取答案")
|
||
elif hasattr(result, 'message'):
|
||
answer = str(result.message)
|
||
logger.Log("Info", f"从result.message提取答案")
|
||
else:
|
||
answer = str(result)
|
||
logger.Log("Info", f"直接转换result为字符串")
|
||
|
||
logger.Log("Info", f"最终答案长度: {len(answer)} 字符")
|
||
|
||
return answer
|
||
|
||
except Exception as e:
|
||
logger.Log("Error", f"回答问题失败: {e}")
|
||
import traceback
|
||
error_trace = traceback.format_exc()
|
||
logger.Log("Error", f"详细错误:\n{error_trace}")
|
||
return f"处理问题时出错: {str(e)}"
|
||
|
||
async def close(self):
|
||
"""关闭客户端"""
|
||
if self.client:
|
||
await self.client.aclose()
|
||
self.client = None
|
||
|
||
class GuideEntry(TypedDict, total=False):
|
||
"""单条图鉴信息。"""
|
||
|
||
title: str
|
||
identifier: str
|
||
description: str
|
||
category: str
|
||
metadata: Dict[str, str]
|
||
icon: str
|
||
badge: str
|
||
links: Sequence[Dict[str, str]]
|
||
tags: Sequence[str]
|
||
details: Sequence[Union[str, Dict[str, Any]]]
|
||
group: str
|
||
|
||
|
||
@dataclass(frozen=True)
|
||
class GuideSection:
|
||
"""图鉴章节。"""
|
||
|
||
title: str
|
||
entries: Sequence[GuideEntry] = field(default_factory=tuple)
|
||
description: str = ""
|
||
layout: str = "grid"
|
||
section_id: str | None = None
|
||
|
||
|
||
@dataclass(frozen=True)
|
||
class GuidePage:
|
||
"""完整图鉴页面。"""
|
||
|
||
title: str
|
||
sections: Sequence[GuideSection] = field(default_factory=tuple)
|
||
subtitle: str = ""
|
||
metadata: Dict[str, str] = field(default_factory=dict)
|
||
related_links: Dict[str, Sequence[Dict[str, str]]] = field(default_factory=dict)
|
||
|
||
|
||
def render_markdown_page(page: GuidePage) -> str:
|
||
"""保留 Markdown 渲染（备用）。"""
|
||
|
||
def _render_section(section: GuideSection) -> str:
|
||
lines: List[str] = [f"## {section.title}"]
|
||
if section.description:
|
||
lines.append(section.description)
|
||
if not section.entries:
|
||
lines.append("> 暂无内容。")
|
||
return "\n".join(lines)
|
||
for entry in section.entries:
|
||
title = entry.get("title", "未命名")
|
||
identifier = entry.get("identifier")
|
||
desc = entry.get("description", "")
|
||
category = entry.get("category")
|
||
metadata = entry.get("metadata", {})
|
||
bullet = f"- **{title}**"
|
||
if identifier:
|
||
bullet += f"｜`{identifier}`"
|
||
if category:
|
||
bullet += f"｜{category}"
|
||
lines.append(bullet)
|
||
if desc:
|
||
lines.append(f" - {desc}")
|
||
for meta_key, meta_value in metadata.items():
|
||
lines.append(f" - {meta_key}：{meta_value}")
|
||
return "\n".join(lines)
|
||
|
||
lines: List[str] = [f"# {page.title}"]
|
||
if page.subtitle:
|
||
lines.append(page.subtitle)
|
||
if page.metadata:
|
||
lines.append("")
|
||
for key, value in page.metadata.items():
|
||
lines.append(f"- {key}：{value}")
|
||
for section in page.sections:
|
||
lines.append("")
|
||
lines.append(_render_section(section))
|
||
return "\n".join(lines)
|
||
|
||
|
||
def render_html_page(page: GuidePage) -> str:
|
||
"""渲染 Apple Store 风格的 HTML 页面。"""
|
||
|
||
def escape(text: Optional[str]) -> str:
|
||
if not text:
|
||
return ""
|
||
return (
|
||
text.replace("&", "&")
|
||
.replace("<", "<")
|
||
.replace(">", ">")
|
||
.replace('"', """)
|
||
)
|
||
|
||
def render_metadata(metadata: Dict[str, str]) -> str:
|
||
if not metadata:
|
||
return ""
|
||
cards = []
|
||
for key, value in metadata.items():
|
||
cards.append(
|
||
f"""
|
||
<div class="meta-card">
|
||
<div class="meta-key">{escape(key)}</div>
|
||
<div class="meta-value">{escape(value)}</div>
|
||
</div>
|
||
"""
|
||
)
|
||
return f'<section class="meta-grid">{"".join(cards)}</section>'
|
||
|
||
def render_links(links: Optional[Sequence[Dict[str, str]]]) -> str:
|
||
if not links:
|
||
return ""
|
||
items = []
|
||
for link in links:
|
||
href = escape(link.get("href", "#"))
|
||
label = escape(link.get("label", "前往"))
|
||
items.append(f'<a class="entry-link" href="{href}" target="_blank">{label}</a>')
|
||
return "".join(items)
|
||
|
||
def render_tags(tags: Optional[Sequence[str]]) -> str:
|
||
if not tags:
|
||
return ""
|
||
chips = "".join(f'<span class="entry-tag">{escape(tag)}</span>' for tag in tags)
|
||
return f'<div class="entry-tags-extra">{chips}</div>'
|
||
|
||
def render_details(details: Optional[Sequence[Union[str, Dict[str, Any]]]]) -> str:
|
||
if not details:
|
||
return ""
|
||
blocks: List[str] = []
|
||
for detail in details:
|
||
if isinstance(detail, str):
|
||
blocks.append(f'<p class="entry-detail-paragraph">{escape(detail)}</p>')
|
||
elif isinstance(detail, dict):
|
||
kind = detail.get("type")
|
||
if kind == "list":
|
||
items = "".join(
|
||
f'<li>{escape(str(item))}</li>' for item in detail.get("items", [])
|
||
)
|
||
blocks.append(f'<ul class="entry-detail-list">{items}</ul>')
|
||
elif kind == "steps":
|
||
items = "".join(
|
||
f'<li><span class="step-index">{idx+1}</span><span>{escape(str(item))}</span></li>'
|
||
for idx, item in enumerate(detail.get("items", []))
|
||
)
|
||
blocks.append(f'<ol class="entry-detail-steps">{items}</ol>')
|
||
elif kind == "table":
|
||
rows = []
|
||
for row in detail.get("rows", []):
|
||
cols = "".join(f"<td>{escape(str(col))}</td>" for col in row)
|
||
rows.append(f"<tr>{cols}</tr>")
|
||
head = ""
|
||
headers = detail.get("headers")
|
||
if headers:
|
||
head = "".join(f"<th>{escape(str(col))}</th>" for col in headers)
|
||
head = f"<thead><tr>{head}</tr></thead>"
|
||
blocks.append(
|
||
f'<table class="entry-detail-table">{head}<tbody>{"".join(rows)}</tbody></table>'
|
||
)
|
||
if not blocks:
|
||
return ""
|
||
return f'<div class="entry-details">{"".join(blocks)}</div>'
|
||
|
||
def render_entry(entry: GuideEntry) -> str:
|
||
icon = escape(entry.get("icon"))
|
||
badge = escape(entry.get("badge"))
|
||
title = escape(entry.get("title"))
|
||
identifier = escape(entry.get("identifier"))
|
||
description = escape(entry.get("description"))
|
||
category = escape(entry.get("category"))
|
||
metadata_items = []
|
||
for meta_key, meta_value in entry.get("metadata", {}).items():
|
||
metadata_items.append(
|
||
f'<li><span>{escape(meta_key)}</span><span>{escape(str(meta_value))}</span></li>'
|
||
)
|
||
metadata_html = ""
|
||
if metadata_items:
|
||
metadata_html = f'<ul class="entry-meta">{"".join(metadata_items)}</ul>'
|
||
identifier_html = f'<code class="entry-id">{identifier}</code>' if identifier else ""
|
||
category_html = f'<span class="entry-category">{category}</span>' if category else ""
|
||
badge_html = f'<span class="entry-badge">{badge}</span>' if badge else ""
|
||
icon_html = f'<div class="entry-icon">{icon}</div>' if icon else ""
|
||
links_html = render_links(entry.get("links"))
|
||
tags_html = render_tags(entry.get("tags"))
|
||
details_html = render_details(entry.get("details"))
|
||
group = escape(entry.get("group"))
|
||
group_attr = f' data-group="{group}"' if group else ""
|
||
return f"""
|
||
<article class="entry-card"{group_attr}>
|
||
{icon_html}
|
||
<div class="entry-content">
|
||
<header>
|
||
<h3>{title}{badge_html}</h3>
|
||
<div class="entry-tags">{identifier_html}{category_html}</div>
|
||
</header>
|
||
<p class="entry-desc">{description}</p>
|
||
{metadata_html}
|
||
{tags_html}
|
||
{details_html}
|
||
{links_html}
|
||
</div>
|
||
</article>
|
||
"""
|
||
|
||
def render_section(section: GuideSection) -> str:
|
||
layout_class = "entries-grid" if section.layout == "grid" else "entries-list"
|
||
section_attr = f' id="{escape(section.section_id)}"' if section.section_id else ""
|
||
cards = "".join(render_entry(entry) for entry in section.entries)
|
||
description_html = (
|
||
f'<p class="section-desc">{escape(section.description)}</p>'
|
||
if section.description
|
||
else ""
|
||
)
|
||
if not cards:
|
||
cards = '<div class="empty-placeholder">暂无内容</div>'
|
||
return f"""
|
||
<section class="guide-section"{section_attr}>
|
||
<div class="section-header">
|
||
<h2>{escape(section.title)}</h2>
|
||
{description_html}
|
||
</div>
|
||
<div class="{layout_class}">
|
||
{cards}
|
||
</div>
|
||
</section>
|
||
"""
|
||
|
||
def render_related(related: Dict[str, Sequence[Dict[str, str]]]) -> str:
|
||
if not related:
|
||
return ""
|
||
blocks: List[str] = []
|
||
for label, links in related.items():
|
||
if not links:
|
||
continue
|
||
items = "".join(
|
||
f'<a class="related-link" href="{escape(link.get("href", "#"))}">{escape(link.get("label", ""))}</a>'
|
||
for link in links
|
||
)
|
||
blocks.append(
|
||
f"""
|
||
<div class="related-block">
|
||
<div class="related-label">{escape(label)}</div>
|
||
<div class="related-items">{items}</div>
|
||
</div>
|
||
"""
|
||
)
|
||
if not blocks:
|
||
return ""
|
||
return f'<section class="related-section">{"".join(blocks)}</section>'
|
||
|
||
sections_html = "".join(render_section(section) for section in page.sections)
|
||
metadata_html = render_metadata(page.metadata)
|
||
related_html = render_related(page.related_links)
|
||
subtitle_html = f'<p class="hero-subtitle">{escape(page.subtitle)}</p>' if page.subtitle else ""
|
||
|
||
return f"""
|
||
<!DOCTYPE html>
|
||
<html lang="zh-CN">
|
||
<head>
|
||
<meta charset="utf-8" />
|
||
<meta name="viewport" content="width=device-width, initial-scale=1" />
|
||
<title>{escape(page.title)}</title>
|
||
<style>
|
||
:root {{
|
||
color-scheme: dark;
|
||
--bg-primary: radial-gradient(120% 120% at 10% 10%, #222 0%, #050505 100%);
|
||
--bg-card: rgba(255, 255, 255, 0.06);
|
||
--bg-card-hover: rgba(255, 255, 255, 0.12);
|
||
--border-soft: rgba(255, 255, 255, 0.1);
|
||
--text-main: #f5f5f7;
|
||
--text-sub: rgba(245, 245, 247, 0.64);
|
||
--accent: linear-gradient(135deg, #4b7bec, #34e7e4);
|
||
--accent-strong: linear-gradient(135deg, #ff9f1a, #ff3f34);
|
||
}}
|
||
* {{
|
||
box-sizing: border-box;
|
||
}}
|
||
body {{
|
||
margin: 0;
|
||
font-family: "SF Pro Display", "SF Pro SC", "PingFang SC", "Microsoft YaHei", sans-serif;
|
||
background: var(--bg-primary);
|
||
color: var(--text-main);
|
||
min-height: 100vh;
|
||
padding: 0 0 64px;
|
||
}}
|
||
a {{
|
||
color: #9fc9ff;
|
||
text-decoration: none;
|
||
}}
|
||
a:hover {{
|
||
text-decoration: underline;
|
||
}}
|
||
header.hero {{
|
||
padding: 80px 24px 40px;
|
||
text-align: center;
|
||
position: relative;
|
||
}}
|
||
header.hero::after {{
|
||
content: "";
|
||
position: absolute;
|
||
inset: 0;
|
||
background: radial-gradient(circle at 50% -20%, rgba(255, 255, 255, 0.18), transparent 55%);
|
||
pointer-events: none;
|
||
z-index: -1;
|
||
}}
|
||
.hero-title {{
|
||
font-size: clamp(32px, 5vw, 48px);
|
||
font-weight: 700;
|
||
margin: 0;
|
||
background: var(--accent);
|
||
-webkit-background-clip: text;
|
||
color: transparent;
|
||
letter-spacing: 0.8px;
|
||
}}
|
||
.hero-subtitle {{
|
||
margin: 16px auto 0;
|
||
max-width: 640px;
|
||
font-size: 18px;
|
||
color: var(--text-sub);
|
||
line-height: 1.6;
|
||
}}
|
||
main {{
|
||
width: min(1100px, 92vw);
|
||
margin: 0 auto;
|
||
}}
|
||
.meta-grid {{
|
||
display: grid;
|
||
gap: 20px;
|
||
margin: 0 auto 48px;
|
||
padding: 0 8px;
|
||
grid-template-columns: repeat(auto-fit, minmax(180px, 1fr));
|
||
}}
|
||
.related-section {{
|
||
margin: 0 auto 48px;
|
||
padding: 0 8px;
|
||
display: grid;
|
||
gap: 18px;
|
||
}}
|
||
.related-block {{
|
||
display: grid;
|
||
gap: 8px;
|
||
}}
|
||
.related-label {{
|
||
font-size: 13px;
|
||
text-transform: uppercase;
|
||
letter-spacing: 0.5px;
|
||
color: var(--text-sub);
|
||
}}
|
||
.related-items {{
|
||
display: flex;
|
||
gap: 10px;
|
||
flex-wrap: wrap;
|
||
}}
|
||
.related-link {{
|
||
display: inline-flex;
|
||
align-items: center;
|
||
padding: 6px 14px;
|
||
border-radius: 999px;
|
||
background: rgba(255, 255, 255, 0.08);
|
||
font-size: 13px;
|
||
letter-spacing: 0.4px;
|
||
transition: background 0.2s ease;
|
||
}}
|
||
.related-link:hover {{
|
||
background: rgba(255, 255, 255, 0.16);
|
||
text-decoration: none;
|
||
}}
|
||
.meta-card {{
|
||
border-radius: 18px;
|
||
padding: 22px;
|
||
border: 1px solid var(--border-soft);
|
||
background: rgba(255, 255, 255, 0.04);
|
||
backdrop-filter: blur(16px);
|
||
box-shadow: 0 12px 30px rgba(0, 0, 0, 0.35);
|
||
}}
|
||
.meta-key {{
|
||
font-size: 13px;
|
||
letter-spacing: 1px;
|
||
text-transform: uppercase;
|
||
color: var(--text-sub);
|
||
}}
|
||
.meta-value {{
|
||
margin-top: 6px;
|
||
font-size: 22px;
|
||
font-weight: 600;
|
||
}}
|
||
.guide-section {{
|
||
margin: 0 auto 56px;
|
||
padding: 0 8px;
|
||
}}
|
||
.section-header {{
|
||
display: flex;
|
||
flex-direction: column;
|
||
gap: 12px;
|
||
margin-bottom: 28px;
|
||
}}
|
||
.section-header h2 {{
|
||
margin: 0;
|
||
font-size: clamp(26px, 3vw, 32px);
|
||
letter-spacing: 0.5px;
|
||
}}
|
||
.section-desc {{
|
||
margin: 0;
|
||
font-size: 16px;
|
||
color: var(--text-sub);
|
||
max-width: 640px;
|
||
}}
|
||
.entries-grid {{
|
||
display: grid;
|
||
gap: 20px;
|
||
grid-template-columns: repeat(auto-fit, minmax(260px, 1fr));
|
||
}}
|
||
.entries-list {{
|
||
display: flex;
|
||
flex-direction: column;
|
||
gap: 18px;
|
||
}}
|
||
.entry-card {{
|
||
display: flex;
|
||
gap: 18px;
|
||
border-radius: 20px;
|
||
padding: 24px;
|
||
border: 1px solid transparent;
|
||
background: var(--bg-card);
|
||
transition: border 0.2s ease, background 0.2s ease, transform 0.2s ease;
|
||
}}
|
||
.entry-card:hover {{
|
||
background: var(--bg-card-hover);
|
||
border-color: rgba(255, 255, 255, 0.18);
|
||
transform: translateY(-4px);
|
||
}}
|
||
.entry-icon {{
|
||
font-size: 36px;
|
||
}}
|
||
.entry-content {{
|
||
flex: 1;
|
||
}}
|
||
.entry-content header {{
|
||
display: flex;
|
||
flex-direction: column;
|
||
gap: 8px;
|
||
margin-bottom: 10px;
|
||
}}
|
||
.entry-content h3 {{
|
||
margin: 0;
|
||
font-size: 20px;
|
||
display: inline-flex;
|
||
align-items: center;
|
||
gap: 10px;
|
||
}}
|
||
.entry-badge {{
|
||
display: inline-flex;
|
||
padding: 4px 10px;
|
||
border-radius: 999px;
|
||
font-size: 12px;
|
||
letter-spacing: 0.5px;
|
||
background: var(--accent-strong);
|
||
}}
|
||
.entry-tags {{
|
||
display: flex;
|
||
gap: 12px;
|
||
flex-wrap: wrap;
|
||
}}
|
||
.entry-tags-extra {{
|
||
display: flex;
|
||
gap: 8px;
|
||
flex-wrap: wrap;
|
||
margin-top: 10px;
|
||
}}
|
||
.entry-id {{
|
||
background: rgba(255, 255, 255, 0.1);
|
||
border-radius: 999px;
|
||
padding: 4px 10px;
|
||
font-size: 12px;
|
||
letter-spacing: 0.5px;
|
||
}}
|
||
.entry-category {{
|
||
font-size: 12px;
|
||
color: var(--text-sub);
|
||
}}
|
||
.entry-desc {{
|
||
margin: 0 0 12px;
|
||
color: var(--text-sub);
|
||
line-height: 1.6;
|
||
}}
|
||
.entry-meta {{
|
||
list-style: none;
|
||
margin: 0;
|
||
padding: 0;
|
||
display: grid;
|
||
gap: 6px;
|
||
font-size: 13px;
|
||
color: var(--text-sub);
|
||
}}
|
||
.entry-meta li {{
|
||
display: flex;
|
||
justify-content: space-between;
|
||
}}
|
||
.entry-link {{
|
||
display: inline-flex;
|
||
align-items: center;
|
||
gap: 6px;
|
||
margin-top: 10px;
|
||
font-size: 14px;
|
||
}}
|
||
.entry-tag {{
|
||
display: inline-flex;
|
||
align-items: center;
|
||
padding: 4px 12px;
|
||
border-radius: 999px;
|
||
background: rgba(255, 255, 255, 0.08);
|
||
font-size: 12px;
|
||
letter-spacing: 0.4px;
|
||
}}
|
||
.entry-details {{
|
||
margin-top: 14px;
|
||
display: grid;
|
||
gap: 10px;
|
||
}}
|
||
.entry-detail-paragraph {{
|
||
margin: 0;
|
||
color: var(--text-sub);
|
||
line-height: 1.6;
|
||
}}
|
||
.entry-detail-list, .entry-detail-steps {{
|
||
margin: 0;
|
||
padding-left: 20px;
|
||
color: var(--text-sub);
|
||
}}
|
||
.entry-detail-steps {{
|
||
list-style: none;
|
||
padding-left: 0;
|
||
}}
|
||
.entry-detail-steps li {{
|
||
display: grid;
|
||
grid-template-columns: 28px 1fr;
|
||
align-items: start;
|
||
gap: 10px;
|
||
margin-bottom: 6px;
|
||
}}
|
||
.entry-detail-steps .step-index {{
|
||
width: 28px;
|
||
height: 28px;
|
||
border-radius: 50%;
|
||
display: inline-flex;
|
||
align-items: center;
|
||
justify-content: center;
|
||
background: rgba(255, 255, 255, 0.12);
|
||
font-size: 12px;
|
||
}}
|
||
.entry-detail-table {{
|
||
width: 100%;
|
||
border-collapse: collapse;
|
||
border-radius: 14px;
|
||
overflow: hidden;
|
||
}}
|
||
.entry-detail-table th,
|
||
.entry-detail-table td {{
|
||
border: 1px solid rgba(255, 255, 255, 0.1);
|
||
padding: 8px 12px;
|
||
font-size: 13px;
|
||
text-align: left;
|
||
}}
|
||
.entry-detail-table th {{
|
||
background: rgba(255, 255, 255, 0.08);
|
||
font-weight: 600;
|
||
}}
|
||
.entry-detail-table tr:nth-child(even) {{
|
||
background: rgba(255, 255, 255, 0.04);
|
||
}}
|
||
.empty-placeholder {{
|
||
padding: 40px;
|
||
text-align: center;
|
||
border-radius: 20px;
|
||
border: 1px dashed rgba(255, 255, 255, 0.2);
|
||
color: var(--text-sub);
|
||
}}
|
||
@media (max-width: 680px) {{
|
||
.entry-card {{
|
||
flex-direction: column;
|
||
}}
|
||
.entry-content h3 {{
|
||
font-size: 18px;
|
||
}}
|
||
.meta-grid {{
|
||
grid-template-columns: 1fr 1fr;
|
||
}}
|
||
}}
|
||
@media (max-width: 480px) {{
|
||
.meta-grid {{
|
||
grid-template-columns: 1fr;
|
||
}}
|
||
}}
|
||
</style>
|
||
</head>
|
||
<body>
|
||
<header class="hero">
|
||
<h1 class="hero-title">{escape(page.title)}</h1>
|
||
{subtitle_html}
|
||
</header>
|
||
<main>
|
||
{metadata_html}
|
||
{related_html}
|
||
{sections_html}
|
||
</main>
|
||
</body>
|
||
</html>
|
||
"""
|
||
|
||
|
||
class MessageSender:
|
||
"""消息发送器"""
|
||
|
||
def __init__(self, webhook_url: str):
|
||
"""初始化消息发送器
|
||
|
||
Args:
|
||
webhook_url: Webhook URL
|
||
"""
|
||
self.webhook_url = webhook_url
|
||
self.client: Optional[httpx.AsyncClient] = None
|
||
|
||
async def _get_client(self) -> httpx.AsyncClient:
|
||
"""获取HTTP客户端（懒加载）"""
|
||
if self.client is None:
|
||
self.client = httpx.AsyncClient(timeout=10.0)
|
||
return self.client
|
||
|
||
async def send_message(self, message: Dict[str, Any]) -> bool:
|
||
"""发送消息到WPS
|
||
|
||
Args:
|
||
message: 消息字典
|
||
|
||
Returns:
|
||
是否发送成功
|
||
"""
|
||
try:
|
||
client = await self._get_client()
|
||
response = await client.post(self.webhook_url, json=message)
|
||
|
||
if response.status_code == 200:
|
||
logger.Log("Info", f"消息发送成功: {message.get('msgtype')}")
|
||
return True
|
||
else:
|
||
logger.Log("Error", f"消息发送失败: status={response.status_code}, body={response.text}")
|
||
return False
|
||
|
||
except Exception as e:
|
||
logger.Log("Error", f"发送消息异常: {e}")
|
||
return False
|
||
|
||
async def send_text(self, content: str, at_user_id: Optional[int] = None) -> bool:
|
||
"""发送文本消息
|
||
|
||
Args:
|
||
content: 文本内容
|
||
at_user_id: @用户ID（可选）
|
||
|
||
Returns:
|
||
是否发送成功
|
||
"""
|
||
# 如果需要@人
|
||
if at_user_id:
|
||
content = f'<at user_id="{at_user_id}"></at> {content}'
|
||
|
||
message = {
|
||
"msgtype": "text",
|
||
"text": {
|
||
"content": content
|
||
}
|
||
}
|
||
return await self.send_message(message)
|
||
|
||
async def send_markdown(self, text: str) -> bool:
|
||
"""发送Markdown消息
|
||
|
||
Args:
|
||
text: Markdown文本
|
||
|
||
Returns:
|
||
是否发送成功
|
||
"""
|
||
message = {
|
||
"msgtype": "markdown",
|
||
"markdown": {
|
||
"text": text
|
||
}
|
||
}
|
||
return await self.send_message(message)
|
||
|
||
async def send_link(self, title: str, text: str,
|
||
message_url: str = "", btn_title: str = "查看详情") -> bool:
|
||
"""发送链接消息
|
||
|
||
Args:
|
||
title: 标题
|
||
text: 文本内容
|
||
message_url: 跳转URL
|
||
btn_title: 按钮文字
|
||
|
||
Returns:
|
||
是否发送成功
|
||
"""
|
||
message = {
|
||
"msgtype": "link",
|
||
"link": {
|
||
"title": title,
|
||
"text": text,
|
||
"messageUrl": message_url,
|
||
"btnTitle": btn_title
|
||
}
|
||
}
|
||
return await self.send_message(message)
|
||
|
||
async def close(self):
|
||
"""关闭HTTP客户端"""
|
||
if self.client:
|
||
await self.client.aclose()
|
||
self.client = None
|
||
|
||
|
||
class BasicWPSInterface(PluginInterface):
|
||
user_id_to_username: Optional[Callable[[int], str]] = None
|
||
|
||
@override
|
||
def is_enable_plugin(self) -> bool:
|
||
return False
|
||
|
||
def get_webhook_url(self, message: str, chat_id: int, user_id: int) -> str:
|
||
'''
|
||
根据消息和用户ID获取Webhook URL, 返回空字符串表示不需要回复消息
|
||
Args:
|
||
message: 消息内容
|
||
chat_id: 聊天ID
|
||
user_id: 用户ID
|
||
Returns:
|
||
Webhook URL
|
||
'''
|
||
return ""
|
||
|
||
def get_webhook_request(self, data:Any|None) -> None:
|
||
pass
|
||
|
||
def get_message_sender(self, webhook_url: str) -> MessageSender:
|
||
return MessageSender(webhook_url)
|
||
|
||
def get_message_sender_type(self) -> Literal["text", "markdown", "link"]:
|
||
return "markdown"
|
||
|
||
def get_message_sender_function(self, webhook_url: str, type: Literal["text", "markdown", "link"]) -> Coroutine[Any, Any, bool]:
|
||
if type == "text":
|
||
return self.get_message_sender(webhook_url).send_text
|
||
elif type == "markdown":
|
||
return self.get_message_sender(webhook_url).send_markdown
|
||
elif type == "link":
|
||
return self.get_message_sender(webhook_url).send_link
|
||
else:
|
||
raise ValueError(f"Invalid message sender type: {type}")
|
||
|
||
def parse_message_after_at(self, message: str) -> str:
|
||
'''
|
||
已过时
|
||
'''
|
||
return message
|
||
|
||
async def send_markdown_message(self, message: str, chat_id: int, user_id: int) -> str|None:
|
||
webhook_url = self.get_webhook_url(message, chat_id, user_id)
|
||
if get_internal_debug():
|
||
logger.Log("Info", f"Webhook URL: {webhook_url}, Message: {LimitStringLength(message)}, Chat ID: {chat_id}, User ID: {user_id}")
|
||
if webhook_url == "" or webhook_url == None:
|
||
return None
|
||
|
||
username: str = self.user_id_to_username(user_id) if self.user_id_to_username else ""
|
||
send_message = f"""## <at user_id="{user_id}">{username}</at>
|
||
---
|
||
{message}
|
||
"""
|
||
|
||
result = await self.get_message_sender_function(webhook_url, self.get_message_sender_type())(send_message)
|
||
if get_internal_verbose():
|
||
logger.Log("Info", f"Webhook URL: {webhook_url}, Message: {LimitStringLength(message)}, Result: {result}")
|
||
return None
|
||
|
||
@override
|
||
async def callback(self, message: str, chat_id: int, user_id: int) -> str|None:
|
||
message = self.parse_message_after_at(message)
|
||
if message == "":
|
||
return None
|
||
return await self.send_markdown_message(message, chat_id, user_id)
|
||
|
||
|
||
class WPSAPI(BasicWPSInterface):
|
||
"""核心 WPS 插件基类，提供图鉴模板设施。"""
|
||
|
||
guide_section_labels: Dict[str, str] = {
|
||
"commands": "指令一览",
|
||
"items": "物品与资源",
|
||
"recipes": "配方与合成",
|
||
"guides": "系统指引",
|
||
}
|
||
|
||
def get_guide_title(self) -> str:
|
||
return self.__class__.__name__
|
||
|
||
def get_guide_subtitle(self) -> str:
|
||
return ""
|
||
|
||
def get_guide_metadata(self) -> Dict[str, str]:
|
||
return {}
|
||
|
||
def collect_command_entries(self) -> Sequence[GuideEntry]:
|
||
return ()
|
||
|
||
def collect_item_entries(self) -> Sequence[GuideEntry]:
|
||
return ()
|
||
|
||
def collect_recipe_entries(self) -> Sequence[GuideEntry]:
|
||
return ()
|
||
|
||
def collect_guide_entries(self) -> Sequence[GuideEntry]:
|
||
return ()
|
||
|
||
def collect_additional_sections(self) -> Sequence[GuideSection]:
|
||
return ()
|
||
|
||
def collect_guide_sections(self) -> Sequence[GuideSection]:
|
||
sections: List[GuideSection] = []
|
||
|
||
command_entries = tuple(self.collect_command_entries())
|
||
if command_entries:
|
||
sections.append(
|
||
GuideSection(
|
||
title=self.guide_section_labels["commands"],
|
||
entries=command_entries,
|
||
layout="list",
|
||
)
|
||
)
|
||
|
||
item_entries = tuple(self.collect_item_entries())
|
||
if item_entries:
|
||
sections.append(
|
||
GuideSection(
|
||
title=self.guide_section_labels["items"],
|
||
entries=item_entries,
|
||
)
|
||
)
|
||
|
||
recipe_entries = tuple(self.collect_recipe_entries())
|
||
if recipe_entries:
|
||
sections.append(
|
||
GuideSection(
|
||
title=self.guide_section_labels["recipes"],
|
||
entries=recipe_entries,
|
||
)
|
||
)
|
||
|
||
guide_entries = tuple(self.collect_guide_entries())
|
||
if guide_entries:
|
||
sections.append(
|
||
GuideSection(
|
||
title=self.guide_section_labels["guides"],
|
||
entries=guide_entries,
|
||
layout="list",
|
||
)
|
||
)
|
||
|
||
additional_sections = tuple(self.collect_additional_sections())
|
||
if additional_sections:
|
||
sections.extend(additional_sections)
|
||
|
||
return tuple(sections)
|
||
|
||
def build_guide_page(self) -> GuidePage:
|
||
metadata: Dict[str, str] = {}
|
||
for key, value in self.get_guide_metadata().items():
|
||
metadata[key] = str(value)
|
||
|
||
related = self.get_related_links()
|
||
|
||
return GuidePage(
|
||
title=self.get_guide_title(),
|
||
subtitle=self.get_guide_subtitle(),
|
||
sections=self.collect_guide_sections(),
|
||
metadata=metadata,
|
||
related_links=related,
|
||
)
|
||
|
||
def render_guide_page(self, page: GuidePage) -> str:
|
||
return render_html_page(page)
|
||
|
||
def render_guide(self) -> str:
|
||
return self.render_guide_page(self.build_guide_page())
|
||
|
||
def get_guide_response(self, content: str) -> HTMLResponse:
|
||
return HTMLResponse(content)
|
||
|
||
@override
|
||
def generate_router_illustrated_guide(self):
|
||
async def handler() -> HTMLResponse:
|
||
return self.get_guide_response(self.render_guide())
|
||
|
||
return handler
|
||
|
||
def get_related_links(self) -> Dict[str, Sequence[Dict[str, str]]]:
|
||
links: Dict[str, Sequence[Dict[str, str]]] = {}
|
||
|
||
parents = []
|
||
for base in self.__class__.__mro__[1:]:
|
||
if not issubclass(base, WPSAPI):
|
||
continue
|
||
if base.__module__.startswith("Plugins."):
|
||
parents.append(base)
|
||
if base is WPSAPI:
|
||
break
|
||
if parents:
|
||
parents_links = [self._build_class_link(cls) for cls in reversed(parents)]
|
||
links["父类链"] = tuple(filter(None, parents_links))
|
||
|
||
child_links = [self._build_class_link(child) for child in self._iter_subclasses(self.__class__)]
|
||
child_links = [link for link in child_links if link]
|
||
if child_links:
|
||
links["子类"] = tuple(child_links)
|
||
|
||
return links
|
||
|
||
def _build_class_link(self, cls: type) -> Optional[Dict[str, str]]:
|
||
if not hasattr(cls, "__module__") or not cls.__module__.startswith("Plugins."):
|
||
return None
|
||
path = f"/api/{cls.__name__}"
|
||
return {
|
||
"label": cls.__name__,
|
||
"href": path,
|
||
}
|
||
|
||
def _iter_subclasses(self, cls: type) -> List[type]:
|
||
collected: List[type] = []
|
||
for subclass in cls.__subclasses__():
|
||
if not issubclass(subclass, WPSAPI):
|
||
continue
|
||
if not subclass.__module__.startswith("Plugins."):
|
||
continue
|
||
collected.append(subclass)
|
||
collected.extend(self._iter_subclasses(subclass))
|
||
return collected
|
||
|
||
def get_guide_subtitle(self) -> str:
|
||
return "核心 Webhook 转发插件"
|
||
|
||
def get_guide_metadata(self) -> Dict[str, str]:
|
||
return {
|
||
"Webhook 状态": "已配置" if MAIN_WEBHOOK_URL else "未配置",
|
||
}
|
||
|
||
def collect_command_entries(self) -> Sequence[GuideEntry]:
|
||
return (
|
||
{
|
||
"title": "say",
|
||
"identifier": "say",
|
||
"description": "将后续消息内容以 Markdown 形式发送到主 Webhook。",
|
||
"metadata": {"别名": "说"},
|
||
"icon": "🗣️",
|
||
"badge": "核心",
|
||
},
|
||
)
|
||
|
||
def collect_guide_entries(self) -> Sequence[GuideEntry]:
|
||
return (
|
||
{
|
||
"title": "Webhook 绑定",
|
||
"description": (
|
||
"在项目配置中设置 `main_webhook_url` 后插件自动启用，"
|
||
"所有注册的命令将调用 `send_markdown_message` 发送富文本。"
|
||
),
|
||
"icon": "🔗",
|
||
},
|
||
{
|
||
"title": "消息格式",
|
||
"description": (
|
||
"默认使用 Markdown 模式发送，支持 `聊天ID` 与 `用户ID` 的 @ 提醒。"
|
||
),
|
||
"icon": "📝",
|
||
},
|
||
)
|
||
|
||
@override
|
||
def is_enable_plugin(self) -> bool:
|
||
if MAIN_WEBHOOK_URL == "":
|
||
logger.Log("Error", f"{ConsoleFrontColor.RED}WPSAPI未配置主Webhook URL{ConsoleFrontColor.RESET}")
|
||
return MAIN_WEBHOOK_URL != ""
|
||
|
||
def get_main_webhook_url(self) -> str:
|
||
return MAIN_WEBHOOK_URL
|
||
|
||
@override
|
||
def get_webhook_url(self, message: str, chat_id: int, user_id: int) -> str:
|
||
webhook_url = ProjectConfig().GetFile(f"webhook_url/{chat_id}",True).LoadAsText()
|
||
if webhook_url == "":
|
||
webhook_url = self.get_main_webhook_url()
|
||
return webhook_url
|
||
|
||
@override
|
||
def get_webhook_request(self, data:Any|None) -> None:
|
||
return None
|
||
|
||
@override
|
||
def wake_up(self) -> None:
|
||
logger.Log("Info", f"{ConsoleFrontColor.GREEN}WPSAPI核心插件已加载{ConsoleFrontColor.RESET}")
|
||
self.register_plugin("say")
|
||
self.register_plugin("说")
|
||
|
||
class WPSAPIHelp(WPSAPI):
|
||
@override
|
||
def dependencies(self) -> List[Type]:
|
||
return [WPSAPI]
|
||
|
||
@override
|
||
def is_enable_plugin(self) -> bool:
|
||
return True
|
||
|
||
@override
|
||
async def callback(self, message: str, chat_id: int, user_id: int) -> str|None:
|
||
mapper: Dict[str, List[str]] = {}
|
||
for key in PluginInterface.plugin_instances.keys():
|
||
plugin = PluginInterface.plugin_instances.get(key)
|
||
if plugin:
|
||
if plugin.__class__.__name__ not in mapper:
|
||
mapper[plugin.__class__.__name__] = []
|
||
mapper[plugin.__class__.__name__].append(key)
|
||
desc = ""
|
||
for plugin_name, keys in mapper.items():
|
||
desc += f"- {plugin_name}: {", ".join(keys)}\n"
|
||
return await self.send_markdown_message(f"""# 指令入口
|
||
{desc}
|
||
""", chat_id, user_id)
|
||
|
||
@override
|
||
def wake_up(self) -> None:
|
||
logger.Log("Info", f"{ConsoleFrontColor.GREEN}WPSAPIHelp 插件已加载{ConsoleFrontColor.RESET}")
|
||
self.register_plugin("help")
|
||
self.register_plugin("帮助")
|
||
|
||
class WPSAPIWebhook(WPSAPI):
|
||
@override
|
||
def dependencies(self) -> List[Type]:
|
||
return [WPSAPI]
|
||
|
||
@override
|
||
def is_enable_plugin(self) -> bool:
|
||
return True
|
||
|
||
@override
|
||
def wake_up(self) -> None:
|
||
logger.Log("Info", f"{ConsoleFrontColor.GREEN}WPSAPIWebhook 插件已加载{ConsoleFrontColor.RESET}")
|
||
self.register_plugin("chat_url_register")
|
||
self.register_plugin("会话注册")
|
||
|
||
@override
|
||
async def callback(self, message: str, chat_id: int, user_id: int) -> str|None:
|
||
ProjectConfig().GetFile(f"webhook_url/{chat_id}",True).SaveAsText(message)
|
||
return await self.send_markdown_message(f"会话注册成功", chat_id, user_id)
|
||
|
||
class NewsAIPlugin(WPSAPI):
|
||
"""新闻AI智能问答插件"""
|
||
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.ai_agent = NewsAIAgent(ollama_url)
|
||
|
||
@override
|
||
def dependencies(self) -> List[Type]:
|
||
return [WPSAPI]
|
||
|
||
@override
|
||
def is_enable_plugin(self) -> bool:
|
||
return True
|
||
|
||
def get_guide_title(self) -> str:
|
||
return "新闻AI智能问答"
|
||
|
||
def get_guide_subtitle(self) -> str:
|
||
return "基于AI的新闻联播智能问答系统"
|
||
|
||
def get_guide_metadata(self) -> Dict[str, str]:
|
||
return {
|
||
"AI模型": "qwen3:0.6b",
|
||
"数据源": "每日新闻联播",
|
||
"功能": "智能问答",
|
||
}
|
||
|
||
def collect_command_entries(self) -> Sequence[GuideEntry]:
|
||
return (
|
||
{
|
||
"title": "问新闻",
|
||
"identifier": "ask_news",
|
||
"description": "询问新闻内容，支持自动识别日期或查询今天/昨天的新闻。",
|
||
"metadata": {"别名": "新闻问答, 问"},
|
||
"icon": "🤖",
|
||
"badge": "AI",
|
||
"details": [
|
||
{
|
||
"type": "list",
|
||
"items": [
|
||
"支持日期格式：2024年11月19日、2024-11-19、2024/11/19",
|
||
"支持相对日期：今天、昨天、今日、昨日",
|
||
"自动使用AI分析新闻内容并回答问题",
|
||
"示例：问 今天有什么重要新闻？",
|
||
"示例：问 2024年11月19日 经济相关的新闻有哪些？",
|
||
]
|
||
}
|
||
]
|
||
},
|
||
{
|
||
"title": "新闻摘要",
|
||
"identifier": "news_summary",
|
||
"description": "获取指定日期新闻的AI摘要。",
|
||
"metadata": {"别名": "摘要"},
|
||
"icon": "📝",
|
||
"badge": "AI",
|
||
},
|
||
)
|
||
|
||
def collect_guide_entries(self) -> Sequence[GuideEntry]:
|
||
return (
|
||
{
|
||
"title": "智能问答",
|
||
"description": (
|
||
"使用AI理解用户问题，从新闻内容中提取相关信息进行回答。"
|
||
"支持自然语言提问，可以询问特定主题、人物、事件等。"
|
||
),
|
||
"icon": "💬",
|
||
},
|
||
{
|
||
"title": "日期识别",
|
||
"description": (
|
||
"自动从问题中识别日期，支持多种格式。"
|
||
"如果未指定日期，默认查询当天新闻。"
|
||
),
|
||
"icon": "📅",
|
||
},
|
||
{
|
||
"title": "内容抓取",
|
||
"description": (
|
||
"自动从新闻网站抓取指定日期的新闻联播文字版内容。"
|
||
),
|
||
"icon": "🌐",
|
||
},
|
||
)
|
||
|
||
@override
|
||
def wake_up(self) -> None:
|
||
logger.Log("Info", f"{ConsoleFrontColor.GREEN}NewsAIPlugin 新闻AI智能问答插件已加载{ConsoleFrontColor.RESET}")
|
||
self.register_plugin("default")
|
||
|
||
@override
|
||
async def callback(self, message: str, chat_id: int, user_id: int) -> str|None:
|
||
"""处理用户问题"""
|
||
try:
|
||
if not message or message.strip() == "":
|
||
help_text = """# 📰 新闻AI智能问答使用帮助
|
||
|
||
**直接提问即可，智能体会自动：**
|
||
1. 识别你想查询的日期
|
||
2. 获取对应日期的新闻内容
|
||
3. 基于新闻内容回答你的问题
|
||
|
||
**支持的日期格式：**
|
||
- 今天、昨天、今日、昨日
|
||
- 2025年11月19日
|
||
- 2025-11-19
|
||
- 2025/11/19
|
||
|
||
**示例问题：**
|
||
- `今天有什么重要新闻？`
|
||
- `2025年11月17日有什么新闻？`
|
||
- `昨天的新闻中有关于经济的内容吗？`
|
||
- `今天习近平主席有什么活动？`
|
||
- `请总结今天新闻联播的主要内容`
|
||
|
||
**特性：**
|
||
✨ 智能理解问题意图
|
||
🤖 自动调用工具获取信息
|
||
💾 新闻内容自动缓存
|
||
🚀 基于LlamaIndex + Ollama"""
|
||
return await self.send_markdown_message(help_text, chat_id, user_id)
|
||
|
||
# 使用智能体回答问题
|
||
answer = await self.ai_agent.answer_question(message)
|
||
|
||
# 格式化返回结果
|
||
formatted_answer = f"""📰 **新闻AI智能问答**
|
||
|
||
{answer}
|
||
|
||
---
|
||
*由 LlamaIndex + Ollama 驱动*"""
|
||
|
||
return await self.send_markdown_message(formatted_answer, chat_id, user_id)
|
||
|
||
except Exception as e:
|
||
logger.Log("Error", f"新闻AI问答异常: {e}")
|
||
import traceback
|
||
error_detail = traceback.format_exc()
|
||
logger.Log("Error", f"详细错误: {error_detail}")
|
||
error_msg = f"""❌ **处理问题时出错**
|
||
|
||
错误信息：{str(e)}
|
||
|
||
请稍后重试或联系管理员。"""
|
||
return await self.send_markdown_message(error_msg, chat_id, user_id)
|
||
|
||
logger.SaveProperties() |