Skip to content

Commit ba5b91b

Browse files
committed
fix
1 parent 50e8d8e commit ba5b91b

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

packages/inference/src/lib/makeRequestOptions.ts

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import { HF_HEADER_X_BILL_TO, HF_HUB_URL } from "../config.js";
22
import { PACKAGE_NAME, PACKAGE_VERSION } from "../package.js";
33
import type { InferenceTask, Options, RequestArgs } from "../types.js";
4-
import type { InferenceProviderModelMapping } from "./getInferenceProviderMapping.js";
4+
import type { InferenceProviderMappingEntry } from "./getInferenceProviderMapping.js";
55
import { getInferenceProviderMapping } from "./getInferenceProviderMapping.js";
66
import type { getProviderHelper } from "./getProviderHelper.js";
77
import { isUrl } from "./isUrl.js";
@@ -64,14 +64,15 @@ export async function makeRequestOptions(
6464

6565
const inferenceProviderMapping = providerHelper.clientSideRoutingOnly
6666
? ({
67+
provider: provider,
6768
// eslint-disable-next-line @typescript-eslint/no-non-null-assertion
6869
providerId: removeProviderPrefix(maybeModel!, provider),
6970
// eslint-disable-next-line @typescript-eslint/no-non-null-assertion
7071
hfModelId: maybeModel!,
7172
status: "live",
7273
// eslint-disable-next-line @typescript-eslint/no-non-null-assertion
7374
task: task!,
74-
} satisfies InferenceProviderModelMapping)
75+
} satisfies InferenceProviderMappingEntry)
7576
: await getInferenceProviderMapping(
7677
{
7778
modelId: hfModel,
@@ -109,7 +110,7 @@ export function makeRequestOptionsFromResolvedModel(
109110
data?: Blob | ArrayBuffer;
110111
stream?: boolean;
111112
},
112-
mapping: InferenceProviderModelMapping | undefined,
113+
mapping: InferenceProviderMappingEntry | undefined,
113114
options?: Options & {
114115
task?: InferenceTask;
115116
}

0 commit comments

Comments
 (0)