refactor(knowledge): 移除未使用的HistoryManager类及其引用

清理不再使用的历史管理功能,简化代码结构
This commit is contained in:
Wenjie Zhang 2025-08-24 13:01:52 +08:00
parent 061a246159
commit e3bb35a383
3 changed files with 0 additions and 51 deletions

View File

@ -11,7 +11,6 @@ from sqlalchemy.orm import Session
from pydantic import BaseModel
from src import executor, config
from src.knowledge import HistoryManager
from src.agents import agent_manager
from src.models import select_model
from src.utils.logging_config import logger

View File

@ -1,2 +1 @@
from .history import HistoryManager
from .graphbase import GraphDatabase

View File

@ -1,49 +0,0 @@
from src.utils.prompts import get_system_prompt
class HistoryManager:
def __init__(self, history=None, system_prompt=None):
self.messages = []
system_prompt = system_prompt or get_system_prompt()
self.add_system(system_prompt)
if history:
self.messages.extend(history)
def add(self, role, content):
self.messages.append({"role": role, "content": content})
return self.messages
def add_user(self, content):
return self.add("user", content)
def add_system(self, content):
return self.add("system", content)
def add_ai(self, content):
return self.add("assistant", content)
def update_ai(self, content):
if self.messages[-1]["role"] == "assistant":
self.messages[-1]["content"] = content
return self.messages
else:
self.add_ai(content)
return self.messages
def get_history_with_msg(self, msg, role="user", max_rounds=None):
"""Get history with new message, but not append it to history."""
if max_rounds is None:
history = self.messages[:]
else:
history = self.messages[-(2*max_rounds):]
history.append({"role": role, "content": msg})
return history
def __str__(self):
history_str = ""
for message in self.messages:
msg = message["content"].replace('\n', ' ')
history_str += f"\n{message['role']}: {msg}"
return history_str