| 12345678910111213141516171819202122232425262728 |
- 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)
|