fix: avoid generating empty content for Stability (#56)

epam · Jan 12, 2024 · eeff435 · eeff435
1 parent 9f42f1e
commit eeff435
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/aidial_adapter_bedrock/llm/model/stability.py b/aidial_adapter_bedrock/llm/model/stability.py
@@ -44,7 +44,11 @@ class StabilityResponse(BaseModel):
 
     def content(self) -> str:
         self._throw_if_error()
-        return ""
+        # NOTE: text-to-text models aren't expected to generate empty strings.
+        # So since we represent text-to-image model (Stability) as
+        # a text-to-text model (via chat completion interface),
+        # we need to return something.
+        return " "
 
     def attachments(self) -> list[Attachment]:
         self._throw_if_error()