fix: regext empty prompt in Gemini Pro Vision (#43)

epam · Jan 15, 2024 · dbe4401 · dbe4401
1 parent 1de1c1d
commit dbe4401
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/aidial_adapter_vertexai/llm/gemini_prompt.py b/aidial_adapter_vertexai/llm/gemini_prompt.py
@@ -77,6 +77,11 @@ async def parse_vision(
         if image_count == 0:
             return UserError("No image inputs were found", usage_message)
 
+        if any(msg.has_empty_content() for msg in download_result):
+            return UserError(
+                "Messages with empty prompts are not allowed", usage_message
+            )
+
         history = list(map(to_content, download_result))
         return cls(history=history[:-1], prompt=history[-1].parts)
 

diff --git a/aidial_adapter_vertexai/llm/process_inputs.py b/aidial_adapter_vertexai/llm/process_inputs.py
@@ -20,6 +20,9 @@ class MessageWithInputs(BaseModel):
     message: Message
     image_inputs: List[ImageDataURL]
 
+    def has_empty_content(self) -> bool:
+        return (self.message.content or "").strip() == ""
+
 
 class DownloadErrors(BaseModel):
     """Download errors for a particular message"""