39 lines
1.3 KiB
TypeScript
39 lines
1.3 KiB
TypeScript
|
import type { PluginInput } from "@opencode-ai/plugin"
|
||
|
|
import { findNearestMessageWithFields } from "../../features/hook-message-injector"
|
||
|
|
import { getMessageDir } from "../../shared/session-utils"
|
||
|
|
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
|
||
|
|
}
|
||
|
|
|
||
|
|
const messageDir = getMessageDir(sessionID)
|
||
|
|
const currentMessage = messageDir ? findNearestMessageWithFields(messageDir) : null
|
||
|
|
const model = currentMessage?.model
|
||
|
|
if (!model?.providerID || !model?.modelID) {
|
||
|
|
return undefined
|
||
|
|
}
|
||
|
|
return { providerID: model.providerID, modelID: model.modelID }
|
||
|
|
}
|