parent
e42d3ae46c
commit
427c73d2a8
3 changed files with 7 additions and 2 deletions
|
|
@ -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:
|
||||
|
|
|
|||
1
graphiti_core/prompts/prompt_helpers.py
Normal file
1
graphiti_core/prompts/prompt_helpers.py
Normal file
|
|
@ -0,0 +1 @@
|
|||
DO_NOT_ESCAPE_UNICODE = '\nDo not escape unicode characters.\n'
|
||||
|
|
@ -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 <paul@getzep.com>",
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue