From c28bde6b07f828d3a617f830c280549792d7cf6b Mon Sep 17 00:00:00 2001 From: Preston Rasmussen <109292228+prasmussen15@users.noreply.github.com> Date: Mon, 18 Aug 2025 11:15:45 -0400 Subject: [PATCH] fix typo and model selector (#843) * fix typo and model selector * bump version --- graphiti_core/cross_encoder/openai_reranker_client.py | 2 +- graphiti_core/models/nodes/node_db_queries.py | 4 +--- graphiti_core/prompts/dedupe_nodes.py | 2 +- pyproject.toml | 2 +- 4 files changed, 4 insertions(+), 6 deletions(-) diff --git a/graphiti_core/cross_encoder/openai_reranker_client.py b/graphiti_core/cross_encoder/openai_reranker_client.py index 55cfb40f..2e6c5b2f 100644 --- a/graphiti_core/cross_encoder/openai_reranker_client.py +++ b/graphiti_core/cross_encoder/openai_reranker_client.py @@ -84,7 +84,7 @@ class OpenAIRerankerClient(CrossEncoderClient): responses = await semaphore_gather( *[ self.client.chat.completions.create( - model=DEFAULT_MODEL, + model=self.config.model or DEFAULT_MODEL, messages=openai_messages, temperature=0, max_tokens=1, diff --git a/graphiti_core/models/nodes/node_db_queries.py b/graphiti_core/models/nodes/node_db_queries.py index cc28d1de..82a53461 100644 --- a/graphiti_core/models/nodes/node_db_queries.py +++ b/graphiti_core/models/nodes/node_db_queries.py @@ -130,6 +130,4 @@ COMMUNITY_NODE_RETURN = """ n.group_id AS group_id, n.summary AS summary, n.created_at AS created_at -""" - - +""" diff --git a/graphiti_core/prompts/dedupe_nodes.py b/graphiti_core/prompts/dedupe_nodes.py index 5e2507c4..37f007af 100644 --- a/graphiti_core/prompts/dedupe_nodes.py +++ b/graphiti_core/prompts/dedupe_nodes.py @@ -108,7 +108,7 @@ def nodes(context: dict[str, Any]) -> list[Message]: Message( role='system', content='You are a helpful assistant that determines whether or not ENTITIES extracted from a conversation are duplicates' - 'of existing entities.', + ' of existing entities.', ), Message( role='user', diff --git a/pyproject.toml b/pyproject.toml index c76f2f11..1a4ed522 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "graphiti-core" description = "A temporal graph building library" -version = "0.18.7" +version = "0.18.8" authors = [ { name = "Paul Paliychuk", email = "paul@getzep.com" }, { name = "Preston Rasmussen", email = "preston@getzep.com" },