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 0f1d3ce8cf
commit 47e0c6e37a

View file

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