feat:拆分HFC组件,为插件做准备
This commit is contained in:
@@ -14,15 +14,15 @@ class BaseProcessor(ABC):
|
||||
支持处理InfoBase和Observation类型的输入。
|
||||
"""
|
||||
|
||||
log_prefix = "Base信息处理器"
|
||||
|
||||
@abstractmethod
|
||||
def __init__(self):
|
||||
"""初始化处理器"""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
async def process_info(
|
||||
self,
|
||||
infos: List[InfoBase],
|
||||
observations: Optional[List[Observation]] = None,
|
||||
running_memorys: Optional[List[Dict]] = None,
|
||||
**kwargs: Any,
|
||||
|
||||
@@ -12,7 +12,7 @@ from typing import Dict
|
||||
from src.chat.models.utils_model import LLMRequest
|
||||
from src.config.config import global_config
|
||||
|
||||
logger = get_logger("observation")
|
||||
logger = get_logger("processor")
|
||||
|
||||
|
||||
class ChattingInfoProcessor(BaseProcessor):
|
||||
@@ -21,12 +21,14 @@ class ChattingInfoProcessor(BaseProcessor):
|
||||
用于处理Observation对象,将其转换为ObsInfo对象。
|
||||
"""
|
||||
|
||||
log_prefix = "聊天信息处理"
|
||||
|
||||
def __init__(self):
|
||||
"""初始化观察处理器"""
|
||||
super().__init__()
|
||||
self.llm_summary = LLMRequest(
|
||||
model=global_config.llm_observation, temperature=0.7, max_tokens=300, request_type="chat_observation"
|
||||
)
|
||||
super().__init__()
|
||||
|
||||
async def process_info(
|
||||
self,
|
||||
|
||||
@@ -24,17 +24,16 @@ from src.chat.focus_chat.info_processors.processor_utils import (
|
||||
from typing import Dict
|
||||
from src.chat.focus_chat.info.info_base import InfoBase
|
||||
|
||||
logger = get_logger("sub_heartflow")
|
||||
logger = get_logger("processor")
|
||||
|
||||
|
||||
def init_prompt():
|
||||
# --- Group Chat Prompt ---
|
||||
group_prompt = """
|
||||
你的名字是{bot_name}
|
||||
{memory_str}
|
||||
{extra_info}
|
||||
{relation_prompt}
|
||||
你的名字是{bot_name}
|
||||
{mood_info}
|
||||
{cycle_info_block}
|
||||
现在是{time_now},你正在上网,和qq群里的网友们聊天,以下是正在进行的聊天内容:
|
||||
{chat_observe_info}
|
||||
@@ -74,8 +73,11 @@ def init_prompt():
|
||||
|
||||
|
||||
class MindProcessor(BaseProcessor):
|
||||
log_prefix = "聊天思考"
|
||||
|
||||
def __init__(self, subheartflow_id: str):
|
||||
super().__init__()
|
||||
|
||||
self.subheartflow_id = subheartflow_id
|
||||
|
||||
self.llm_model = LLMRequest(
|
||||
@@ -228,7 +230,7 @@ class MindProcessor(BaseProcessor):
|
||||
bot_name=individuality.name,
|
||||
time_now=time_now,
|
||||
chat_observe_info=chat_observe_info,
|
||||
mood_info="mood_info",
|
||||
# mood_info="mood_info",
|
||||
hf_do_next=spark_prompt,
|
||||
last_mind=previous_mind,
|
||||
cycle_info_block=hfcloop_observe_info,
|
||||
|
||||
@@ -14,7 +14,7 @@ from src.chat.heart_flow.observation.observation import Observation
|
||||
from src.chat.heart_flow.observation.working_observation import WorkingObservation
|
||||
from src.chat.focus_chat.info.structured_info import StructuredInfo
|
||||
|
||||
logger = get_logger("tool_use")
|
||||
logger = get_logger("processor")
|
||||
|
||||
|
||||
def init_prompt():
|
||||
@@ -45,6 +45,8 @@ def init_prompt():
|
||||
|
||||
|
||||
class ToolProcessor(BaseProcessor):
|
||||
log_prefix = "工具执行器"
|
||||
|
||||
def __init__(self, subheartflow_id: str):
|
||||
super().__init__()
|
||||
self.subheartflow_id = subheartflow_id
|
||||
|
||||
Reference in New Issue
Block a user