mirror of
https://github.com/infiniflow/ragflow.git
synced 2025-12-08 20:42:30 +08:00
Update cv_model.py (#9472)
### What problem does this PR solve? https://github.com/infiniflow/ragflow/issues/9452 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
@ -372,6 +372,16 @@ class OllamaCV(Base):
|
|||||||
self.keep_alive = kwargs.get("ollama_keep_alive", int(os.environ.get("OLLAMA_KEEP_ALIVE", -1)))
|
self.keep_alive = kwargs.get("ollama_keep_alive", int(os.environ.get("OLLAMA_KEEP_ALIVE", -1)))
|
||||||
Base.__init__(self, **kwargs)
|
Base.__init__(self, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
def _clean_img(self, img):
|
||||||
|
if not isinstance(img, str):
|
||||||
|
return img
|
||||||
|
|
||||||
|
#remove the header like "data/*;base64,"
|
||||||
|
if img.startswith("data:") and ";base64," in img:
|
||||||
|
img = img.split(";base64,")[1]
|
||||||
|
return img
|
||||||
|
|
||||||
def _clean_conf(self, gen_conf):
|
def _clean_conf(self, gen_conf):
|
||||||
options = {}
|
options = {}
|
||||||
if "temperature" in gen_conf:
|
if "temperature" in gen_conf:
|
||||||
@ -390,9 +400,12 @@ class OllamaCV(Base):
|
|||||||
hist.insert(0, {"role": "system", "content": system})
|
hist.insert(0, {"role": "system", "content": system})
|
||||||
if not images:
|
if not images:
|
||||||
return hist
|
return hist
|
||||||
|
temp_images = []
|
||||||
|
for img in images:
|
||||||
|
temp_images.append(self._clean_img(img))
|
||||||
for his in hist:
|
for his in hist:
|
||||||
if his["role"] == "user":
|
if his["role"] == "user":
|
||||||
his["images"] = images
|
his["images"] = temp_images
|
||||||
break
|
break
|
||||||
return hist
|
return hist
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user