langchain
Version:
Typescript bindings for langchain
49 lines (48 loc) • 2.23 kB
TypeScript
import { AgentBuiltInState, Runtime } from "../runtime.js";
import { AgentMiddleware } from "./types.js";
import { SystemMessage } from "@langchain/core/messages";
import * as _langchain_core_tools3 from "@langchain/core/tools";
//#region src/agents/middleware/dynamicSystemPrompt.d.ts
type DynamicSystemPromptMiddlewareConfig<TContextSchema> = (state: AgentBuiltInState, runtime: Runtime<TContextSchema>) => string | SystemMessage | Promise<string | SystemMessage>;
/**
* Dynamic System Prompt Middleware
*
* Allows setting the system prompt dynamically right before each model invocation.
* Useful when the prompt depends on the current agent state or per-invocation context.
*
* @typeParam TContextSchema - The shape of the runtime context available at invocation time.
* If your agent defines a `contextSchema`, pass the inferred type here to get full type-safety
* for `runtime.context`.
*
* @param fn - Function that receives the current agent `state` and `runtime`, and
* returns the system prompt for the next model call as a string.
*
* @returns A middleware instance that sets `systemPrompt` for the next model call.
*
* @example Basic usage with typed context
* ```ts
* import { z } from "zod";
* import { dynamicSystemPrompt } from "langchain";
* import { createAgent, SystemMessage } from "langchain";
*
* const contextSchema = z.object({ region: z.string().optional() });
*
* const middleware = dynamicSystemPrompt<z.infer<typeof contextSchema>>(
* (_state, runtime) => `You are a helpful assistant. Region: ${runtime.context.region ?? "n/a"}`
* );
*
* const agent = createAgent({
* model: "anthropic:claude-3-5-sonnet",
* contextSchema,
* middleware: [middleware],
* });
*
* await agent.invoke({ messages }, { context: { region: "EU" } });
* ```
*
* @public
*/
declare function dynamicSystemPromptMiddleware<TContextSchema = unknown>(fn: DynamicSystemPromptMiddlewareConfig<TContextSchema>): AgentMiddleware<undefined, undefined, unknown, readonly (_langchain_core_tools3.ServerTool | _langchain_core_tools3.ClientTool)[]>;
//#endregion
export { DynamicSystemPromptMiddlewareConfig, dynamicSystemPromptMiddleware };
//# sourceMappingURL=dynamicSystemPrompt.d.ts.map