ToothFairyAI_Public/tf_code
1
0
Fork 0
mirror of https://gitea.toothfairyai.com/ToothFairyAI/tf_code.git synced 2026-04-25 18:14:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: types

Browse Source
This commit is contained in:
Gab
2026-03-30 11:36:11 +11:00
parent 478d930c4a
commit 43048f89c5
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
packages/tfcode/src/provider/sdk/toothfairyai/toothfairyai-provider.ts
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,8 +132,8 @@ 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", {
chunk: data.slice(0, 100), chunk: data.slice(0, 100),
@@ -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 })