graphiti-core 0.12.0rc2__tar.gz → 0.12.0rc4__tar.gz

{graphiti_core-0.12.0rc2 → graphiti_core-0.12.0rc4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: graphiti-core
-Version: 0.12.0rc2
+Version: 0.12.0rc4
 Summary: A temporal graph building library
 License: Apache-2.0
 Author: Paul Paliychuk
@@ -22,7 +22,7 @@ Requires-Dist: groq (>=0.2.0) ; extra == "groq"
 Requires-Dist: neo4j (>=5.23.0)
 Requires-Dist: numpy (>=1.0.0)
 Requires-Dist: openai (>=1.53.0)
-Requires-Dist: pydantic (>=2.8.2)
+Requires-Dist: pydantic (>=2.11.5)
 Requires-Dist: python-dotenv (>=1.0.1)
 Requires-Dist: tenacity (>=9.0.0)
 Project-URL: Homepage, https://help.getzep.com/graphiti/graphiti/overview
@@ -236,7 +236,7 @@ Graphiti supports Azure OpenAI for both LLM inference and embeddings. To use Azu
 ```python
 from openai import AsyncAzureOpenAI
 from graphiti_core import Graphiti
-from graphiti_core.llm_client import OpenAIClient
+from graphiti_core.llm_client import LLMConfig, OpenAIClient
 from graphiti_core.embedder.openai import OpenAIEmbedder, OpenAIEmbedderConfig
 from graphiti_core.cross_encoder.openai_reranker_client import OpenAIRerankerClient
 
@@ -252,12 +252,19 @@ azure_openai_client = AsyncAzureOpenAI(
     azure_endpoint=azure_endpoint
 )
 
+# Create LLM Config with your Azure deployed model names
+azure_llm_config = LLMConfig(
+    small_model="gpt-4.1-nano",
+    model="gpt-4.1-mini",
+)
+
 # Initialize Graphiti with Azure OpenAI clients
 graphiti = Graphiti(
     "bolt://localhost:7687",
     "neo4j",
     "password",
     llm_client=OpenAIClient(
+        llm_config=azure_llm_config,
         client=azure_openai_client
     ),
     embedder=OpenAIEmbedder(
@@ -268,6 +275,7 @@ graphiti = Graphiti(
     ),
     # Optional: Configure the OpenAI cross encoder with Azure OpenAI
     cross_encoder=OpenAIRerankerClient(
+        llm_config=azure_llm_config,
         client=azure_openai_client
     )
 )

{graphiti_core-0.12.0rc2 → graphiti_core-0.12.0rc4}/README.md

@@ -205,7 +205,7 @@ Graphiti supports Azure OpenAI for both LLM inference and embeddings. To use Azu
 ```python
 from openai import AsyncAzureOpenAI
 from graphiti_core import Graphiti
-from graphiti_core.llm_client import OpenAIClient
+from graphiti_core.llm_client import LLMConfig, OpenAIClient
 from graphiti_core.embedder.openai import OpenAIEmbedder, OpenAIEmbedderConfig
 from graphiti_core.cross_encoder.openai_reranker_client import OpenAIRerankerClient
 
@@ -221,12 +221,19 @@ azure_openai_client = AsyncAzureOpenAI(
     azure_endpoint=azure_endpoint
 )
 
+# Create LLM Config with your Azure deployed model names
+azure_llm_config = LLMConfig(
+    small_model="gpt-4.1-nano",
+    model="gpt-4.1-mini",
+)
+
 # Initialize Graphiti with Azure OpenAI clients
 graphiti = Graphiti(
     "bolt://localhost:7687",
     "neo4j",
     "password",
     llm_client=OpenAIClient(
+        llm_config=azure_llm_config,
         client=azure_openai_client
     ),
     embedder=OpenAIEmbedder(
@@ -237,6 +244,7 @@ graphiti = Graphiti(
     ),
     # Optional: Configure the OpenAI cross encoder with Azure OpenAI
     cross_encoder=OpenAIRerankerClient(
+        llm_config=azure_llm_config,
         client=azure_openai_client
     )
 )

{graphiti_core-0.12.0rc2 → graphiti_core-0.12.0rc4}/graphiti_core/search/search_filters.py

@@ -21,8 +21,6 @@ from typing import Any
 from pydantic import BaseModel, Field
 from typing_extensions import LiteralString
 
-from graphiti_core.helpers import lucene_sanitize
-
 
 class ComparisonOperator(Enum):
     equals = '='
@@ -60,7 +58,7 @@ def node_search_filter_query_constructor(
     filter_params: dict[str, Any] = {}
 
     if filters.node_labels is not None:
-        node_labels = '|'.join(list(map(lucene_sanitize, filters.node_labels)))
+        node_labels = '|'.join(filters.node_labels)
         node_label_filter = ' AND n:' + node_labels
         filter_query += node_label_filter
 
@@ -80,7 +78,7 @@ def edge_search_filter_query_constructor(
         filter_params['edge_types'] = edge_types
 
     if filters.node_labels is not None:
-        node_labels = '|'.join(list(map(lucene_sanitize, filters.node_labels)))
+        node_labels = '|'.join(filters.node_labels)
         node_label_filter = '\nAND n:' + node_labels + ' AND m:' + node_labels
         filter_query += node_label_filter
 
@@ -97,12 +95,12 @@ def edge_search_filter_query_constructor(
             and_filter_query = ''
             for j, and_filter in enumerate(and_filters):
                 and_filter_query += and_filter
-                if j != len(and_filter_query) - 1:
+                if j != len(and_filters) - 1:
                     and_filter_query += ' AND '
 
             valid_at_filter += and_filter_query
 
-            if i == len(or_list) - 1:
+            if i == len(filters.valid_at) - 1:
                 valid_at_filter += ')'
             else:
                 valid_at_filter += ' OR '
@@ -122,12 +120,12 @@ def edge_search_filter_query_constructor(
             and_filter_query = ''
             for j, and_filter in enumerate(and_filters):
                 and_filter_query += and_filter
-                if j != len(and_filter_query) - 1:
+                if j != len(and_filters) - 1:
                     and_filter_query += ' AND '
 
             invalid_at_filter += and_filter_query
 
-            if i == len(or_list) - 1:
+            if i == len(filters.invalid_at) - 1:
                 invalid_at_filter += ')'
             else:
                 invalid_at_filter += ' OR '
@@ -147,12 +145,12 @@ def edge_search_filter_query_constructor(
             and_filter_query = ''
             for j, and_filter in enumerate(and_filters):
                 and_filter_query += and_filter
-                if j != len(and_filter_query) - 1:
+                if j != len(and_filters) - 1:
                     and_filter_query += ' AND '
 
             created_at_filter += and_filter_query
 
-            if i == len(or_list) - 1:
+            if i == len(filters.created_at) - 1:
                 created_at_filter += ')'
             else:
                 created_at_filter += ' OR '
@@ -172,12 +170,12 @@ def edge_search_filter_query_constructor(
             and_filter_query = ''
             for j, and_filter in enumerate(and_filters):
                 and_filter_query += and_filter
-                if j != len(and_filter_query) - 1:
+                if j != len(and_filters) - 1:
                     and_filter_query += ' AND '
 
             expired_at_filter += and_filter_query
 
-            if i == len(or_list) - 1:
+            if i == len(filters.expired_at) - 1:
                 expired_at_filter += ')'
             else:
                 expired_at_filter += ' OR '

{graphiti_core-0.12.0rc2 → graphiti_core-0.12.0rc4}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "graphiti-core"
 description = "A temporal graph building library"
-version = "0.12.0pre2"
+version = "0.12.0pre4"
 authors = [
     { "name" = "Paul Paliychuk", "email" = "paul@getzep.com" },
     { "name" = "Preston Rasmussen", "email" = "preston@getzep.com" },
@@ -12,7 +12,7 @@ license = "Apache-2.0"
 requires-python = ">=3.10,<4"
 packages = [{ include = "graphiti_core", from = "." }]
 dependencies = [
-    "pydantic>=2.8.2",
+    "pydantic>=2.11.5",
     "neo4j>=5.23.0",
     "diskcache>=5.6.3",
     "openai>=1.53.0",