UNPKG

create-agent-chat-app

Version:
33 lines (29 loc) 868 B
/** * Define the configurable parameters for the agent. */ import { Annotation } from "@langchain/langgraph"; import { SYSTEM_PROMPT_TEMPLATE } from "./prompts.js"; import { RunnableConfig } from "@langchain/core/runnables"; export const ConfigurationSchema = Annotation.Root({ /** * The system prompt to be used by the agent. */ systemPromptTemplate: Annotation<string>, /** * The name of the language model to be used by the agent. */ model: Annotation<string>, }); export function ensureConfiguration( config: RunnableConfig, ): typeof ConfigurationSchema.State { /** * Ensure the defaults are populated. */ const configurable = config.configurable ?? {}; return { systemPromptTemplate: configurable.systemPromptTemplate ?? SYSTEM_PROMPT_TEMPLATE, model: configurable.model ?? "claude-3-7-sonnet-latest", }; }