Priya-0914 commited on
Commit
2b371d5
·
verified ·
1 Parent(s): 19a9438

Create metadata_filtering.py

Browse files
Files changed (1) hide show
  1. metadata_filtering.py +24 -0
metadata_filtering.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from llama_index.core.vector_stores import (
2
+ MetadataFilters,
3
+ MetadataFilter,
4
+ FilterOperator,
5
+ FilterCondition,
6
+ )
7
+
8
+ def build_filters_from_query(query: str):
9
+ terms = [t.lower() for t in query.split() if len(t) > 2]
10
+
11
+ if not terms:
12
+ return None
13
+
14
+ return MetadataFilters(
15
+ filters=[
16
+ MetadataFilter(
17
+ key="excerpt_keywords",
18
+ operator=FilterOperator.TEXT_MATCH,
19
+ value=term,
20
+ )
21
+ for term in terms
22
+ ],
23
+ condition=FilterCondition.OR
24
+ )