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,
|
versions as invalidate_edges_versions,
|
||||||
)
|
)
|
||||||
from .models import Message, PromptFunction
|
from .models import Message, PromptFunction
|
||||||
|
from .prompt_helpers import DO_NOT_ESCAPE_UNICODE
|
||||||
from .summarize_nodes import Prompt as SummarizeNodesPrompt
|
from .summarize_nodes import Prompt as SummarizeNodesPrompt
|
||||||
from .summarize_nodes import Versions as SummarizeNodesVersions
|
from .summarize_nodes import Versions as SummarizeNodesVersions
|
||||||
from .summarize_nodes import versions as summarize_nodes_versions
|
from .summarize_nodes import versions as summarize_nodes_versions
|
||||||
|
|
@ -106,7 +107,10 @@ class VersionWrapper:
|
||||||
self.func = func
|
self.func = func
|
||||||
|
|
||||||
def __call__(self, context: dict[str, Any]) -> list[Message]:
|
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:
|
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]
|
[tool.poetry]
|
||||||
name = "graphiti-core"
|
name = "graphiti-core"
|
||||||
version = "0.4.2"
|
version = "0.4.3"
|
||||||
description = "A temporal graph building library"
|
description = "A temporal graph building library"
|
||||||
authors = [
|
authors = [
|
||||||
"Paul Paliychuk <paul@getzep.com>",
|
"Paul Paliychuk <paul@getzep.com>",
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue