import { Agent, type Plugin } from "@downcity/agent";

export const messageAugmentPlugin: Plugin = {
  name: "message_augment",
  title: "Message Augment",
  description: "Adds extra fields to inbound message payloads.",
  hooks: {
    pipeline: {
      "chat.enrich_message": [
        async ({ value }) => {
          const body = value as Record<string, unknown>;
          return {
            ...body,
            source: body.source || "web",
            pageTitle: body.pageTitle || "Untitled Page",
          };
        },
      ],
    },
  },
};

const agent = new Agent({
  id: "repo-helper",
  path: "/path/to/project",
  tools: {},
  plugins: [messageAugmentPlugin],
});

const message = await agent.plugins.pipeline("chat.enrich_message", {
  text: "Please summarize this page",
});