OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-10 15:40:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Cleaning up

Browse Source
This commit is contained in:
Andrew Tait Gehrhardt 2024-06-18 21:12:47 -04:00
parent 7d817b6b6a
commit c360d2bda8
1 changed files with 2 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
examples/filters/dynamic_ollama_vision_filter_pipeline.py
View File

@ -74,28 +74,18 @@ class Pipeline:
body = json.loads(body)
model = body.get("model", "")
print(f"MODEL NAME: {model}")
# Get the content of the most recent message
user_message = get_last_user_message(body["messages"])
print("CURRENT MESSAGE:", user_message)
if model in self.valves.model_to_override:
messages = body.get("messages", [])
for message in messages:
if "images" in message:
images.extend(message["images"])
print("IMAGES: True")
llava_response = await self.process_images_with_llava(images, user_message, self.valves.vision_model,self.valves.ollama_base_url)
raw_llava_response = await self.process_images_with_llava(images, user_message, self.valves.vision_model,self.valves.ollama_base_url)
llava_response = f"REPEAT THIS BACK: {raw_llava_response}"
message["content"] = llava_response
print("LLAVA RESPONSE:", llava_response)
message.pop("images", None) # This will safely remove the 'images' key if it exists
else:
print("IMAGES: False")
print(f"""
THIS IS THE BODY OBJECT:
{body}
""")
return body