allow multi-track drifting (multiple images to retcon)

2026-03-30 13:06:31 -04:00
parent 677ea83fa1
commit 0003ac29e9
2 changed files with 8 additions and 7 deletions
--- a/vibe_bot/llama_wrapper.py
+++ b/vibe_bot/llama_wrapper.py
@@ -5,8 +5,6 @@
 import openai
 from typing import Iterable
 from openai.types.chat import ChatCompletionMessageParam
-from openai._types import FileTypes, SequenceNotStr
-from typing import Union
 from io import BufferedReader, BytesIO


@@ -138,7 +136,7 @@ def image_generation(prompt: str, openai_url: str, openai_api_key: str, n=1) ->


 def image_edit(
-    image: BufferedReader | BytesIO,
+    image: BufferedReader | BytesIO | list[BufferedReader] | list[BytesIO],
    prompt: str,
    openai_url: str,
    openai_api_key: str,