gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update

Browse source
This commit is contained in:
prestonrasmussen 2025-09-09 10:45:55 -04:00
parent 568ef77e04
commit 49a91ecca1
1 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
graphiti_core/search/search_filters.py
View file

@ -54,6 +54,16 @@ class SearchFilters(BaseModel):
expired_at: list[list[DateFilter]] | None = Field(default=None) expired_at: list[list[DateFilter]] | None = Field(default=None)
def cypher_to_opensearch_operator(op: ComparisonOperator) -> str:
mapping = {
ComparisonOperator.greater_than: 'gt',
ComparisonOperator.less_than: 'lt',
ComparisonOperator.greater_than_equal: 'gte',
ComparisonOperator.less_than_equal: 'lte',
}
return mapping.get(op, op.value)
def node_search_filter_query_constructor( def node_search_filter_query_constructor(
filters: SearchFilters, filters: SearchFilters,
provider: GraphProvider, provider: GraphProvider,
@ -259,7 +269,11 @@ def build_aoss_edge_filters(group_ids: list[str], search_filters: SearchFilters)
for and_group in ranges: for and_group in ranges:
and_filters = [] and_filters = []
for df in and_group: # df is a DateFilter for df in and_group: # df is a DateFilter
range_query = {'range': {field: {df.comparison_operator.value: df.date}}} range_query = {
'range': {
field: {cypher_to_opensearch_operator(df.comparison_operator): df.date}
}
}
and_filters.append(range_query) and_filters.append(range_query)
should_clauses.append({'bool': {'filter': and_filters}}) should_clauses.append({'bool': {'filter': and_filters}})
filters.append({'bool': {'should': should_clauses, 'minimum_should_match': 1}}) filters.append({'bool': {'should': should_clauses, 'minimum_should_match': 1}})