WIN-JHFT4D3SIVT\caoxiaozhu
4972b4e6b1
Align the L3 graph, agent service, and sync tool shims on one canonical continuity contract so clarification resumes and persisted snapshots behave consistently. Add targeted regressions and hardening notes covering system-message coalescing, async bridge usage, and continuity rehydration.
130 lines
3.8 KiB
Python
130 lines
3.8 KiB
Python
"""Agent 工具集 - 论坛相关"""
|
||
|
||
from langchain_core.tools import tool
|
||
from app.database import async_session
|
||
from app.models.forum import ForumPost, ForumReply
|
||
from app.agents.context import get_current_user
|
||
from app.agents.tools.async_bridge import run_async
|
||
from sqlalchemy import select
|
||
|
||
|
||
def _run_async(coro, timeout: int = 30):
|
||
return run_async(coro, timeout=timeout)
|
||
|
||
|
||
@tool
|
||
def get_forum_posts(category: str | None = None, limit: int = 10) -> str:
|
||
"""
|
||
获取论坛帖子列表。
|
||
|
||
Args:
|
||
category: 可选，筛选分类 (discussion/instruction/question)
|
||
limit: 返回数量，默认10
|
||
|
||
Returns:
|
||
帖子列表
|
||
"""
|
||
uid = get_current_user()
|
||
|
||
async def _get():
|
||
async with async_session() as db:
|
||
from app.models.user import User
|
||
query = (
|
||
select(ForumPost)
|
||
.join(User, User.id == ForumPost.user_id)
|
||
.where(User.id == uid)
|
||
)
|
||
if category:
|
||
query = query.where(ForumPost.category == category)
|
||
query = query.order_by(ForumPost.created_at.desc()).limit(limit)
|
||
result = await db.execute(query)
|
||
posts = result.scalars().all()
|
||
if not posts:
|
||
return "暂无帖子"
|
||
lines = []
|
||
for p in posts:
|
||
exec_mark = " [已执行]" if p.is_executed else ""
|
||
lines.append(
|
||
f"- [{p.id[:8]}] [{p.category}] {p.title} | "
|
||
f"{p.content[:50]}...{exec_mark}"
|
||
)
|
||
return "\n".join(lines)
|
||
|
||
try:
|
||
return _run_async(_get())
|
||
except Exception as e:
|
||
return f"获取帖子失败: {str(e)}"
|
||
|
||
|
||
@tool
|
||
def create_forum_post(title: str, content: str, category: str = "discussion") -> str:
|
||
"""
|
||
在论坛发布新帖子。
|
||
|
||
Args:
|
||
title: 帖子标题
|
||
content: 帖子内容
|
||
category: 分类 (discussion/instruction/question)，默认discussion
|
||
|
||
Returns:
|
||
创建结果
|
||
"""
|
||
uid = get_current_user()
|
||
|
||
async def _create():
|
||
async with async_session() as db:
|
||
post = ForumPost(
|
||
user_id=uid,
|
||
title=title,
|
||
content=content,
|
||
category=category,
|
||
)
|
||
db.add(post)
|
||
await db.commit()
|
||
await db.refresh(post)
|
||
return f"帖子发布成功: [{post.id[:8]}] {title}"
|
||
|
||
try:
|
||
return _run_async(_create())
|
||
except Exception as e:
|
||
return f"发布帖子失败: {str(e)}"
|
||
|
||
|
||
@tool
|
||
def scan_forum_for_instructions() -> str:
|
||
"""
|
||
扫描论坛中的指令类帖子，检查是否有待执行的指令。
|
||
|
||
Returns:
|
||
待执行指令的列表
|
||
"""
|
||
uid = get_current_user()
|
||
|
||
async def _scan():
|
||
async with async_session() as db:
|
||
from app.models.user import User
|
||
result = await db.execute(
|
||
select(ForumPost)
|
||
.join(User, User.id == ForumPost.user_id)
|
||
.where(ForumPost.user_id == uid)
|
||
.where(ForumPost.category == "instruction")
|
||
.where(ForumPost.is_executed == False)
|
||
.order_by(ForumPost.created_at.desc())
|
||
.limit(10)
|
||
)
|
||
posts = result.scalars().all()
|
||
if not posts:
|
||
return "暂无待执行的指令"
|
||
lines = ["待执行的指令:"]
|
||
for p in posts:
|
||
lines.append(f"- [{p.id[:8]}] {p.title}\n 内容: {p.content[:100]}...")
|
||
return "\n".join(lines)
|
||
|
||
try:
|
||
return _run_async(_scan())
|
||
except Exception as e:
|
||
return f"扫描论坛失败: {str(e)}"
|
||
|
||
|
||
__all__ = ["get_forum_posts", "create_forum_post", "scan_forum_for_instructions"]
|