| import type { QuestionAnsweringInput, QuestionAnsweringOutput } from "@huggingface/tasks"; |
|
|
| import { resolveProvider } from "../../lib/getInferenceProviderMapping.js"; |
| import { getProviderHelper } from "../../lib/getProviderHelper.js"; |
| import type { BaseArgs, Options } from "../../types.js"; |
| import { innerRequest } from "../../utils/request.js"; |
|
|
| export type QuestionAnsweringArgs = BaseArgs & QuestionAnsweringInput; |
|
|
| |
| |
| |
| export async function questionAnswering( |
| args: QuestionAnsweringArgs, |
| options?: Options |
| ): Promise<QuestionAnsweringOutput[number]> { |
| const provider = await resolveProvider(args.provider, args.model, args.endpointUrl); |
| const providerHelper = getProviderHelper(provider, "question-answering"); |
| const { data: res } = await innerRequest<QuestionAnsweringOutput | QuestionAnsweringOutput[number]>( |
| args, |
| providerHelper, |
| { |
| ...options, |
| task: "question-answering", |
| } |
| ); |
| return providerHelper.getResponse(res); |
| } |
|
|