import { type LanguageModel, stepCountIs, ToolLoopAgent } from "ai";

import type { Logger } from "../../logger";

import { createGetCurrentTime } from "../tools/get-current-time";

const SYSTEM_PROMPT = `你是 Alfred，一个 AI 助手。

## 输出规范
- 使用中文回复
- 代码块用 Markdown 围栏语法，标注语言
- 给出结论时简洁直接，不要长篇铺垫
- 不确定的事明确说"不确定"`;

export function createAlfredAgent(model: LanguageModel, logger?: Logger) {
  return new ToolLoopAgent({
    instructions: SYSTEM_PROMPT,
    model,
    stopWhen: stepCountIs(20),
    tools: { getCurrentTime: createGetCurrentTime(logger) },
  });
}