references/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-06-14 03:30:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-06-01 13:10:19 -07:00
parent d64ef1803d
commit eb4eebc3ce
2 changed files with 37 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/routers/openai.py
+33 -3
View File
@@ -687,9 +687,18 @@ async def verify_connection(
if auth_type not in ('azure_ad', 'microsoft_entra_id'):
headers['api-key'] = key
api_version = api_config.get('api_version', '') or '2023-03-15-preview'
# Azure v1 format: base URL already ends with /openai/v1,
# use standard /models endpoint without api-version.
is_azure_v1 = bool(re.search(r'/openai/v1(?:/|$)', url))
if is_azure_v1:
verify_url = f'{url.rstrip("/")}/models'
else:
api_version = api_config.get('api_version', '') or '2023-03-15-preview'
verify_url = f'{url}/openai/models?api-version={api_version}'
async with session.get(
url=f'{url}/openai/models?api-version={api_version}',
url=verify_url,
headers=headers,
cookies=cookies,
ssl=AIOHTTP_CLIENT_SESSION_SSL,
@@ -1305,11 +1314,32 @@ async def embeddings(request: Request, form_data: dict, user):
streaming = False
headers, cookies = await get_headers_and_cookies(request, url, key, api_config, user=user)
if api_config.get('azure') or api_config.get('provider') == 'azure':
# Only set api-key header if not using Azure Entra ID authentication
auth_type = api_config.get('auth_type', 'bearer')
if auth_type not in ('azure_ad', 'microsoft_entra_id'):
headers['api-key'] = key
# Azure v1 format: base URL already ends with /openai/v1,
# model stays in the payload, no deployment URL rewriting.
is_azure_v1 = bool(re.search(r'/openai/v1(?:/|$)', url))
if is_azure_v1:
embeddings_url = f'{url.rstrip("/")}/embeddings'
else:
api_version = api_config.get('api_version', '2023-03-15-preview')
model = _sanitize_model_for_url(form_data.get('model', ''))
embeddings_url = f'{url}/openai/deployments/{model}/embeddings?api-version={api_version}'
headers['api-version'] = api_version
else:
embeddings_url = f'{url}/embeddings'
try:
session = await get_session()
r = await session.request(
method='POST',
url=f'{url}/embeddings',
url=embeddings_url,
data=body,
headers=headers,
cookies=cookies,
src/lib/components/AddConnectionModal.svelte
+4 -5
View File
@@ -41,7 +41,8 @@
provider === 'azure' ||
((url.includes('azure.') || url.includes('cognitive.microsoft.com')) &&
!direct &&
provider === '');
provider === '' &&
!/\/openai\/v1(\/|$)/.test(url));
let prefixId = '';
let enable = true;
@@ -392,9 +393,7 @@
<option value="session">{$i18n.t('Session')}</option>
{#if !direct}
<option value="system_oauth">{$i18n.t('OAuth')}</option>
{#if azure}
<option value="microsoft_entra_id">{$i18n.t('Entra ID')}</option>
{/if}
<option value="microsoft_entra_id">{$i18n.t('Entra ID')}</option>
{/if}
{/if}
</select>
@@ -523,7 +522,7 @@
<label
for="api-version-input"
class={`mb-0.5 text-xs text-gray-500
${($settings?.highContrastMode ?? false) ? 'text-gray-800 dark:text-gray-100' : ''}`}
${($settings?.highContrastMode ?? false) ? 'text-gray-800 dark:text-gray-100' : ''}`}
>{$i18n.t('API Version')}</label
>