47 lines
1.5 KiB
TypeScript
47 lines
1.5 KiB
TypeScript
import type { PluginInput } from "@opencode-ai/plugin"
|
|
import {
|
|
findNearestMessageWithFields,
|
|
findNearestMessageWithFieldsFromSDK,
|
|
} from "../../features/hook-message-injector"
|
|
import { getMessageDir, isSqliteBackend } from "../../shared"
|
|
import type { ModelInfo } from "./types"
|
|
|
|
export async function resolveRecentModelForSession(
|
|
ctx: PluginInput,
|
|
sessionID: string
|
|
): Promise<ModelInfo | undefined> {
|
|
try {
|
|
const messagesResp = await ctx.client.session.messages({ path: { id: sessionID } })
|
|
const messages = (messagesResp.data ?? []) as Array<{
|
|
info?: { model?: ModelInfo; modelID?: string; providerID?: string }
|
|
}>
|
|
|
|
for (let i = messages.length - 1; i >= 0; i--) {
|
|
const info = messages[i].info
|
|
const model = info?.model
|
|
if (model?.providerID && model?.modelID) {
|
|
return { providerID: model.providerID, modelID: model.modelID }
|
|
}
|
|
|
|
if (info?.providerID && info?.modelID) {
|
|
return { providerID: info.providerID, modelID: info.modelID }
|
|
}
|
|
}
|
|
} catch {
|
|
// ignore - fallback to message storage
|
|
}
|
|
|
|
let currentMessage = null
|
|
if (isSqliteBackend()) {
|
|
currentMessage = await findNearestMessageWithFieldsFromSDK(ctx.client, sessionID)
|
|
} else {
|
|
const messageDir = getMessageDir(sessionID)
|
|
currentMessage = messageDir ? findNearestMessageWithFields(messageDir) : null
|
|
}
|
|
const model = currentMessage?.model
|
|
if (!model?.providerID || !model?.modelID) {
|
|
return undefined
|
|
}
|
|
return { providerID: model.providerID, modelID: model.modelID }
|
|
}
|