🤖 自动格式化代码 [skip ci]
This commit is contained in:
@@ -51,7 +51,7 @@ class SearchKnowledgeTool(BaseTool):
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def get_info_from_db(
|
def get_info_from_db(
|
||||||
query_embedding: list, limit: int = 1, threshold: float = 0.5, return_raw: bool = False
|
query_embedding: list, limit: int = 1, threshold: float = 0.5, return_raw: bool = False
|
||||||
) -> Union[str, list]:
|
) -> Union[str, list]:
|
||||||
"""从数据库中获取相关信息
|
"""从数据库中获取相关信息
|
||||||
|
|
||||||
|
|||||||
@@ -26,7 +26,7 @@ class ToolUser:
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def _build_tool_prompt(
|
async def _build_tool_prompt(
|
||||||
message_txt: str, sender_name: str, chat_stream: ChatStream, subheartflow: SubHeartflow = None
|
message_txt: str, sender_name: str, chat_stream: ChatStream, subheartflow: SubHeartflow = None
|
||||||
):
|
):
|
||||||
"""构建工具使用的提示词
|
"""构建工具使用的提示词
|
||||||
|
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ class DirectMessageSender:
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def send_message(
|
async def send_message(
|
||||||
chat_stream: ChatStream,
|
chat_stream: ChatStream,
|
||||||
content: str,
|
content: str,
|
||||||
reply_to_message: Optional[Message] = None,
|
reply_to_message: Optional[Message] = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
|
|||||||
@@ -375,7 +375,7 @@ class PromptBuilder:
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def get_info_from_db(
|
def get_info_from_db(
|
||||||
query_embedding: list, limit: int = 1, threshold: float = 0.5, return_raw: bool = False
|
query_embedding: list, limit: int = 1, threshold: float = 0.5, return_raw: bool = False
|
||||||
) -> Union[str, list]:
|
) -> Union[str, list]:
|
||||||
if not query_embedding:
|
if not query_embedding:
|
||||||
return "" if not return_raw else []
|
return "" if not return_raw else []
|
||||||
|
|||||||
@@ -66,7 +66,7 @@ class PromptBuilder:
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def _build_prompt(
|
async def _build_prompt(
|
||||||
chat_stream, message_txt: str, sender_name: str = "某人", stream_id: Optional[int] = None
|
chat_stream, message_txt: str, sender_name: str = "某人", stream_id: Optional[int] = None
|
||||||
) -> tuple[str, str]:
|
) -> tuple[str, str]:
|
||||||
current_mind_info = heartflow.get_subheartflow(stream_id).current_mind
|
current_mind_info = heartflow.get_subheartflow(stream_id).current_mind
|
||||||
|
|
||||||
@@ -171,7 +171,7 @@ class PromptBuilder:
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def _build_prompt_simple(
|
async def _build_prompt_simple(
|
||||||
chat_stream, message_txt: str, sender_name: str = "某人", stream_id: Optional[int] = None
|
chat_stream, message_txt: str, sender_name: str = "某人", stream_id: Optional[int] = None
|
||||||
) -> tuple[str, str]:
|
) -> tuple[str, str]:
|
||||||
current_mind_info = heartflow.get_subheartflow(stream_id).current_mind
|
current_mind_info = heartflow.get_subheartflow(stream_id).current_mind
|
||||||
|
|
||||||
@@ -241,7 +241,7 @@ class PromptBuilder:
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def _build_prompt_check_response(
|
async def _build_prompt_check_response(
|
||||||
chat_stream,
|
chat_stream,
|
||||||
message_txt: str,
|
message_txt: str,
|
||||||
sender_name: str = "某人",
|
sender_name: str = "某人",
|
||||||
stream_id: Optional[int] = None,
|
stream_id: Optional[int] = None,
|
||||||
|
|||||||
@@ -316,7 +316,7 @@ class PersonInfoManager:
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def get_specific_value_list(
|
async def get_specific_value_list(
|
||||||
field_name: str,
|
field_name: str,
|
||||||
way: Callable[[Any], bool], # 接受任意类型值
|
way: Callable[[Any], bool], # 接受任意类型值
|
||||||
) -> Dict[str, Any]:
|
) -> Dict[str, Any]:
|
||||||
"""
|
"""
|
||||||
|
|||||||
Reference in New Issue
Block a user