gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
294 commits 43 branches 175 tags 21 MiB
Commit graph

146 commits

Author SHA1 Message Date
neonconsultingllc
d0b1b2e5db
Fix for using non default neo4j database (#329) 
Pass database_ correctly to driver.session to fix using non default database
 2025-04-18 13:06:31 -04:00
Preston Rasmussen
f578ee2177
prompt update (#378) 2025-04-18 00:09:12 -04:00
Pavlo Paliychuk
daaa0beab9
Fix edge deletion check to handle empty episodes list (#370) 2025-04-17 16:36:40 -04:00
FuJiaJie123
9ca7ff6cf9
Fix bug of label_propagation (#302) 
Fix bug of issue #297
 2025-04-17 15:22:24 -04:00
Preston Rasmussen
e73aaf8171
mmr update (#369) 
* mmr update

* bump version

* format
 2025-04-17 10:14:50 -04:00
Preston Rasmussen
5274970be3
reduce entity type attribute hallucinations (#365) 
* reduce entity type attribute hallucinations

* reduce entity type attribute hallucinations

* reduce entity type attribute hallucinations

* mypy fix

* mypy fix

* mypy fix
 2025-04-16 19:09:25 -04:00
Evan Schultz
113179f674
Anthropic client (#361) 
* update Anthropic client to use tool calling and add tests

* fix linting errors before creating pull request by making literal types for anthropic models
 2025-04-16 12:35:07 -07:00
Preston Rasmussen
bad33f698e
make node record loading more robust (#363) 
don't throw an error if certain pieces of the dict don't exist
 2025-04-16 10:32:03 -04:00
Preston Rasmussen
45b15a06f2
add episode scope to search (#362) 
* add episode scope to search

* bump version

* linter

* Update graphiti_core/search/search_helpers.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

* mypy

---------

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-04-15 19:27:56 -04:00
Preston Rasmussen
c19f9d09d3
add get_by_entity_node_uuid (#357) 
* add get_by_entity_node_uuid

* Update graphiti_core/nodes.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

---------

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-04-15 12:37:23 -04:00
Preston Rasmussen
11e19a35b7
add reranker_min_score (#355) 
* add reranker_min_score

* update divide by 0 case

* center node always gets a score of .1

* linter
 2025-04-15 12:33:37 -04:00
Preston Rasmussen
c8d5c45269
update to 4.1 models (#352) 2025-04-14 21:02:36 -04:00
Preston Rasmussen
ed26852531
Extract entities and facts in their language (#351) 
* Extract entities and facts in their language

* linter
 2025-04-14 16:28:59 -04:00
Preston Rasmussen
5dce26722e
e2e graph builder eval (#343) 
* add partial eval platform

* dedupe updates

* add e2e eval

* Update graphiti_core/prompts/eval.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

* clear all outputs

* clear all outputs

* squash eval commits

* Update tests/evals/data/utils.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

* add longmemeval disclaimer

* remove gitignore

* add copyright headers

* add cli

* Update tests/evals/data/longmemeval_data/README.md

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

* Update tests/evals/eval_e2e_graph_building.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

* updates

---------

Co-authored-by: jackaldenryan <jackaldenryan@gmail.com>
Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-04-12 10:35:22 -04:00
Preston Rasmussen
6aa25a1901
update context string (#346) 
* update context string

* Update graphiti_core/search/search_helpers.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

* remove unused imports

* bump version

---------

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-04-10 06:57:58 -04:00
Preston Rasmussen
502b6da1c7
Add search_ and deprecate _search (#342) 
* add search_ and deprecate _search. Add formatting helper

* add search helpers file

* move SearchResults

* Update graphiti_core/search/search_helpers.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

* remove unused imports

---------

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-04-09 15:59:21 -04:00
Daniel Chalef
7f7a17c926
chore: update dependencies and refactor type hinting (#339) 
* Bump version from 0.9.0 to 0.9.1 in pyproject.toml and update google-genai dependency to >=0.1.0

* Bump version from 0.9.1 to 0.9.2 in pyproject.toml

* Update google-genai dependency version to >=0.8.0 in pyproject.toml

* loc file

* Update pyproject.toml to version 0.9.3, restructure dependencies, and modify author format. Remove outdated Google API key note from README.md.

* upgrade poetry and ruff

* Update README.md to include installation instructions for Graphiti with Google Gemini support

* fix to deps since peotry doesn't fully implement PEP 735

* Refactor string formatting in various files to use single quotes for consistency and improve readability. This includes updates in agent.ipynb, quickstart.py, multiple prompt files, and ingest.py and retrieve.py modules.

* Remove optional dependencies from pyproject.toml to streamline project requirements.
 2025-04-09 08:05:26 -07:00
Daniel Chalef
0f6ac57dab
chore: update version to 0.9.3 and restructure dependencies (#338) 
* Bump version from 0.9.0 to 0.9.1 in pyproject.toml and update google-genai dependency to >=0.1.0

* Bump version from 0.9.1 to 0.9.2 in pyproject.toml

* Update google-genai dependency version to >=0.8.0 in pyproject.toml

* loc file

* Update pyproject.toml to version 0.9.3, restructure dependencies, and modify author format. Remove outdated Google API key note from README.md.

* upgrade poetry and ruff
 2025-04-08 20:47:38 -07:00
Preston Rasmussen
6c0def79d8
Use UUID as cursor (#335) 
* update pagination

* update pagination

* Update graphiti_core/edges.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

* uuid_cursor

---------

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-04-08 17:59:40 -04:00
Pavlo Paliychuk
65b5eeae88
Refactor ORDER BY clauses in edges and nodes to use created_at instead of uuid for better chronological sorting (#322) 
* Refactor ORDER BY clauses in edges and nodes to use created_at instead of uuid for better chronological sorting

* Update ORDER BY clauses in edges and nodes to include uuid for improved sorting consistency
 2025-04-08 14:48:36 -04:00
Pavlo Paliychuk
807a402ba4
Add set max_tokens attribute in OpenAIClient initialization (#330) 2025-04-07 12:33:15 -04:00
Daniel Chalef
9e78890f2e
Gemini support (#324) 
* first cut

* Update dependencies and enhance README for optional LLM providers

- Bump aiohttp version from 3.11.14 to 3.11.16
- Update yarl version from 1.18.3 to 1.19.0
- Modify pyproject.toml to include optional extras for Anthropic, Groq, and Google Gemini
- Revise README.md to reflect new optional LLM provider installation instructions and clarify API key requirements

* Remove deprecated packages from poetry.lock and update content hash

- Removed cachetools, google-auth, google-genai, pyasn1, pyasn1-modules, rsa, and websockets from the lock file.
- Added new extras for anthropic, google-genai, and groq.
- Updated content hash to reflect changes.

* Refactor import paths for GeminiClient in README and __init__.py

- Updated import statement in README.md to reflect the new module structure for GeminiClient.
- Removed GeminiClient from the __all__ list in __init__.py as it is no longer directly imported.

* Refactor import paths for GeminiEmbedder in README and __init__.py

- Updated import statement in README.md to reflect the new module structure for GeminiEmbedder.
- Removed GeminiEmbedder and GeminiEmbedderConfig from the __all__ list in __init__.py as they are no longer directly imported.
 2025-04-06 09:27:04 -07:00
Preston Rasmussen
f4dd39128c
add entity type validation handling (#320) 
* add entity type validation handling

* bump version
 2025-04-04 13:53:04 -04:00
Preston Rasmussen
fb37565112
Retrieve episodes override (#319) 
* add episode override

* add test

* make format
 2025-04-02 17:19:24 -04:00
Preston Rasmussen
f9aa90f612
update rate limits (#316) 
* update rate limits

* default none
 2025-04-02 11:43:34 -04:00
Preston Rasmussen
f536c4e948
get edges by uuids return empty list instead of errors (#308) 
* get edges by uuids return empty list instead of errors

* lint

* fix null bugs

* guard rails for null summary

* supress
 2025-03-27 15:00:05 -04:00
Preston Rasmussen
f91570a359
update get edges by uuids (#307) 
* update get edges by uuids

* Update graphiti_core/errors.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

* remove sample test

---------

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-03-27 10:45:06 -04:00
Preston Rasmussen
04203506d9
fix bug with updating node type attributes (#305) 
fix bug with saving new properties
 2025-03-26 12:37:48 -04:00
Daniel Chalef
4307274967
Add MCP Server (#301) 
* experimental

* experimental

* experimental

* wip

* wip

* wip

* wip

* code cleanup

* refactor and cleanup

* fix lint

* remove unneeded mcp dep

* polish
 2025-03-24 17:08:19 -07:00
Preston Rasmussen
4e52c57675
fix node save errors (#303) 2025-03-24 13:52:14 -04:00
Preston Rasmussen
d91aaa052c
node classification failsafe - remove node labels not part of the custom ontology (#295) 
* node classification failsafe

* Update graphiti_core/utils/maintenance/node_operations.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

* test update

* dict.keys update

---------

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-03-18 18:32:56 -04:00
Preston Rasmussen
f651984bd8
remove BGE reranker client from cross_encoder initializatio nto preve… (#296) 
* remove BGE reranker client from cross_encoder initializatio nto prevent loading of sentence transformers

* remove bge reranker client import
 2025-03-18 17:48:14 -04:00
Tomek Słoma
5cad6c8504
Allow usage of different openai compatible clients in embedder and encoder (#279) 
* allow usage of different openai compatible clients in embedder and encoder

* azure openai

* cross encoder example

---------

Co-authored-by: Daniel Chalef <131175+danielchalef@users.noreply.github.com>
 2025-03-15 20:46:22 -07:00
Daniel Chalef
55e308fb9f
feat: add Azure OpenAI Embedding support to Graphiti (#293) 
support for Azure OpenAI
 2025-03-15 14:15:45 -07:00
Preston Rasmussen
dff0b3c42e
get_nodes_and_edges_by_episode (#292) 
update name
 2025-03-13 15:50:03 -04:00
Preston Rasmussen
daf14a6509
get edges by node uuid (#291) 
* get edges by node uuid

* lint
 2025-03-13 15:44:02 -04:00
Preston Rasmussen
e83bcbb435
swap type_description to docstring (#287) 
* swap type_description to docstring

* remove unused imports

* bump version

* removed unused imports
 2025-03-05 15:27:03 -05:00
Preston Rasmussen
f73867e0fa
Entity classification updates (#285) 
* updates

* tested

* remove unused imports

* llm outputs will be dicts rather than pydantic models

* removed unused imports
 2025-03-05 12:08:11 -05:00
Preston Rasmussen
7f20b21572
Entity attributes in prompts (#284) 
* add node attributes to prompts

* tested

* attribute update
 2025-03-04 16:34:19 -05:00
Preston Rasmussen
a18dde36be
Catch classification exceptions (#283) 
* Catch classification exceptions

* except

* Update graphiti_core/utils/maintenance/node_operations.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

---------

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-02-28 17:06:25 -05:00
Preston Rasmussen
6f874730f3
Entity classification updates (#281) 
* node classification updates

* update

* remove unused code

* update
 2025-02-27 15:12:50 -05:00
Preston Rasmussen
1d2417ec26
Search optimizations (#280) 
fix node distance search
 2025-02-27 11:51:10 -05:00
Preston Rasmussen
db0df06975
remove core properties from attribute list (#277) 2025-02-25 15:55:06 -05:00
Preston Rasmussen
9efa6762d7
entity typo (#274) 2025-02-24 12:44:17 -05:00
Preston Rasmussen
088029a80c
node label filters (#265) 
* node label filters

* update

* add search filters

* updates

* bump versions

* update tests

* test update
 2025-02-21 12:38:01 -05:00
Preston Rasmussen
29a071b2b8
Custom ontology (#262) 
* ontology

* extract and save node labels

* extract entity type properties

* neo4j upgrade needed

* add entity types

* update typing

* update types

* updates

* Update graphiti_core/utils/maintenance/node_operations.py

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>

* fix warning

* mypy updates

* update properties

* mypy ignore

* mypy types

* bump version

---------

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-02-13 12:17:52 -05:00
Preston Rasmussen
0e45d15462
Add triple update (#263) 
* update add-triplet

* test fixes
 2025-02-12 12:04:43 -05:00
Preston Rasmussen
a99aad59de
Remove Episode by uuid (#261) 
* add remove_episode

* delete episodes

* update

* bump version
 2025-02-05 15:17:08 -05:00
Preston Rasmussen
0d41efe0c9
update output token limits to 2048 (#258) 
update token limits
 2025-01-31 21:33:10 -05:00
Preston Rasmussen
6ef2f5e097
Date filters (#240) 
* add search filters

* add search filters

* mypy

* mypy

* update filtering

* date-filters

* update

* update filter queries

* update dictionary
 2025-01-28 11:52:53 -05:00