graphiti-core 0.17.2__py3-none-any.whl → 0.17.3__py3-none-any.whl

graphiti_core/embedder/gemini.py

@@ -37,7 +37,7 @@ from .client import EmbedderClient, EmbedderConfig
 
 logger = logging.getLogger(__name__)
 
-DEFAULT_EMBEDDING_MODEL = 'text-embedding-001' # gemini-embedding-001 or text-embedding-005
+DEFAULT_EMBEDDING_MODEL = 'text-embedding-001'  # gemini-embedding-001 or text-embedding-005
 
 DEFAULT_BATCH_SIZE = 100
 
@@ -78,7 +78,7 @@ class GeminiEmbedder(EmbedderClient):
 
         if batch_size is None and self.config.embedding_model == 'gemini-embedding-001':
             # Gemini API has a limit on the number of instances per request
-            #https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/text-embeddings-api
+            # https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/text-embeddings-api
             self.batch_size = 1
         elif batch_size is None:
             self.batch_size = DEFAULT_BATCH_SIZE
@@ -113,32 +113,34 @@ class GeminiEmbedder(EmbedderClient):
     async def create_batch(self, input_data_list: list[str]) -> list[list[float]]:
         """
         Create embeddings for a batch of input data using Google's Gemini embedding model.
-        
+
         This method handles batching to respect the Gemini API's limits on the number
         of instances that can be processed in a single request.
-        
+
         Args:
             input_data_list: A list of strings to create embeddings for.
-            
+
         Returns:
             A list of embedding vectors (each vector is a list of floats).
         """
         if not input_data_list:
             return []
-        
+
         batch_size = self.batch_size
         all_embeddings = []
-        
+
         # Process inputs in batches
         for i in range(0, len(input_data_list), batch_size):
-            batch = input_data_list[i:i + batch_size]
-            
+            batch = input_data_list[i : i + batch_size]
+
             try:
                 # Generate embeddings for this batch
                 result = await self.client.aio.models.embed_content(
                     model=self.config.embedding_model or DEFAULT_EMBEDDING_MODEL,
                     contents=batch,  # type: ignore[arg-type]  # mypy fails on broad union type
-                    config=types.EmbedContentConfig(output_dimensionality=self.config.embedding_dim),
+                    config=types.EmbedContentConfig(
+                        output_dimensionality=self.config.embedding_dim
+                    ),
                 )
 
                 if not result.embeddings or len(result.embeddings) == 0:
@@ -149,29 +151,33 @@ class GeminiEmbedder(EmbedderClient):
                     if not embedding.values:
                         raise ValueError('Empty embedding values returned')
                     all_embeddings.append(embedding.values)
-                    
+
             except Exception as e:
                 # If batch processing fails, fall back to individual processing
-                logger.warning(f"Batch embedding failed for batch {i//batch_size + 1}, falling back to individual processing: {e}")
-                
+                logger.warning(
+                    f'Batch embedding failed for batch {i // batch_size + 1}, falling back to individual processing: {e}'
+                )
+
                 for item in batch:
                     try:
                         # Process each item individually
                         result = await self.client.aio.models.embed_content(
                             model=self.config.embedding_model or DEFAULT_EMBEDDING_MODEL,
                             contents=[item],  # type: ignore[arg-type]  # mypy fails on broad union type
-                            config=types.EmbedContentConfig(output_dimensionality=self.config.embedding_dim),
+                            config=types.EmbedContentConfig(
+                                output_dimensionality=self.config.embedding_dim
+                            ),
                         )
-                        
+
                         if not result.embeddings or len(result.embeddings) == 0:
                             raise ValueError('No embeddings returned from Gemini API')
                         if not result.embeddings[0].values:
                             raise ValueError('Empty embedding values returned')
-                            
+
                         all_embeddings.append(result.embeddings[0].values)
-                        
+
                     except Exception as individual_error:
-                        logger.error(f"Failed to embed individual item: {individual_error}")
+                        logger.error(f'Failed to embed individual item: {individual_error}')
                         raise individual_error
-        
+
         return all_embeddings

graphiti_core/llm_client/client.py

@@ -172,13 +172,13 @@ class LLMClient(ABC):
         """
         Log the full input messages, the raw output (if any), and the exception for debugging failed generations.
         """
-        log = ""
-        log += f"Input messages: {json.dumps([m.model_dump() for m in messages], indent=2)}\n"
+        log = ''
+        log += f'Input messages: {json.dumps([m.model_dump() for m in messages], indent=2)}\n'
         if output is not None:
             if len(output) > 4000:
-                log += f"Raw output: {output[:2000]}... (truncated) ...{output[-2000:]}\n"
+                log += f'Raw output: {output[:2000]}... (truncated) ...{output[-2000:]}\n'
             else:
-                log += f"Raw output: {output}\n"
+                log += f'Raw output: {output}\n'
         else:
-            log += "No raw output available"
+            log += 'No raw output available'
         return log

graphiti_core/llm_client/gemini_client.py

@@ -219,14 +219,14 @@ class GeminiClient(LLMClient):
         array_match = re.search(r'\]\s*$', raw_output)
         if array_match:
             try:
-                return json.loads(raw_output[:array_match.end()])
+                return json.loads(raw_output[: array_match.end()])
             except Exception:
                 pass
         # Try to salvage a JSON object
         obj_match = re.search(r'\}\s*$', raw_output)
         if obj_match:
             try:
-                return json.loads(raw_output[:obj_match.end()])
+                return json.loads(raw_output[: obj_match.end()])
             except Exception:
                 pass
         return None
@@ -323,12 +323,14 @@ class GeminiClient(LLMClient):
                     return validated_model.model_dump()
                 except Exception as e:
                     if raw_output:
-                        logger.error("🦀 LLM generation failed parsing as JSON, will try to salvage.")
+                        logger.error(
+                            '🦀 LLM generation failed parsing as JSON, will try to salvage.'
+                        )
                         logger.error(self._get_failed_generation_log(gemini_messages, raw_output))
                         # Try to salvage
                         salvaged = self.salvage_json(raw_output)
                         if salvaged is not None:
-                            logger.warning("Salvaged partial JSON from truncated/malformed output.")
+                            logger.warning('Salvaged partial JSON from truncated/malformed output.')
                             return salvaged
                     raise Exception(f'Failed to parse structured response: {e}') from e
 
@@ -384,7 +386,11 @@ class GeminiClient(LLMClient):
                     max_tokens=max_tokens,
                     model_size=model_size,
                 )
-                last_output = response.get('content') if isinstance(response, dict) and 'content' in response else None
+                last_output = (
+                    response.get('content')
+                    if isinstance(response, dict) and 'content' in response
+                    else None
+                )
                 return response
             except RateLimitError as e:
                 # Rate limit errors should not trigger retries (fail fast)
@@ -416,7 +422,7 @@ class GeminiClient(LLMClient):
                 )
 
         # If we exit the loop without returning, all retries are exhausted
-        logger.error("🦀 LLM generation failed and retries are exhausted.")
+        logger.error('🦀 LLM generation failed and retries are exhausted.')
         logger.error(self._get_failed_generation_log(messages, last_output))
         logger.error(f'Max retries ({self.MAX_RETRIES}) exceeded. Last error: {last_error}')
-        raise last_error or Exception("Max retries exceeded")
+        raise last_error or Exception('Max retries exceeded')

graphiti_core/models/edges/edge_db_queries.py

@@ -31,9 +31,9 @@ EPISODIC_EDGE_SAVE_BULK = """
 """
 
 ENTITY_EDGE_SAVE = """
-        MATCH (source:Entity {uuid: $source_uuid}) 
-        MATCH (target:Entity {uuid: $target_uuid}) 
-        MERGE (source)-[r:RELATES_TO {uuid: $uuid}]->(target)
+        MATCH (source:Entity {uuid: $edge_data.source_uuid}) 
+        MATCH (target:Entity {uuid: $edge_data.target_uuid}) 
+        MERGE (source)-[r:RELATES_TO {uuid: $edge_data.uuid}]->(target)
         SET r = $edge_data
         WITH r CALL db.create.setRelationshipVectorProperty(r, "fact_embedding", $edge_data.fact_embedding)
         RETURN r.uuid AS uuid"""

{graphiti_core-0.17.2.dist-info → graphiti_core-0.17.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphiti-core
-Version: 0.17.2
+Version: 0.17.3
 Summary: A temporal graph building library
 Project-URL: Homepage, https://help.getzep.com/graphiti/graphiti/overview
 Project-URL: Repository, https://github.com/getzep/graphiti

{graphiti_core-0.17.2.dist-info → graphiti_core-0.17.3.dist-info}/RECORD

@@ -19,16 +19,16 @@ graphiti_core/driver/neo4j_driver.py,sha256=0MCAWAPay0LdcqrFSkY91GooUtrn1yX1CTKu
 graphiti_core/embedder/__init__.py,sha256=EL564ZuE-DZjcuKNUK_exMn_XHXm2LdO9fzdXePVKL4,179
 graphiti_core/embedder/azure_openai.py,sha256=OyomPwC1fIsddI-3n6g00kQFdQznZorBhHwkQKCLUok,2384
 graphiti_core/embedder/client.py,sha256=qEpSHceL_Gc4QQPJWIOnuNLemNuR_TYA4r28t2Vldbg,1115
-graphiti_core/embedder/gemini.py,sha256=GdpnmRKunruLB4ViJMo6K-WEv8RqZvuLfgyKXtRcEMI,7218
+graphiti_core/embedder/gemini.py,sha256=s3_2xjHdFTIuF-fJlBFwh64XK5BLPHHThuBymDpM5Jo,7194
 graphiti_core/embedder/openai.py,sha256=bIThUoLMeGlHG2-3VikzK6JZfOHKn4PKvUMx5sHxJy8,2192
 graphiti_core/embedder/voyage.py,sha256=oJHAZiNqjdEJOKgoKfGWcxK2-Ewqn5UB3vrBwIwP2u4,2546
 graphiti_core/llm_client/__init__.py,sha256=QgBWUiCeBp6YiA_xqyrDvJ9jIyy1hngH8g7FWahN3nw,776
 graphiti_core/llm_client/anthropic_client.py,sha256=xTFcrgMDK77BwnChBhYj51Jaa2mRNI850oJv2pKZI0A,12892
 graphiti_core/llm_client/azure_openai_client.py,sha256=ekERggAekbb7enes1RJqdRChf_mjaZTFXsnMbxO7azQ,2497
-graphiti_core/llm_client/client.py,sha256=fgNnJgmoZN7v7PNoJGtt4MMdKkDNsmT9F2XOLKZOU38,6473
+graphiti_core/llm_client/client.py,sha256=cUwwCZEhP9jJAI04AhHxsFPecggajSgCRCM3frrYJqA,6473
 graphiti_core/llm_client/config.py,sha256=90IgSBxZE_3nWdaEONVLUznI8lytPA7ZyexQz-_c55U,2560
 graphiti_core/llm_client/errors.py,sha256=pn6brRiLW60DAUIXJYKBT6MInrS4ueuH1hNLbn_JbQo,1243
-graphiti_core/llm_client/gemini_client.py,sha256=LKB6nktFMIn2fuRNRoGeBOmxlE3WuhB5sWI7yUwGGaA,17583
+graphiti_core/llm_client/gemini_client.py,sha256=m0-6SFUs8qqoR5rGTrASAcMtTbJKfZqO4-MaDr4CYCQ,17719
 graphiti_core/llm_client/groq_client.py,sha256=bYLE_cg1QEhugsJOXh4b1vPbxagKeMWqk48240GCzMs,2922
 graphiti_core/llm_client/openai_base_client.py,sha256=gfMcKPyLrylz_ouRdoenDWXyitmgfFZ17Zthbkq3Qs4,8126
 graphiti_core/llm_client/openai_client.py,sha256=ykBK94gxzE7iXux5rvOzVNA8q0Sqzq-8njPB75XcRe8,3240
@@ -36,7 +36,7 @@ graphiti_core/llm_client/openai_generic_client.py,sha256=WElMnPqdb1CxzYH4p2-m_9r
 graphiti_core/llm_client/utils.py,sha256=zKpxXEbKa369m4W7RDEf-m56kH46V1Mx3RowcWZEWWs,1000
 graphiti_core/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 graphiti_core/models/edges/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graphiti_core/models/edges/edge_db_queries.py,sha256=4vSWdmE5MKoDrlHJmmr2xNhVSQ-buE1O7mCX_H0Wtfk,2294
+graphiti_core/models/edges/edge_db_queries.py,sha256=YyGc0UT4eeOHQrYuXfAGIwxiX4xCc4YHrz9c2bVT7Lw,2324
 graphiti_core/models/nodes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 graphiti_core/models/nodes/node_db_queries.py,sha256=AQgRGVO-GgFWfLq1G6k8s86WItwpXruy3Mj4DBli-vM,2145
 graphiti_core/prompts/__init__.py,sha256=EA-x9xUki9l8wnu2l8ek_oNf75-do5tq5hVq7Zbv8Kw,101
@@ -71,7 +71,7 @@ graphiti_core/utils/maintenance/node_operations.py,sha256=4jMlmbB3zwK9KzIm2QXRxz
 graphiti_core/utils/maintenance/temporal_operations.py,sha256=mJkw9xLB4W2BsLfC5POr0r-PHWL9SIfNj_l_xu0B5ug,3410
 graphiti_core/utils/maintenance/utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 graphiti_core/utils/ontology_utils/entity_types_utils.py,sha256=QJX5cG0GSSNF_Mm_yrldr69wjVAbN_MxLhOSznz85Hk,1279
-graphiti_core-0.17.2.dist-info/METADATA,sha256=bqv7KcvfhQ5qp4F2BYrWK5ziw2uJzAA2rPxi-KlJTp8,23791
-graphiti_core-0.17.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-graphiti_core-0.17.2.dist-info/licenses/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
-graphiti_core-0.17.2.dist-info/RECORD,,
+graphiti_core-0.17.3.dist-info/METADATA,sha256=lKv8hXBNV6eA3NXeWdCJRfm5QaJT5Eth936n9ZxuMtI,23791
+graphiti_core-0.17.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+graphiti_core-0.17.3.dist-info/licenses/LICENSE,sha256=KCUwCyDXuVEgmDWkozHyniRyWjnWUWjkuDHfU6o3JlA,11325
+graphiti_core-0.17.3.dist-info/RECORD,,