diff --git a/graphiti_core/prompts/lib.py b/graphiti_core/prompts/lib.py index cb971d0f..6757a1bd 100644 --- a/graphiti_core/prompts/lib.py +++ b/graphiti_core/prompts/lib.py @@ -74,6 +74,7 @@ from .invalidate_edges import ( versions as invalidate_edges_versions, ) from .models import Message, PromptFunction +from .prompt_helpers import DO_NOT_ESCAPE_UNICODE from .summarize_nodes import Prompt as SummarizeNodesPrompt from .summarize_nodes import Versions as SummarizeNodesVersions from .summarize_nodes import versions as summarize_nodes_versions @@ -106,7 +107,10 @@ class VersionWrapper: self.func = func def __call__(self, context: dict[str, Any]) -> list[Message]: - return self.func(context) + messages = self.func(context) + for message in messages: + message.content += DO_NOT_ESCAPE_UNICODE if message.role == 'system' else '' + return messages class PromptTypeWrapper: diff --git a/graphiti_core/prompts/prompt_helpers.py b/graphiti_core/prompts/prompt_helpers.py new file mode 100644 index 00000000..aeb1145d --- /dev/null +++ b/graphiti_core/prompts/prompt_helpers.py @@ -0,0 +1 @@ +DO_NOT_ESCAPE_UNICODE = '\nDo not escape unicode characters.\n' diff --git a/pyproject.toml b/pyproject.toml index bc000c7d..262bb50e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "graphiti-core" -version = "0.4.2" +version = "0.4.3" description = "A temporal graph building library" authors = [ "Paul Paliychuk ",