langchain/dist/agents/annotation.cjs

Typescript bindings for langchain

const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
const __langchain_langgraph = require_rolldown_runtime.__toESM(require("@langchain/langgraph"));
const __langchain_langgraph_zod = require_rolldown_runtime.__toESM(require("@langchain/langgraph/zod"));
const __langchain_core_utils_types = require_rolldown_runtime.__toESM(require("@langchain/core/utils/types"));

//#region src/agents/annotation.ts
function createAgentState(hasStructuredResponse = true, stateSchema, middlewareList = []) {
	/**
	* Collect fields from state schemas
	*/
	const stateFields = { jumpTo: new __langchain_langgraph.UntrackedValue() };
	const inputFields = {};
	const outputFields = {};
	const applySchema = (schema) => {
		if (__langchain_langgraph.StateSchema.isInstance(schema)) {
			for (const [key, field] of Object.entries(schema.fields)) {
				if (key.startsWith("_")) continue;
				if (!(key in stateFields)) {
					stateFields[key] = field;
					if (__langchain_langgraph.ReducedValue.isInstance(field)) {
						inputFields[key] = field.inputSchema || field.valueSchema;
						outputFields[key] = field.valueSchema;
					} else {
						inputFields[key] = field;
						outputFields[key] = field;
					}
				}
			}
			return;
		}
		const shape = (0, __langchain_core_utils_types.getInteropZodObjectShape)(schema);
		for (const [key, fieldSchema] of Object.entries(shape)) {
			if (key.startsWith("_")) continue;
			if (!(key in stateFields)) {
				if ((0, __langchain_core_utils_types.isZodSchemaV4)(fieldSchema)) {
					const meta = __langchain_langgraph_zod.schemaMetaRegistry.get(fieldSchema);
					if (meta?.reducer) {
						if (meta.reducer.schema) {
							stateFields[key] = new __langchain_langgraph.ReducedValue(fieldSchema, {
								inputSchema: meta.reducer.schema,
								reducer: meta.reducer.fn
							});
							inputFields[key] = meta.reducer.schema;
							outputFields[key] = fieldSchema;
						} else {
							stateFields[key] = new __langchain_langgraph.ReducedValue(fieldSchema, { reducer: meta.reducer.fn });
							inputFields[key] = fieldSchema;
							outputFields[key] = fieldSchema;
						}
						continue;
					}
				}
				stateFields[key] = fieldSchema;
				inputFields[key] = fieldSchema;
				outputFields[key] = fieldSchema;
			}
		}
	};
	/**
	* Add state schema properties from user-provided schema.
	* Supports both StateSchema and Zod v3/v4 objects.
	*/
	if (stateSchema && (__langchain_langgraph.StateSchema.isInstance(stateSchema) || (0, __langchain_core_utils_types.isInteropZodObject)(stateSchema))) applySchema(stateSchema);
	/**
	* Add state schema properties from middleware.
	* Supports both StateSchema and Zod v3/v4 objects.
	*/
	for (const middleware of middlewareList) if (middleware.stateSchema && (__langchain_langgraph.StateSchema.isInstance(middleware.stateSchema) || (0, __langchain_core_utils_types.isInteropZodObject)(middleware.stateSchema))) applySchema(middleware.stateSchema);
	if (hasStructuredResponse) outputFields.structuredResponse = new __langchain_langgraph.UntrackedValue();
	/**
	* Create StateSchema instances for state, input, and output.
	* Using MessagesValue provides the proper message reducer behavior.
	*/
	return {
		state: new __langchain_langgraph.StateSchema({
			messages: __langchain_langgraph.MessagesValue,
			...stateFields
		}),
		input: new __langchain_langgraph.StateSchema({
			messages: __langchain_langgraph.MessagesValue,
			...inputFields
		}),
		output: new __langchain_langgraph.StateSchema({
			messages: __langchain_langgraph.MessagesValue,
			...outputFields
		})
	};
}

//#endregion
exports.createAgentState = createAgentState;
//# sourceMappingURL=annotation.cjs.map