|
1 | 1 | import { HF_HEADER_X_BILL_TO, HF_HUB_URL } from "../config.js"; |
2 | 2 | import { PACKAGE_NAME, PACKAGE_VERSION } from "../package.js"; |
3 | 3 | import type { InferenceTask, Options, RequestArgs } from "../types.js"; |
4 | | -import type { InferenceProviderModelMapping } from "./getInferenceProviderMapping.js"; |
| 4 | +import type { InferenceProviderMappingEntry } from "./getInferenceProviderMapping.js"; |
5 | 5 | import { getInferenceProviderMapping } from "./getInferenceProviderMapping.js"; |
6 | 6 | import type { getProviderHelper } from "./getProviderHelper.js"; |
7 | 7 | import { isUrl } from "./isUrl.js"; |
@@ -64,14 +64,15 @@ export async function makeRequestOptions( |
64 | 64 |
|
65 | 65 | const inferenceProviderMapping = providerHelper.clientSideRoutingOnly |
66 | 66 | ? ({ |
| 67 | + provider: provider, |
67 | 68 | // eslint-disable-next-line @typescript-eslint/no-non-null-assertion |
68 | 69 | providerId: removeProviderPrefix(maybeModel!, provider), |
69 | 70 | // eslint-disable-next-line @typescript-eslint/no-non-null-assertion |
70 | 71 | hfModelId: maybeModel!, |
71 | 72 | status: "live", |
72 | 73 | // eslint-disable-next-line @typescript-eslint/no-non-null-assertion |
73 | 74 | task: task!, |
74 | | - } satisfies InferenceProviderModelMapping) |
| 75 | + } satisfies InferenceProviderMappingEntry) |
75 | 76 | : await getInferenceProviderMapping( |
76 | 77 | { |
77 | 78 | modelId: hfModel, |
@@ -109,7 +110,7 @@ export function makeRequestOptionsFromResolvedModel( |
109 | 110 | data?: Blob | ArrayBuffer; |
110 | 111 | stream?: boolean; |
111 | 112 | }, |
112 | | - mapping: InferenceProviderModelMapping | undefined, |
| 113 | + mapping: InferenceProviderMappingEntry | undefined, |
113 | 114 | options?: Options & { |
114 | 115 | task?: InferenceTask; |
115 | 116 | } |
|
0 commit comments