aidial_adapter_openai/endpoints/embeddings.py

from fastapi import Request

from aidial_adapter_openai.dial_api.storage import create_file_storage
from aidial_adapter_openai.embeddings.azure_ai_vision import (
    embeddings as azure_ai_vision_embeddings,
)
from aidial_adapter_openai.embeddings.openai import (
    embeddings as openai_embeddings,
)
from aidial_adapter_openai.utils.auth import get_credentials
from aidial_adapter_openai.utils.parsers import parse_body
from aidial_adapter_openai.utils.request import (
    get_api_version,
    get_request_app_config,
)


async def embedding(deployment_id: str, request: Request):
    app_config = get_request_app_config(request)
    data = await parse_body(request)

    # See note for /chat/completions endpoint
    data["model"] = deployment_id

    creds = await get_credentials(request)
    api_version = get_api_version(request)
    upstream_endpoint = request.headers["X-UPSTREAM-ENDPOINT"]

    if deployment_id in app_config.AZURE_AI_VISION_DEPLOYMENTS:
        storage = create_file_storage("images", request.headers)
        return await azure_ai_vision_embeddings(
            creds, deployment_id, upstream_endpoint, storage, data
        )

    return await openai_embeddings(creds, upstream_endpoint, api_version, data)