fix: types

This commit is contained in:
Gab
2026-03-30 11:36:11 +11:00
parent 478d930c4a
commit 43048f89c5

View File

@@ -1,5 +1,10 @@
import type { LanguageModelV2 } from "@ai-sdk/provider" import type { LanguageModelV2 } from "@ai-sdk/provider"
import { type FetchFunction, withoutTrailingSlash, safeParseJSON, EventSourceParserStream } from "@ai-sdk/provider-utils" import {
type FetchFunction,
withoutTrailingSlash,
safeParseJSON,
EventSourceParserStream,
} from "@ai-sdk/provider-utils"
import { OpenAICompatibleChatLanguageModel } from "../copilot/chat/openai-compatible-chat-language-model" import { OpenAICompatibleChatLanguageModel } from "../copilot/chat/openai-compatible-chat-language-model"
import { Log } from "@/util/log" import { Log } from "@/util/log"
@@ -127,7 +132,7 @@ export function createToothFairyAI(options: ToothFairyAIProviderSettings = {}):
return return
} }
const parsed = await safeParseJSON({ text: data, schema: null }) const parsed = await safeParseJSON({ text: data })
if (!parsed.success) { if (!parsed.success) {
log.error("Failed to parse SSE chunk", { log.error("Failed to parse SSE chunk", {
@@ -138,7 +143,11 @@ export function createToothFairyAI(options: ToothFairyAIProviderSettings = {}):
return return
} }
const value = parsed.value const value = parsed.value as {
status?: string
choices?: Array<{ finish_reason?: string }>
usage?: { prompt_tokens?: number; completion_tokens?: number; total_tokens?: number }
}
if (value.status === "initialising" || value.status === "connected") { if (value.status === "initialising" || value.status === "connected") {
log.debug("filtered connection status", { status: value.status }) log.debug("filtered connection status", { status: value.status })