File tree 1 file changed +11
-2
lines changed
1 file changed +11
-2
lines changed Original file line number Diff line number Diff line change @@ -67,9 +67,10 @@ def construct_query(self):
67
67
self .error (f"elastic_query unexpectedly blank" )
68
68
69
69
self .query_to_provider = elastic_query
70
+ logger .debug (f"Constructed query_to_provider: { self .query_to_provider } " )
70
71
return
71
72
72
- def execute_search (self , session = None ):
73
+ def execute_search (self , size , session = None ):
73
74
74
75
logger .debug (f"{ self } : execute_search()" )
75
76
@@ -129,10 +130,18 @@ def execute_search(self, session=None):
129
130
else :
130
131
self .status = "ERR_NO_QUERY_SPECIFIED"
131
132
return
133
+
134
+ # Extract size (int) - Optional
135
+ size_pattern = r"size=(\d+)"
136
+ match = re .search (size_pattern , self .query_to_provider )
137
+ if match :
138
+ size = int (match .group (1 ))
139
+ else :
140
+ size = 10 # Default size if not specified
132
141
133
142
response = None
134
143
try :
135
- response = es .search (index = index , query = query )
144
+ response = es .search (index = index , query = query , size = size )
136
145
except ConnectionError as err :
137
146
self .error (f"es.search reports: { err } " )
138
147
except NotFoundError :
You can’t perform that action at this time.
0 commit comments