UNPKG

@huggingface/inference

Version:

Typescript client for the Hugging Face Inference Providers and Inference Endpoints

30 lines (29 loc) 1.2 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.FireworksConversationalTask = void 0; /** * See the registered mapping of HF model ID => Fireworks model ID here: * * https://huggingface.co/api/partners/fireworks/models * * This is a publicly available mapping. * * If you want to try to run inference for a new model locally before it's registered on huggingface.co, * you can add it to the dictionary "HARDCODED_MODEL_ID_MAPPING" in consts.ts, for dev purposes. * * - If you work at Fireworks and want to update this mapping, please use the model mapping API we provide on huggingface.co * - If you're a community member and want to add a new supported HF model to Fireworks, please open an issue on the present repo * and we will tag Fireworks team members. * * Thanks! */ const providerHelper_js_1 = require("./providerHelper.js"); class FireworksConversationalTask extends providerHelper_js_1.BaseConversationalTask { constructor() { super("fireworks-ai", "https://api.fireworks.ai"); } makeRoute() { return "/inference/v1/chat/completions"; } } exports.FireworksConversationalTask = FireworksConversationalTask;