UNPKG

@huggingface/tasks

Version:
29 lines (28 loc) 900 B
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.HF_HUB_INFERENCE_PROXY_TEMPLATE = void 0; exports.openAIbaseUrl = openAIbaseUrl; /// This list is for illustration purposes only. /// in the `tasks` sub-package, we do not need actual strong typing of the inference providers. const INFERENCE_PROVIDERS = [ "cerebras", "cohere", "fal-ai", "fireworks-ai", "hf-inference", "hyperbolic", "ovhcloud", "replicate", "sambanova", "together", ]; exports.HF_HUB_INFERENCE_PROXY_TEMPLATE = `https://router.huggingface.co/{{PROVIDER}}`; /** * URL to set as baseUrl in the OpenAI SDK. * * TODO(Expose this from InferenceClient in the future?) */ function openAIbaseUrl(provider) { const url = exports.HF_HUB_INFERENCE_PROXY_TEMPLATE.replace("{{PROVIDER}}", provider); return provider === "hf-inference" ? `${url}/v1` : url; }