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

Fixed metadata. Fixed image retention.

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

47
examples/filters/dynamic_ollama_vision_filter_pipeline.py
View File

@ -1,10 +1,10 @@
"""
title: HomeAssistant Filter Pipeline
title: Ollama Dynamic Vision Pipeline
author: Andrew Tait Gehrhardt
date: 2024-06-15
date: 2024-06-18
version: 1.0
license: MIT
description: A pipeline for controlling Home Assistant entities based on their easy names. Only supports lights at the moment.
description: A pipeline for dynamically processing images when current model is a text only model
requirements: pydantic, aiohttp
"""
@ -12,15 +12,15 @@ from typing import List, Optional
from pydantic import BaseModel
import json
import aiohttp
from utils.pipelines.main import get_last_user_message, get_last_assistant_message
from utils.pipelines.main import get_last_user_message
class Pipeline:
class Valves(BaseModel):
pipelines: List[str] = []
priority: int = 0
target_user_roles: List[str] = ["admin", "user"]
vision_model: str = "llava"
ollama_base_url: str = "http://host.docker.interal:11434"
ollama_base_url: str = ""
model_to_override: str = ""
def __init__(self):
self.type = "filter"
@ -72,31 +72,26 @@ class Pipeline:
# Ensure the body is a dictionary
if isinstance(body, str):
body = json.loads(body)
model = body.get("model", "")
print(f"MODEL NAME: {model}")
if user and user.get("role", "admin") in self.valves.target_user_roles:
# 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"])
# Get the content of the most recent message
if messages:
user_message = get_last_user_message(body["messages"])
print("CURRENT MESSAGE:", user_message)
# Process the images with LLava
if images:
print("IMAGES: True")
llava_response = await self.process_images_with_llava(images, user_message, self.valves.vision_model,self.valves.ollama_base_url)
print("LLAVA RESPONSE:", llava_response)
# Override the content for the user role
for message in messages:
if message.get("role") == "user":
message["content"] = llava_response
else:
print("IMAGES: False")
print("IMAGES: True")
llava_response = await self.process_images_with_llava(images, user_message, self.valves.vision_model,self.valves.ollama_base_url)
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: