patch_ollama_embeddings.py 924 B

12345678910111213141516171819202122232425262728
  1. from pathlib import Path
  2. p = Path("/app/backend/open_webui/retrieval/utils.py")
  3. txt = p.read_text()
  4. # 1) /api/embed -> /api/embeddings
  5. txt = txt.replace("/api/embed", "/api/embeddings")
  6. # 2) asenda embeddingu plokk ühtse versiooniga
  7. old_block = """ data = await r.json()
  8. if "embeddings" in data:
  9. return data["embeddings"]
  10. else:
  11. raise Exception("Something went wrong :/")"""
  12. new_block = """ data = await r.json()
  13. if "embeddings" in data:
  14. return data["embeddings"]
  15. elif "embedding" in data:
  16. # Ollama: üksiku embeddingu vastus
  17. return [data["embedding"]]
  18. else:
  19. raise Exception("Something went wrong :/")"""
  20. if old_block in txt:
  21. txt = txt.replace(old_block, new_block)
  22. p.write_text(txt)