mirror of
https://github.com/langchain-ai/create-agent-chat-app.git
synced 2026-07-01 21:24:02 -04:00
bracesproul
21 lines
783 B
TypeScript
21 lines
783 B
TypeScript
import { initChatModel } from "langchain/chat_models/universal";
|
|
|
|
/**
|
|
* Load a chat model from a fully specified name.
|
|
* @param fullySpecifiedName - String in the format 'provider/model' or 'provider/account/provider/model'.
|
|
* @returns A Promise that resolves to a BaseChatModel instance.
|
|
*/
|
|
export async function loadChatModel(
|
|
fullySpecifiedName: string,
|
|
): Promise<ReturnType<typeof initChatModel>> {
|
|
const index = fullySpecifiedName.indexOf("/");
|
|
if (index === -1) {
|
|
// If there's no "/", assume it's just the model
|
|
return await initChatModel(fullySpecifiedName);
|
|
} else {
|
|
const provider = fullySpecifiedName.slice(0, index);
|
|
const model = fullySpecifiedName.slice(index + 1);
|
|
return await initChatModel(model, { modelProvider: provider });
|
|
}
|
|
}
|