fc / apps /litigation_ai /agent /__init__.py
jun2024's picture
Upload 2241 files
b4f2238 verified
Raw History Blame Contribute Delete
1.46 kB
"""
诉讼文书生成 Agent 模块
基于统一 LLM 服务的 Agent 架构,实现智能诉讼文书生成.
主要组件:
- LitigationAgent: Agent 实例,封装 LLM 和工具调用
- LitigationAgentFactory: Agent 工厂,创建和配置 Agent 实例
- LitigationAgentState: Agent 状态定义
- ILitigationAgentService: Agent 服务接口
- IAgentFactory: Agent 工厂接口
- Tools: Agent 工具集
- Middleware: 对话历史和摘要中间件
- Prompts: 系统提示词
"""
from .factory import LitigationAgent, LitigationAgentFactory
from .interfaces import IAgentFactory, ILitigationAgentService, IMemoryMiddleware
from .middleware import LitigationMemoryMiddleware, LitigationSummarizationMiddleware, SummarizationConfig
from .prompts import build_full_prompt, get_system_prompt
from .schemas import AgentResponse, DraftOutput, ToolCallRecord
from .state import LitigationAgentState
from .tools import get_litigation_tools
__all__ = [
# Agent
"LitigationAgent",
# 工厂
"LitigationAgentFactory",
# 状态
"LitigationAgentState",
# 接口
"ILitigationAgentService",
"IAgentFactory",
"IMemoryMiddleware",
# 数据结构
"AgentResponse",
"DraftOutput",
"ToolCallRecord",
# 工具
"get_litigation_tools",
# 中间件
"LitigationMemoryMiddleware",
"LitigationSummarizationMiddleware",
"SummarizationConfig",
# 提示词
"get_system_prompt",
"build_full_prompt",
]