Merge pull request #13522 from salbahra/bug/fix-missing-api-config

fix: Correct missing `api_config` for embedding Ollama proxy endpoints
This commit is contained in:
Tim Jaeryang Baek 2025-05-05 23:49:40 +04:00 committed by GitHub
commit 0a8199a056
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -883,6 +883,10 @@ async def embed(
)
url = request.app.state.config.OLLAMA_BASE_URLS[url_idx]
api_config = request.app.state.config.OLLAMA_API_CONFIGS.get(
str(url_idx),
request.app.state.config.OLLAMA_API_CONFIGS.get(url, {}), # Legacy support
)
key = get_api_key(url_idx, url, request.app.state.config.OLLAMA_API_CONFIGS)
prefix_id = api_config.get("prefix_id", None)
@ -966,6 +970,10 @@ async def embeddings(
)
url = request.app.state.config.OLLAMA_BASE_URLS[url_idx]
api_config = request.app.state.config.OLLAMA_API_CONFIGS.get(
str(url_idx),
request.app.state.config.OLLAMA_API_CONFIGS.get(url, {}), # Legacy support
)
key = get_api_key(url_idx, url, request.app.state.config.OLLAMA_API_CONFIGS)
prefix_id = api_config.get("prefix_id", None)