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

76 commits

Author SHA1 Message Date
James.
7ce07942b1
Fix: Add missing name_embedding field to community search queries (#664) 
Enhanced queries in search_utils.py to include 'name_embedding' field in community full-text and similarity search functions.
 2025-07-02 11:45:25 -04:00
Daniel Chalef
8213d10d44
migrate to pyright (#646) 
* migrate to pyright

* Refactor type checking to use Pyright, update dependencies, and clean up code.

- Replaced MyPy with Pyright in configuration files and CI workflows.
- Updated `pyproject.toml` and `uv.lock` to reflect new dependencies and versions.
- Adjusted type hints and fixed minor code issues across various modules for better compatibility with Pyright.
- Added new packages `backoff` and `posthog` to the project dependencies.

* Update CI workflows to install all extra dependencies for type checking and unit tests

* Update dependencies in uv.lock to replace MyPy with Pyright and add nodeenv package. Adjust type hinting in config.py for compatibility with Pyright.
 2025-06-30 12:04:21 -07:00
Gal Shubeli
6e6115c134
FalkorDB Integration: Bug Fixes and Unit Tests (#607) 
* fixes-and-tests

* update-workflow

* lint-fixes

* mypy-fixes

* fix-falkor-tests

* Update poetry.lock after pyproject.toml changes

* update-yml

* fix-tests

* comp-tests

* typo

* fix-tests

---------

Co-authored-by: Guy Korland <gkorland@gmail.com>
 2025-06-30 11:01:44 -04:00
Daniel Chalef
7537f0c972
fix: correct spacing in group IDs filter concatenation in fulltext_query function (#636) 2025-06-27 14:09:01 -07:00
Preston Rasmussen
97593550a9
fix fulltext query (#626) 
* fix fulltext query

* updates
 2025-06-25 18:09:56 -04:00
Preston Rasmussen
19fde653a6
update driver (#583) 
* update driver

* mypy updates

* mypy updates

* mypy updates

* Update graphiti_core/graph_queries.py

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

* mypy updates

* mypy

* mypy updates

* mypy updates

* mypy updates

* mypy updates

---------

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-06-13 14:12:09 -04:00
Preston Rasmussen
14146dc46f
Add support for falkordb (#575) 
* [wip] add support for falkordb

* updates

* fix-async

* progress

* fix-issues

* rm-date-handler

* red-code

* rm-uns-try

* fix-exm

* rm-un-lines

* fix-comments

* fix-se-utils

* fix-falkor-readme

* fix-falkor-cosine-score

* update-falkor-ver

* fix-vec-sim

* min-updates

* make format

* update graph driver abstraction

* poetry lock

* updates

* linter

* Update graphiti_core/search/search_utils.py

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

---------

Co-authored-by: Dudi Zimberknopf <zimber.dudi@gmail.com>
Co-authored-by: Gal Shubeli <galshubeli93@gmail.com>
Co-authored-by: Gal Shubeli <124919062+galshubeli@users.noreply.github.com>
Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
 2025-06-13 12:06:57 -04:00
TheEpTic
735b020624
BUG FIX: Fix trailing AND in edge_search_filter_query_constructor Cypher query (#541) 
Fix trailing AND in edge_search_filter_query_constructor Cypher query

Corrected the edge_search_filter_query_constructor function to prevent
trailing AND operators in generated Cypher queries, which caused
Neo.ClientError.Statement.SyntaxError. Changed condition from
`j != len(and_filter_query) - 1` to `j != len(and_filters) - 1` for
valid_at, invalid_at, created_at, and expired_at filter blocks. Also
fixed outer loop condition to use `len(filters.<field>)` instead of
`len(or_list)`. Ensures valid Cypher syntax for single DateFilter cases.

Co-authored-by: TheEpTic <326774+TheEpTic@users.noreply.github.com>
 2025-05-29 12:55:16 -04:00
Preston Rasmussen
1eea232ef1
remove sanitize (#540) 
* remove sanitize

* format
 2025-05-28 19:34:44 -04:00
Preston Rasmussen
5fe2f588a6
Edge type search (#537) 
* add filters

* search filter

* Update graphiti_core/search/search_utils.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-05-27 13:16:28 -04:00
Preston Rasmussen
db7595fe63
Edge types (#501) 
* update entity edge attributes

* Adding prompts

* extract fact attributes

* edge types

* edge types no regressions

* mypy

* mypy update

* Update graphiti_core/prompts/dedupe_edges.py

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

* Update graphiti_core/prompts/dedupe_edges.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-05-19 13:30:56 -04:00
Preston Rasmussen
9baa9b7b8a
Mmr optimizations (#481) 
* update mmr calculations

* update search

* fixes and updates

* mypy
 2025-05-12 22:30:23 -04:00
Preston Rasmussen
4198483993
improve memory leak (#478) 2025-05-12 16:32:27 -04:00
Preston Rasmussen
1f2f1eeab5
Size optimizations (#456) 
* memory optimizations for vectors

* debugged

* unused import

* Update graphiti_core/edges.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-05-07 20:08:30 -04:00
Preston Rasmussen
8b19771d86
search update (#426) 2025-04-30 18:25:43 -04:00
Preston Rasmussen
50b3df03c4
Lucene sanitize (#423) 
* lucene sanitize

* bump version
 2025-04-30 15:00:29 -04:00
Preston Rasmussen
1193b25fa3
add_episode() refactor (#421) 
* temporal updates

* update resolve nodes

* dedupe edge updates

* edge dedupe

* extract attributes

* update dynamic pydantic model

* first pass of extract node attributes

* no errors

* bug fixes

* bug fixes

* prompt updates

* prompt updates

* updates

* updates

* remove unused imports

* update tests based on changes

* remove unused import
 2025-04-30 12:08:52 -04:00
Preston Rasmussen
a26b25dc06
Add episode refactor (#399) 
* partial refactor

* get relevant nodes refactor

* load edges updates

* refactor triplets

* not there yet

* node search update

* working refactor

* updates

* mypy

* mypy
 2025-04-26 00:24:23 -04:00
Preston Rasmussen
432d2295c6
Revert episodes (#387) 
* episode search fixes and optimizations

* remove extra return string

* Update graphiti_core/utils/maintenance/graph_data_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-04-22 12:03:09 -04:00
Preston Rasmussen
009467650f
Node episodes list (#381) 
* added episode list virtual field

* in progress tests

* add tests

* update search return type

* linter

* copyright notice

* mark integration tests
 2025-04-20 23:20:19 -04:00
Preston Rasmussen
e73aaf8171
mmr update (#369) 
* mmr update

* bump version

* format
 2025-04-17 10:14:50 -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
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
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
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
1d2417ec26
Search optimizations (#280) 
fix node distance search
 2025-02-27 11:51:10 -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
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
Preston Rasmussen
00fe87679e
Bounded semaphore - limiting concurrency (#244) 
* WIP

* add semaphore

* remove unused imports

* remove unused imports

* lower concurrency limit
 2024-12-17 13:08:18 -05:00
Preston Rasmussen
34496ffa6a
Abstract Neo4j filters in search queries (#243) 
* move null check for search queries to python

* update search filtering

* update

* update
 2024-12-16 21:45:45 -05:00
Preston Rasmussen
6a152ab91a
fix node distance reranker (#231) 2024-12-06 12:08:54 -05:00
Preston Rasmussen
0fbe5c0704
Pagination for get by group_id (#218) 
* add pagination to subgraphs

* update pagination

* update LiteralString import

* cleanup

* cleanup

* update embedding dims
 2024-12-02 11:17:37 -05:00
Preston Rasmussen
52c590878a
Update edge search (#216) 
* update edge fulltext search

* bump version
 2024-11-15 14:32:11 -05:00
Preston Rasmussen
281fe072cb
add fulltext search limit (#215) 
* add fulltext search limit

* format

* update

* update

* update tests

* remove unused imports

* format

* mypy
 2024-11-14 12:18:18 -05:00
Preston Rasmussen
eba9f40ca2
add reflexion (#212) 
* add reflexion

* clean up boolean logic

* update conditional

* cap reflexion iterations

* don't do an extra reflection step
 2024-11-13 11:58:56 -05:00
Preston Rasmussen
857a8f61cf
add search recipes (#210) 2024-11-06 14:59:17 -05:00
Preston Rasmussen
6536401c8c
return no results with empty search string (#206) 
* return no results with empty search string

* update

* bump version
 2024-11-04 10:50:49 -05:00
Preston Rasmussen
b8f52670ce
Bulk add nodes and edges (#205) 
* test

* only use parallel runtime if set to true

* add and test bulk add

* remove group_ids

* format

* bump version

* update readme
 2024-10-31 12:31:37 -04:00
Preston Rasmussen
63a1b11142
update new names with input_data (#204) 2024-10-29 11:03:31 -04:00
Preston Rasmussen
7bb0c78d5d
Update reranker limits (#203) 
* update reranker limits

* update versions

* format

* update names

* fix: voyage linter

---------

Co-authored-by: paulpaliychuk <pavlo.paliychuk.ca@gmail.com>
 2024-10-28 14:50:16 -04:00
Preston Rasmussen
ceb60a3d33
Cross encoder reranker in search query (#202) 
* cross encoder reranker

* update reranker

* add openai reranker

* format

* mypy

* update

* updates

* MyPy typing

* bump version
 2024-10-25 12:29:27 -04:00
Pavlo Paliychuk
544f9e3fba
chore: Set up cross encoder client (#201) 
* chore: Set up cross encoder client

* fix: deps

* chore: move voyage to dev deps
 2024-10-24 11:36:10 -04:00
Preston Rasmussen
47ba11e08d
Number of Neo4j Connections Optimizations (#199) 
* improve node distance performance

* update episode mentions

* format

* swap to debug log
 2024-10-23 13:08:47 -04:00
Preston Rasmussen
50d2308c93
Use sessions search (#197) 
* use sessions for searches

* correct DB name

* fix typo
 2024-10-22 10:01:56 -04:00
Preston Rasmussen
b217d1e51f
Make default DB explicit (#195) 
* add default database

* update

* init tests

* update test

* bump version

* removed unused imports
 2024-10-21 12:33:32 -04:00
Preston Rasmussen
ecb704364e
update lucene search (#193) 
* update lucene search

* update max query length

* bump version
 2024-10-20 14:37:40 -07:00
Preston Rasmussen
737a3bcda6
ZEPAI-676: Remove fuzzy query in fulltext search to avoid errors with large queries (#192) 
* Remove fuzzy query in fulltext search to avoid errors with large queries

* bump version
 2024-10-19 11:35:25 -04:00