contextSchema

const agent = createAgent({
  llm: model,
  tools: [getWeather],
  contextSchema: z.object({
    capital: z.string(),
  }),
  prompt: (state, config) => {
    return [
      new SystemMessage(`You are a helpful assistant. The capital of France is ${config.context.capital}.`),
    ];
  },
});

const result = await agent.invoke({
  messages: [
    new SystemMessage("You are a helpful assistant."),
    new HumanMessage("What is the capital of France?"),
  ],
}, {
  context: {
    capital: "Paris",
  },
});