mirror of
https://github.com/open-webui/open-webui.git
synced 2026-05-06 02:48:13 -05:00
[PR #4715] [MERGED] feat: support ollama batch processing embeddings #37174
Reference in New Issue
Block a user
Delete Branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
📋 Pull Request Information
Original PR: https://github.com/open-webui/open-webui/pull/4715
Author: @Peter-De-Ath
Created: 8/18/2024
Status: ✅ Merged
Merged: 10/8/2024
Merged by: @tjbck
Base:
dev← Head:ollama-batch-embeddings📝 Commits (3)
885b9f1refactor: Update GenerateEmbeddingsForm to support batch processing3fcdca2backwards comp: use RAG_EMBEDDING_OPENAI_BATCH_SIZEb242460formatting📊 Changes
6 files changed (+84 additions, -97 deletions)
View changed files
📝
backend/open_webui/apps/ollama/main.py(+29 -55)📝
backend/open_webui/apps/retrieval/main.py(+10 -14)📝
backend/open_webui/apps/retrieval/utils.py(+25 -17)📝
backend/open_webui/config.py(+7 -4)📝
src/lib/apis/retrieval/index.ts(+1 -1)📝
src/lib/components/admin/Settings/Documents.svelte(+12 -6)📄 Description
feat: support batch embedding with ollama
Pull Request Checklist
Note to first-time contributors: Please open a discussion post in Discussions and describe your changes before submitting a pull request.
Before submitting, make sure you've checked the following:
devbranch.Changelog Entry
Description
Add support for batch embedding with ollama api
Added
Support for batch embedding in a single ollama request
Removed
RAG_EMBEDDING_OPENAI_BATCH_SIZE environment variable
Breaking Changes
changed RAG_EMBEDDING_OPENAI_BATCH_SIZE to RAG_EMBEDDING_BATCH_SIZE
Screenshots or Videos
🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.