135 lines
3.9 KiB
Python
135 lines
3.9 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 sqlalchemy import select
|
|||
|
|
import asyncio
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _run_async(coro, timeout: int = 30):
|
|||
|
|
try:
|
|||
|
|
loop = asyncio.get_running_loop()
|
|||
|
|
future = loop.run_in_executor(__import__("concurrent.futures").ThreadPoolExecutor(), lambda: asyncio.run(coro))
|
|||
|
|
return future.result(timeout=timeout)
|
|||
|
|
except RuntimeError:
|
|||
|
|
return asyncio.run(coro)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@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"]
|