pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: correct single base64 image handling in image prompt (#9220)

Browse Source 
### What problem does this PR solve?

Correct single base64 image handling in image prompt.


![img_v3_02or_ec4757c2-a9d4-4774-9a76-f7c6be633ebg](https://github.com/user-attachments/assets/872a86bf-e2a8-48d1-9b71-2a0c7a35ba9e)

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Yongteng Lei
2025-08-05 09:26:42 +08:00
committed by GitHub
parent dda5a0080a
commit 30ccc4a66c
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rag/llm/cv_model.py
View File

@ -59,6 +59,10 @@ class Base(ABC):
def _image_prompt(self, text, images): def _image_prompt(self, text, images):
if not images: if not images:
return text return text
if isinstance(images, str):
images = [images]
pmpt = [{"type": "text", "text": text}] pmpt = [{"type": "text", "text": text}]
for img in images: for img in images:
pmpt.append({ pmpt.append({
@ -795,4 +799,4 @@ class GoogleCV(AnthropicCV, GeminiCV):
yield ans yield ans
else: else:
for ans in GeminiCV.chat_streamly(self, system, history, gen_conf, images): for ans in GeminiCV.chat_streamly(self, system, history, gen_conf, images):
yield ans yield ans