adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(workflow/llm): replace regex substitution with string replace for context rendering

Browse Source
This commit is contained in:
Timebomb2018
2026-04-22 20:05:45 +08:00
parent 2a03f70287
commit b0bb5e883c
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/app/core/workflow/nodes/llm/node.py
View File

@@ -5,7 +5,6 @@ LLM 节点实现
""" """
import logging import logging
import re
from typing import Any from typing import Any
from langchain_core.messages import AIMessage from langchain_core.messages import AIMessage
@@ -81,7 +80,7 @@ class LLMNode(BaseNode):
def _render_context(self, message: str, variable_pool: VariablePool): def _render_context(self, message: str, variable_pool: VariablePool):
context = f"<context>{self._render_template(self.typed_config.context, variable_pool)}</context>" context = f"<context>{self._render_template(self.typed_config.context, variable_pool)}</context>"
return re.sub(r"{{context}}", context, message) return message.replace("{{context}}", context)
async def _prepare_llm( async def _prepare_llm(
self, self,