mini: fix default model (#35515)
This commit is contained in:
@@ -20,6 +20,8 @@ type Turn = {
|
||||
export type RunSession = {
|
||||
first: boolean
|
||||
turns: Turn[]
|
||||
model?: NonNullable<RunInput["model"]>
|
||||
variant?: string
|
||||
}
|
||||
|
||||
function fileName(url: string, filename?: string) {
|
||||
@@ -157,9 +159,11 @@ export async function resolveCurrentSession(
|
||||
sessionID: string,
|
||||
limit = LIMIT,
|
||||
): Promise<RunSession> {
|
||||
const response = await sdk.v2.session.messages({ sessionID, limit, order: "desc" }, { throwOnError: true })
|
||||
const [response, session] = await Promise.all([
|
||||
sdk.v2.session.messages({ sessionID, limit, order: "desc" }, { throwOnError: true }),
|
||||
sdk.v2.session.get({ sessionID }, { throwOnError: true }),
|
||||
])
|
||||
const messages = response.data.data.toReversed()
|
||||
const session = await sdk.v2.session.get({ sessionID }, { throwOnError: true })
|
||||
return {
|
||||
first: messages.length === 0,
|
||||
turns: messages.flatMap((message) => {
|
||||
@@ -197,6 +201,13 @@ export async function resolveCurrentSession(
|
||||
},
|
||||
]
|
||||
}),
|
||||
...(session.data.data.model && {
|
||||
model: {
|
||||
providerID: session.data.data.model.providerID,
|
||||
modelID: session.data.data.model.id,
|
||||
},
|
||||
variant: session.data.data.model.variant,
|
||||
}),
|
||||
}
|
||||
}
|
||||
|
||||
@@ -223,6 +234,10 @@ export function sessionVariant(session: RunSession, model: RunInput["model"]): s
|
||||
return undefined
|
||||
}
|
||||
|
||||
if (session.model?.providerID === model.providerID && session.model.modelID === model.modelID) {
|
||||
return session.variant
|
||||
}
|
||||
|
||||
for (let idx = session.turns.length - 1; idx >= 0; idx -= 1) {
|
||||
const turn = session.turns[idx]
|
||||
if (turn.provider !== model.providerID || turn.model !== model.modelID) {
|
||||
|
||||
Reference in New Issue
Block a user