Skip to content

Commit 8a482e0

Browse files
committed
fix typing
1 parent 2cbdcec commit 8a482e0

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

packages/inference/src/snippets/python.ts

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import type { PipelineType, WidgetType } from "@huggingface/tasks/src/pipelines.js";
22
import type { ChatCompletionInputMessage, GenerationParameters } from "@huggingface/tasks/src/tasks/index.js";
33
import { type InferenceSnippet, type ModelDataMinimal, getModelInputSnippet } from "@huggingface/tasks";
4-
import type { InferenceProvider } from "../types";
4+
import type { InferenceProvider, InferenceTask, RequestArgs } from "../types";
55
import { Template } from "@huggingface/jinja";
66
import { makeRequestOptionsFromResolvedModel } from "../lib/makeRequestOptions";
77
import fs from "fs";
@@ -110,14 +110,14 @@ const snippetGenerator = (templateName: string, inputPreparationFn?: InputPrepar
110110
const inputs = inputPreparationFn ? inputPreparationFn(model, opts) : { inputs: getModelInputSnippet(model) };
111111
const request = makeRequestOptionsFromResolvedModel(
112112
providerModelId ?? model.id,
113-
{ accessToken: accessToken, model: providerModelId, provider: provider, ...inputs },
114-
{ chatCompletion: templateName.includes("conversational"), task: model.pipeline_tag }
113+
{ accessToken: accessToken, model: providerModelId, provider: provider, ...inputs } as RequestArgs,
114+
{ chatCompletion: templateName.includes("conversational"), task: model.pipeline_tag as InferenceTask }
115115
);
116116

117117
/// Prepare template injection data
118118
const params: TemplateParams = {
119119
accessToken,
120-
authorizationHeader: request.info.headers?.Authorization!,
120+
authorizationHeader: (request.info.headers as Record<string, string>)?.Authorization,
121121
baseUrl: removeSuffix(request.url, "/chat/completions"),
122122
fullUrl: request.url,
123123
inputs: {

0 commit comments

Comments
 (0)