From c360d2bda8fc518d2ba5c7be5b51ffbe1dede4cb Mon Sep 17 00:00:00 2001 From: Andrew Tait Gehrhardt Date: Tue, 18 Jun 2024 21:12:47 -0400 Subject: [PATCH] Cleaning up --- .../dynamic_ollama_vision_filter_pipeline.py | 14 ++------------ 1 file changed, 2 insertions(+), 12 deletions(-) diff --git a/examples/filters/dynamic_ollama_vision_filter_pipeline.py b/examples/filters/dynamic_ollama_vision_filter_pipeline.py index d5458cd..9eb01d2 100644 --- a/examples/filters/dynamic_ollama_vision_filter_pipeline.py +++ b/examples/filters/dynamic_ollama_vision_filter_pipeline.py @@ -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