Return empty embeddings list as empty list through gateway

This commit is contained in:
Cyber MacGeddon 2025-09-11 00:33:42 +01:00
parent 6a1cc61e52
commit 73d3f45103

View file

@ -81,7 +81,7 @@ class GraphEmbeddingsResponseTranslator(MessageTranslator):
def from_pulsar(self, obj: GraphEmbeddingsResponse) -> Dict[str, Any]:
result = {}
if obj.entities:
if obj.entities is not None:
result["entities"] = [
self.value_translator.from_pulsar(entity)
for entity in obj.entities
@ -91,4 +91,4 @@ class GraphEmbeddingsResponseTranslator(MessageTranslator):
def from_response_with_completion(self, obj: GraphEmbeddingsResponse) -> Tuple[Dict[str, Any], bool]:
"""Returns (response_dict, is_final)"""
return self.from_pulsar(obj), True
return self.from_pulsar(obj), True