42 lines
687 B
TypeScript
42 lines
687 B
TypeScript
|
import { ChatChromeAI } from "./ChatChromeAi"
|
||
|
|
import { ChatOllama } from "./ChatOllama"
|
||
|
|
|
||
|
|
export const pageAssistModel = async ({
|
||
|
|
model,
|
||
|
|
baseUrl,
|
||
|
|
keepAlive,
|
||
|
|
temperature,
|
||
|
|
topK,
|
||
|
|
topP,
|
||
|
|
numCtx,
|
||
|
|
seed
|
||
|
|
}: {
|
||
|
|
model: string
|
||
|
|
baseUrl: string
|
||
|
|
keepAlive: string
|
||
|
|
temperature: number
|
||
|
|
topK: number
|
||
|
|
topP: number
|
||
|
|
numCtx: number
|
||
|
|
seed: number
|
||
|
|
}) => {
|
||
|
|
switch (model) {
|
||
|
|
case "chrome::gemini-nano::page-assist":
|
||
|
|
return new ChatChromeAI({
|
||
|
|
temperature,
|
||
|
|
topK
|
||
|
|
})
|
||
|
|
default:
|
||
|
|
return new ChatOllama({
|
||
|
|
baseUrl,
|
||
|
|
keepAlive,
|
||
|
|
temperature,
|
||
|
|
topK,
|
||
|
|
topP,
|
||
|
|
numCtx,
|
||
|
|
seed,
|
||
|
|
model
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|