from pathlib import Path

p = Path("/app/backend/open_webui/retrieval/utils.py")
txt = p.read_text()

# 1) /api/embed -> /api/embeddings
txt = txt.replace("/api/embed", "/api/embeddings")

# 2) asenda embeddingu plokk ühtse versiooniga
old_block = """                data = await r.json()
                if "embeddings" in data:
                    return data["embeddings"]
                else:
                    raise Exception("Something went wrong :/")"""

new_block = """                data = await r.json()
                if "embeddings" in data:
                    return data["embeddings"]
                elif "embedding" in data:
                    # Ollama: üksiku embeddingu vastus
                    return [data["embedding"]]
                else:
                    raise Exception("Something went wrong :/")"""

if old_block in txt:
    txt = txt.replace(old_block, new_block)

p.write_text(txt)