Skip to content

Commit 517bef3

Browse files
authored
fix: handle empty requests in milvus search to avoid connection attempts (#36388)
Prevent unnecessary connection attempts by returning early when receiving empty requests
1 parent 6043660 commit 517bef3

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

sdks/python/apache_beam/ml/rag/enrichment/milvus_search.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -417,6 +417,9 @@ def __enter__(self):
417417
def __call__(self, request: Union[Chunk, List[Chunk]], *args,
418418
**kwargs) -> List[Tuple[Chunk, Dict[str, Any]]]:
419419
reqs = request if isinstance(request, list) else [request]
420+
# Early return for empty requests to avoid unnecessary connection attempts
421+
if not reqs:
422+
return []
420423
search_result = self._search_documents(reqs)
421424
return self._get_call_response(reqs, search_result)
422425

0 commit comments

Comments
 (0)