graphiti-core 0.15.1__tar.gz → 0.17.0__tar.gz

{graphiti_core-0.15.1 → graphiti_core-0.17.0}/.env.example

@@ -12,7 +12,6 @@ FALKORDB_PORT=
 FALKORDB_USER=
 FALKORDB_PASSWORD=
 
-DEFAULT_DATABASE=
 USE_PARALLEL_RUNTIME=
 SEMAPHORE_LIMIT=
 GITHUB_SHA=

graphiti_core-0.17.0/.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,54 @@
+---
+name: Bug Report
+about: Create a report to help us improve Graphiti
+title: '[BUG] '
+labels: bug
+assignees: ''
+---
+
+## Bug Description
+A clear and concise description of what the bug is.
+
+## Steps to Reproduce
+Provide a minimal code example that reproduces the issue:
+
+```python
+# Your code here
+```
+
+## Expected Behavior
+A clear and concise description of what you expected to happen.
+
+## Actual Behavior
+A clear and concise description of what actually happened.
+
+## Environment
+- **Graphiti Version**: [e.g. 0.15.1]
+- **Python Version**: [e.g. 3.11.5]
+- **Operating System**: [e.g. macOS 14.0, Ubuntu 22.04]
+- **Database Backend**: [e.g. Neo4j 5.26, FalkorDB 1.1.2]
+- **LLM Provider & Model**: [e.g. OpenAI gpt-4.1, Anthropic claude-4-sonnet, Google gemini-2.5-flash]
+
+## Installation Method
+- [ ] pip install
+- [ ] uv add
+- [ ] Development installation (git clone)
+
+## Error Messages/Traceback
+```
+Paste the full error message and traceback here
+```
+
+## Configuration
+```python
+# Relevant configuration or initialization code
+```
+
+## Additional Context
+- Does this happen consistently or intermittently?
+- Which component are you using? (core library, REST server, MCP server)
+- Any recent changes to your environment?
+- Related issues or similar problems you've encountered?
+
+## Possible Solution
+If you have ideas about what might be causing the issue or how to fix it, please share them here.

graphiti_core-0.17.0/.github/pull_request_template.md

@@ -0,0 +1,32 @@
+## Summary
+Brief description of the changes in this PR.
+
+## Type of Change
+- [ ] Bug fix
+- [ ] New feature
+- [ ] Performance improvement
+- [ ] Documentation/Tests
+
+## Objective
+**For new features and performance improvements:** Clearly describe the objective and rationale for this change.
+
+## Testing
+- [ ] Unit tests added/updated
+- [ ] Integration tests added/updated
+- [ ] All existing tests pass
+
+## Breaking Changes
+- [ ] This PR contains breaking changes
+
+If this is a breaking change, describe:
+- What functionality is affected
+- Migration path for existing users
+
+## Checklist
+- [ ] Code follows project style guidelines (`make lint` passes)
+- [ ] Self-review completed
+- [ ] Documentation updated where necessary
+- [ ] No secrets or sensitive information committed
+
+## Related Issues
+Closes #[issue number]

{graphiti_core-0.15.1 → graphiti_core-0.17.0}/.github/workflows/cla.yml

@@ -28,7 +28,7 @@ jobs:
           path-to-document: "https://github.com/getzep/graphiti/blob/main/Zep-CLA.md" # e.g. a CLA or a DCO document
           # branch should not be protected unless a personal PAT is used
           branch: "main"
-          allowlist: paul-paliychuk,prasmussen15,danielchalef,dependabot[bot],ellipsis-dev
+          allowlist: paul-paliychuk,prasmussen15,danielchalef,dependabot[bot],ellipsis-dev,Claude[bot]
 
           # the followings are the optional inputs - If the optional inputs are not given, then default values will be taken
           #remote-organization-name: enter the remote organization name where the signatures should be stored (Default is storing the signatures in the same repository)

graphiti_core-0.17.0/.github/workflows/claude-code-review.yml

@@ -0,0 +1,78 @@
+name: Claude Code Review
+
+on:
+  pull_request:
+    types: [opened, synchronize]
+    # Optional: Only run on specific file changes
+    # paths:
+    #   - "src/**/*.ts"
+    #   - "src/**/*.tsx"
+    #   - "src/**/*.js"
+    #   - "src/**/*.jsx"
+
+jobs:
+  claude-review:
+    # Optional: Filter by PR author
+    # if: |
+    #   github.event.pull_request.user.login == 'external-contributor' ||
+    #   github.event.pull_request.user.login == 'new-developer' ||
+    #   github.event.pull_request.author_association == 'FIRST_TIME_CONTRIBUTOR'
+    
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      pull-requests: write
+      issues: read
+      id-token: write
+    
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 1
+
+      - name: Run Claude Code Review
+        id: claude-review
+        uses: anthropics/claude-code-action@beta
+        with:
+          anthropic_api_key: ${{ secrets.ANTHROPIC_API_KEY }}
+
+          # Optional: Specify model (defaults to Claude Sonnet 4, uncomment for Claude Opus 4)
+          # model: "claude-opus-4-20250514"
+          
+          # Direct prompt for automated review (no @claude mention needed)
+          direct_prompt: |
+            Please review this pull request and provide feedback on:
+            - Code quality and best practices
+            - Potential bugs or issues
+            - Performance considerations
+            - Security concerns
+            - Test coverage
+            
+            Be constructive and helpful in your feedback.
+
+          # Optional: Use sticky comments to make Claude reuse the same comment on subsequent pushes to the same PR
+          # use_sticky_comment: true
+          
+          # Optional: Customize review based on file types
+          # direct_prompt: |
+          #   Review this PR focusing on:
+          #   - For TypeScript files: Type safety and proper interface usage
+          #   - For API endpoints: Security, input validation, and error handling
+          #   - For React components: Performance, accessibility, and best practices
+          #   - For tests: Coverage, edge cases, and test quality
+          
+          # Optional: Different prompts for different authors
+          # direct_prompt: |
+          #   ${{ github.event.pull_request.author_association == 'FIRST_TIME_CONTRIBUTOR' && 
+          #   'Welcome! Please review this PR from a first-time contributor. Be encouraging and provide detailed explanations for any suggestions.' ||
+          #   'Please provide a thorough code review focusing on our coding standards and best practices.' }}
+          
+          # Optional: Add specific tools for running tests or linting
+          # allowed_tools: "Bash(npm run test),Bash(npm run lint),Bash(npm run typecheck)"
+          
+          # Optional: Skip review for certain conditions
+          # if: |
+          #   !contains(github.event.pull_request.title, '[skip-review]') &&
+          #   !contains(github.event.pull_request.title, '[WIP]')
+

graphiti_core-0.17.0/.github/workflows/claude.yml

@@ -0,0 +1,64 @@
+name: Claude Code
+
+on:
+  issue_comment:
+    types: [created]
+  pull_request_review_comment:
+    types: [created]
+  issues:
+    types: [opened, assigned]
+  pull_request_review:
+    types: [submitted]
+
+jobs:
+  claude:
+    if: |
+      (github.event_name == 'issue_comment' && contains(github.event.comment.body, '@claude')) ||
+      (github.event_name == 'pull_request_review_comment' && contains(github.event.comment.body, '@claude')) ||
+      (github.event_name == 'pull_request_review' && contains(github.event.review.body, '@claude')) ||
+      (github.event_name == 'issues' && (contains(github.event.issue.body, '@claude') || contains(github.event.issue.title, '@claude')))
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      pull-requests: write
+      issues: write
+      id-token: write
+      actions: read # Required for Claude to read CI results on PRs
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 1
+
+      - name: Run Claude Code
+        id: claude
+        uses: anthropics/claude-code-action@beta
+        with:
+          anthropic_api_key: ${{ secrets.ANTHROPIC_API_KEY }}
+
+          # This is an optional setting that allows Claude to read CI results on PRs
+          additional_permissions: |
+            actions: read
+          
+          # Optional: Specify model (defaults to Claude Sonnet 4, uncomment for Claude Opus 4)
+          # model: "claude-opus-4-20250514"
+          
+          # Optional: Customize the trigger phrase (default: @claude)
+          # trigger_phrase: "/claude"
+          
+          # Optional: Trigger when specific user is assigned to an issue
+          # assignee_trigger: "claude-bot"
+          
+          # Optional: Allow Claude to run specific commands
+          # allowed_tools: "Bash(npm install),Bash(npm run build),Bash(npm run test:*),Bash(npm run lint:*)"
+          
+          # Optional: Add custom instructions for Claude to customize its behavior for your project
+          # custom_instructions: |
+          #   Follow our coding standards
+          #   Ensure all new code has tests
+          #   Use TypeScript for new files
+          
+          # Optional: Custom environment variables for Claude
+          # claude_env: |
+          #   NODE_ENV: test
+

{graphiti_core-0.15.1 → graphiti_core-0.17.0}/CLAUDE.md

@@ -24,7 +24,7 @@ uv sync --extra dev
 # Format code (ruff import sorting + formatting)
 make format
 
-# Lint code (ruff + mypy type checking)
+# Lint code (ruff + pyright type checking)
 make lint
 
 # Run tests
@@ -102,7 +102,11 @@ docker-compose up
 ### Database Setup
 
 - **Neo4j**: Version 5.26+ required, available via Neo4j Desktop
+  - Database name defaults to `neo4j` (hardcoded in Neo4jDriver)
+  - Override by passing `database` parameter to driver constructor
 - **FalkorDB**: Version 1.1.2+ as alternative backend
+  - Database name defaults to `default_db` (hardcoded in FalkorDriver)
+  - Override by passing `database` parameter to driver constructor
 
 ## Development Guidelines
 
@@ -111,13 +115,18 @@ docker-compose up
 - Use Ruff for formatting and linting (configured in pyproject.toml)
 - Line length: 100 characters
 - Quote style: single quotes
-- Type checking with MyPy is enforced
+- Type checking with Pyright is enforced
+- Main project uses `typeCheckingMode = "basic"`, server uses `typeCheckingMode = "standard"`
 
 ### Testing Requirements
 
 - Run tests with `make test` or `pytest`
-- Integration tests require database connections
+- Integration tests require database connections and are marked with `_int` suffix
 - Use `pytest-xdist` for parallel test execution
+- Run specific test files: `pytest tests/test_specific_file.py`
+- Run specific test methods: `pytest tests/test_file.py::test_method_name`
+- Run only integration tests: `pytest tests/ -k "_int"`
+- Run only unit tests: `pytest tests/ -k "not _int"`
 
 ### LLM Provider Support
 
@@ -130,4 +139,4 @@ When working with the MCP server, follow the patterns established in `mcp_server
 - Always search for existing knowledge before adding new information
 - Use specific entity type filters (`Preference`, `Procedure`, `Requirement`)
 - Store new information immediately using `add_memory`
-- Follow discovered procedures and respect established preferences
+- Follow discovered procedures and respect established preferences

{graphiti_core-0.15.1 → graphiti_core-0.17.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphiti-core
-Version: 0.15.1
+Version: 0.17.0
 Summary: A temporal graph building library
 Project-URL: Homepage, https://help.getzep.com/graphiti/graphiti/overview
 Project-URL: Repository, https://github.com/getzep/graphiti
@@ -265,6 +265,24 @@ In addition to the Neo4j and OpenAi-compatible credentials, Graphiti also has a
 If you are using one of our supported models, such as Anthropic or Voyage models, the necessary environment variables
 must be set.
 
+### Database Configuration
+
+Database names are configured directly in the driver constructors:
+
+- **Neo4j**: Database name defaults to `neo4j` (hardcoded in Neo4jDriver)
+- **FalkorDB**: Database name defaults to `default_db` (hardcoded in FalkorDriver)
+
+To use a different database name, pass the `database` parameter when creating the driver:
+
+```python
+from graphiti_core.driver.neo4j_driver import Neo4jDriver
+
+# Use custom database name
+driver = Neo4jDriver(uri="bolt://localhost:7687", user="neo4j", password="password", database="my_db")
+```
+
+### Performance Configuration
+
 `USE_PARALLEL_RUNTIME` is an optional boolean variable that can be set to true if you wish
 to enable Neo4j's parallel runtime feature for several of our search queries.
 Note that this feature is not supported for Neo4j Community edition or for smaller AuraDB instances,

{graphiti_core-0.15.1 → graphiti_core-0.17.0}/README.md

@@ -213,6 +213,24 @@ In addition to the Neo4j and OpenAi-compatible credentials, Graphiti also has a
 If you are using one of our supported models, such as Anthropic or Voyage models, the necessary environment variables
 must be set.
 
+### Database Configuration
+
+Database names are configured directly in the driver constructors:
+
+- **Neo4j**: Database name defaults to `neo4j` (hardcoded in Neo4jDriver)
+- **FalkorDB**: Database name defaults to `default_db` (hardcoded in FalkorDriver)
+
+To use a different database name, pass the `database` parameter when creating the driver:
+
+```python
+from graphiti_core.driver.neo4j_driver import Neo4jDriver
+
+# Use custom database name
+driver = Neo4jDriver(uri="bolt://localhost:7687", user="neo4j", password="password", database="my_db")
+```
+
+### Performance Configuration
+
 `USE_PARALLEL_RUNTIME` is an optional boolean variable that can be set to true if you wish
 to enable Neo4j's parallel runtime feature for several of our search queries.
 Note that this feature is not supported for Neo4j Community edition or for smaller AuraDB instances,

{graphiti_core-0.15.1 → graphiti_core-0.17.0}/examples/podcast/podcast_runner.py

@@ -25,6 +25,8 @@ from pydantic import BaseModel, Field
 from transcript_parser import parse_podcast_messages
 
 from graphiti_core import Graphiti
+from graphiti_core.nodes import EpisodeType
+from graphiti_core.utils.bulk_utils import RawEpisode
 from graphiti_core.utils.maintenance.graph_data_operations import clear_data
 
 load_dotenv()
@@ -67,7 +69,7 @@ class IsPresidentOf(BaseModel):
     """Relationship between a person and the entity they are a president of"""
 
 
-async def main():
+async def main(use_bulk: bool = False):
     setup_logging()
     client = Graphiti(neo4j_uri, neo4j_user, neo4j_password)
     await clear_data(client.driver)
@@ -75,21 +77,43 @@ async def main():
     messages = parse_podcast_messages()
     group_id = str(uuid4())
 
-    for i, message in enumerate(messages[3:14]):
-        episodes = await client.retrieve_episodes(message.actual_timestamp, 3, group_ids=[group_id])
-        episode_uuids = [episode.uuid for episode in episodes]
-
-        await client.add_episode(
-            name=f'Message {i}',
-            episode_body=f'{message.speaker_name} ({message.role}): {message.content}',
-            reference_time=message.actual_timestamp,
-            source_description='Podcast Transcript',
+    raw_episodes: list[RawEpisode] = []
+    for i, message in enumerate(messages[3:7]):
+        raw_episodes.append(
+            RawEpisode(
+                name=f'Message {i}',
+                content=f'{message.speaker_name} ({message.role}): {message.content}',
+                reference_time=message.actual_timestamp,
+                source=EpisodeType.message,
+                source_description='Podcast Transcript',
+            )
+        )
+    if use_bulk:
+        await client.add_episode_bulk(
+            raw_episodes,
             group_id=group_id,
             entity_types={'Person': Person},
             edge_types={'IS_PRESIDENT_OF': IsPresidentOf},
             edge_type_map={('Person', 'Entity'): ['PRESIDENT_OF']},
-            previous_episode_uuids=episode_uuids,
         )
-
-
-asyncio.run(main())
+    else:
+        for i, message in enumerate(messages[3:14]):
+            episodes = await client.retrieve_episodes(
+                message.actual_timestamp, 3, group_ids=[group_id]
+            )
+            episode_uuids = [episode.uuid for episode in episodes]
+
+            await client.add_episode(
+                name=f'Message {i}',
+                episode_body=f'{message.speaker_name} ({message.role}): {message.content}',
+                reference_time=message.actual_timestamp,
+                source_description='Podcast Transcript',
+                group_id=group_id,
+                entity_types={'Person': Person},
+                edge_types={'IS_PRESIDENT_OF': IsPresidentOf},
+                edge_type_map={('Person', 'Entity'): ['PRESIDENT_OF']},
+                previous_episode_uuids=episode_uuids,
+            )
+
+
+asyncio.run(main(True))

{graphiti_core-0.15.1 → graphiti_core-0.17.0}/examples/quickstart/README.md

@@ -41,6 +41,8 @@ export NEO4J_PASSWORD=password
 
 # Optional FalkorDB connection parameters (defaults shown)
 export FALKORDB_URI=falkor://localhost:6379
+
+# To use a different database, modify the driver constructor in the script
 ```
 
 3. Run the example:
@@ -71,6 +73,23 @@ After running this example, you can:
 4. Try other predefined search recipes from `graphiti_core.search.search_config_recipes`
 5. Explore the more advanced examples in the other directories
 
+## Troubleshooting
+
+### "Graph not found: default_db" Error
+
+If you encounter the error `Neo.ClientError.Database.DatabaseNotFound: Graph not found: default_db`, this occurs when the driver is trying to connect to a database that doesn't exist.
+
+**Solution:**
+The Neo4j driver defaults to using `neo4j` as the database name. If you need to use a different database, modify the driver constructor in the script:
+
+```python
+# In quickstart_neo4j.py, change:
+driver = Neo4jDriver(uri=neo4j_uri, user=neo4j_user, password=neo4j_password)
+
+# To specify a different database:
+driver = Neo4jDriver(uri=neo4j_uri, user=neo4j_user, password=neo4j_password, database="your_db_name")
+```
+
 ## Understanding the Output
 
 ### Edge Search Results

{graphiti_core-0.15.1 → graphiti_core-0.17.0}/graphiti_core/cross_encoder/gemini_reranker_client.py

@@ -41,6 +41,10 @@ DEFAULT_MODEL = 'gemini-2.5-flash-lite-preview-06-17'
 
 
 class GeminiRerankerClient(CrossEncoderClient):
+    """
+    Google Gemini Reranker Client
+    """
+
     def __init__(
         self,
         config: LLMConfig | None = None,
@@ -57,7 +61,6 @@ class GeminiRerankerClient(CrossEncoderClient):
             config (LLMConfig | None): The configuration for the LLM client, including API key, model, base URL, temperature, and max tokens.
             client (genai.Client | None): An optional async client instance to use. If not provided, a new genai.Client is created.
         """
-
         if config is None:
             config = LLMConfig()
 

{graphiti_core-0.15.1 → graphiti_core-0.17.0}/graphiti_core/driver/driver.py

@@ -19,8 +19,6 @@ from abc import ABC, abstractmethod
 from collections.abc import Coroutine
 from typing import Any
 
-from graphiti_core.helpers import DEFAULT_DATABASE
-
 logger = logging.getLogger(__name__)
 
 
@@ -54,7 +52,7 @@ class GraphDriver(ABC):
         raise NotImplementedError()
 
     @abstractmethod
-    def session(self, database: str) -> GraphDriverSession:
+    def session(self, database: str | None = None) -> GraphDriverSession:
         raise NotImplementedError()
 
     @abstractmethod
@@ -62,5 +60,5 @@ class GraphDriver(ABC):
         raise NotImplementedError()
 
     @abstractmethod
-    def delete_all_indexes(self, database_: str = DEFAULT_DATABASE) -> Coroutine:
+    def delete_all_indexes(self, database_: str | None = None) -> Coroutine:
         raise NotImplementedError()

{graphiti_core-0.15.1 → graphiti_core-0.17.0}/graphiti_core/driver/falkordb_driver.py

@@ -33,7 +33,6 @@ else:
         ) from None
 
 from graphiti_core.driver.driver import GraphDriver, GraphDriverSession
-from graphiti_core.helpers import DEFAULT_DATABASE
 
 logger = logging.getLogger(__name__)
 
@@ -81,6 +80,7 @@ class FalkorDriver(GraphDriver):
         username: str | None = None,
         password: str | None = None,
         falkor_db: FalkorDB | None = None,
+        database: str = 'default_db',
     ):
         """
         Initialize the FalkorDB driver.
@@ -95,15 +95,16 @@ class FalkorDriver(GraphDriver):
             self.client = falkor_db
         else:
             self.client = FalkorDB(host=host, port=port, username=username, password=password)
+            self._database = database
 
     def _get_graph(self, graph_name: str | None) -> FalkorGraph:
-        # FalkorDB requires a non-None database name for multi-tenant graphs; the default is DEFAULT_DATABASE
+        # FalkorDB requires a non-None database name for multi-tenant graphs; the default is "default_db"
         if graph_name is None:
-            graph_name = DEFAULT_DATABASE
+            graph_name = self._database
         return self.client.select_graph(graph_name)
 
     async def execute_query(self, cypher_query_, **kwargs: Any):
-        graph_name = kwargs.pop('database_', DEFAULT_DATABASE)
+        graph_name = kwargs.pop('database_', self._database)
         graph = self._get_graph(graph_name)
 
         # Convert datetime objects to ISO strings (FalkorDB does not support datetime objects directly)
@@ -136,7 +137,7 @@ class FalkorDriver(GraphDriver):
 
         return records, header, None
 
-    def session(self, database: str | None) -> GraphDriverSession:
+    def session(self, database: str | None = None) -> GraphDriverSession:
         return FalkorDriverSession(self._get_graph(database))
 
     async def close(self) -> None:
@@ -148,10 +149,11 @@ class FalkorDriver(GraphDriver):
         elif hasattr(self.client.connection, 'close'):
             await self.client.connection.close()
 
-    async def delete_all_indexes(self, database_: str = DEFAULT_DATABASE) -> None:
+    async def delete_all_indexes(self, database_: str | None = None) -> None:
+        database = database_ or self._database
         await self.execute_query(
             'CALL db.indexes() YIELD name DROP INDEX name',
-            database_=database_,
+            database_=database,
         )
 
 

{graphiti_core-0.15.1 → graphiti_core-0.17.0}/graphiti_core/driver/neo4j_driver.py

@@ -22,7 +22,6 @@ from neo4j import AsyncGraphDatabase, EagerResult
 from typing_extensions import LiteralString
 
 from graphiti_core.driver.driver import GraphDriver, GraphDriverSession
-from graphiti_core.helpers import DEFAULT_DATABASE
 
 logger = logging.getLogger(__name__)
 
@@ -30,34 +29,36 @@ logger = logging.getLogger(__name__)
 class Neo4jDriver(GraphDriver):
     provider: str = 'neo4j'
 
-    def __init__(
-        self,
-        uri: str,
-        user: str | None,
-        password: str | None,
-    ):
+    def __init__(self, uri: str, user: str | None, password: str | None, database: str = 'neo4j'):
         super().__init__()
         self.client = AsyncGraphDatabase.driver(
             uri=uri,
             auth=(user or '', password or ''),
         )
+        self._database = database
 
     async def execute_query(self, cypher_query_: LiteralString, **kwargs: Any) -> EagerResult:
+        # Check if database_ is provided in kwargs.
+        # If not populated, set the value to retain backwards compatibility
         params = kwargs.pop('params', None)
+        if params is None:
+            params = {}
+        params.setdefault('database_', self._database)
+
         result = await self.client.execute_query(cypher_query_, parameters_=params, **kwargs)
 
         return result
 
-    def session(self, database: str) -> GraphDriverSession:
-        return self.client.session(database=database)  # type: ignore
+    def session(self, database: str | None = None) -> GraphDriverSession:
+        _database = database or self._database
+        return self.client.session(database=_database)  # type: ignore
 
     async def close(self) -> None:
         return await self.client.close()
 
-    def delete_all_indexes(
-        self, database_: str = DEFAULT_DATABASE
-    ) -> Coroutine[Any, Any, EagerResult]:
+    def delete_all_indexes(self, database_: str | None = None) -> Coroutine[Any, Any, EagerResult]:
+        database = database_ or self._database
         return self.client.execute_query(
             'CALL db.indexes() YIELD name DROP INDEX name',
-            database_=database_,
+            database_=database,
         )