ee7a37ce70
引入了一套全新的、基于规则的Prompt注入系统，以取代原有的 `injection_point` 机制。这套新系统提供了更强大、更灵活的Prompt内容注入能力。 主要变更包括： - **引入 `InjectionRule` 和 `InjectionType`**：定义了注入规则的数据结构和注入类型（如 `PREPEND`, `APPEND`, `REPLACE`, `REMOVE`, `INSERT_AFTER`），允许插件开发者精确控制注入行为。 - **重构 `PromptComponentManager`**：核心逻辑从简单地拼接字符串 (`execute_components_for`) 重构为按优先级应用注入规则 (`apply_injections`)，支持正则表达式匹配和更复杂的注入操作。 - **向后兼容**：`PromptInfo` 中增加了兼容逻辑，能自动将旧的 `injection_point` 定义转换为新的 `injection_rules`，确保现有插件无需立即修改即可正常工作。 - **更新 `BasePrompt`**：废弃了 `injection_point` 属性，并推荐使用新的 `injection_rules` 属性。 - **更新示例插件**：`hello_world_plugin` 已更新，展示了新注入规则的使用方法。 BREAKING CHANGE: `BasePrompt` 中的 `injection_point` 属性已被废弃。虽然目前存在向后兼容逻辑，但未来版本将移除该属性。所有Prompt组件都应迁移至使用 `injection_rules` 以获得更强的控制力和未来的兼容性。
102 lines
3.6 KiB
Python
102 lines
3.6 KiB
Python
from abc import ABC, abstractmethod
|
||
from typing import Any
|
||
|
||
from src.chat.utils.prompt_params import PromptParameters
|
||
from src.common.logger import get_logger
|
||
from src.plugin_system.base.component_types import ComponentType, InjectionRule, PromptInfo
|
||
|
||
logger = get_logger("base_prompt")
|
||
|
||
|
||
class BasePrompt(ABC):
|
||
"""Prompt组件基类
|
||
|
||
Prompt是插件的一种组件类型，用于动态地向现有的核心Prompt模板中注入额外的上下文信息。
|
||
它的主要作用是在不修改核心代码的情况下，扩展和定制模型的行为。
|
||
|
||
子类可以通过类属性定义其行为：
|
||
- prompt_name: Prompt组件的唯一名称。
|
||
- injection_rules: 定义注入规则的列表。
|
||
"""
|
||
|
||
prompt_name: str = ""
|
||
"""Prompt组件的名称"""
|
||
prompt_description: str = ""
|
||
"""Prompt组件的描述"""
|
||
|
||
# 定义此组件希望如何注入到核心Prompt中
|
||
# 这是一个 InjectionRule 对象的列表，可以实现复杂的注入逻辑
|
||
# 例如: [InjectionRule(target_prompt="planner_prompt", injection_type=InjectionType.APPEND, priority=50)]
|
||
injection_rules: list[InjectionRule] = []
|
||
"""定义注入规则的列表"""
|
||
|
||
# 旧的注入点定义，用于向后兼容。如果定义了这个，它将被自动转换为 injection_rules。
|
||
injection_point: str | list[str] | None = None
|
||
"""[已废弃] 要注入的目标Prompt名称或列表，请使用 injection_rules"""
|
||
|
||
def __init__(self, params: PromptParameters, plugin_config: dict | None = None):
|
||
"""初始化Prompt组件
|
||
|
||
Args:
|
||
params: 统一提示词参数，包含所有构建提示词所需的上下文信息。
|
||
plugin_config: 插件配置字典。
|
||
"""
|
||
self.params = params
|
||
self.plugin_config = plugin_config or {}
|
||
self.log_prefix = "[PromptComponent]"
|
||
|
||
logger.debug(f"{self.log_prefix} Prompt组件 '{self.prompt_name}' 初始化完成")
|
||
|
||
@abstractmethod
|
||
async def execute(self) -> str:
|
||
"""执行Prompt生成的抽象方法，子类必须实现。
|
||
|
||
此方法应根据初始化时传入的 `self.params` 来构建并返回一个字符串。
|
||
返回的字符串将被拼接到目标Prompt的最前面。
|
||
|
||
Returns:
|
||
str: 生成的文本内容。
|
||
"""
|
||
pass
|
||
|
||
def get_config(self, key: str, default: Any = None) -> Any:
|
||
"""获取插件配置值，支持嵌套键访问。
|
||
|
||
Args:
|
||
key: 配置键名，使用点号进行嵌套访问，如 "section.subsection.key"。
|
||
default: 未找到键时返回的默认值。
|
||
|
||
Returns:
|
||
Any: 配置值或默认值。
|
||
"""
|
||
if not self.plugin_config:
|
||
return default
|
||
|
||
keys = key.split(".")
|
||
current = self.plugin_config
|
||
for k in keys:
|
||
if isinstance(current, dict) and k in current:
|
||
current = current[k]
|
||
else:
|
||
return default
|
||
return current
|
||
|
||
@classmethod
|
||
def get_prompt_info(cls) -> "PromptInfo":
|
||
"""从类属性生成PromptInfo，用于组件注册和管理。
|
||
|
||
Returns:
|
||
PromptInfo: 生成的Prompt信息对象。
|
||
"""
|
||
if not cls.prompt_name:
|
||
raise ValueError("Prompt组件必须定义 'prompt_name' 类属性。")
|
||
|
||
# 同时传递新旧两种定义，PromptInfo的__post_init__将处理兼容性问题
|
||
return PromptInfo(
|
||
name=cls.prompt_name,
|
||
component_type=ComponentType.PROMPT,
|
||
description=cls.prompt_description,
|
||
injection_rules=cls.injection_rules,
|
||
injection_point=cls.injection_point,
|
||
)
|