Migrate from CrewAI to Guild

"use agent";
import { agent, pick, userInterfaceTools, type Task } from "@guildai/agents-sdk";
import { z } from "zod";

const inputSchema = z.object({ topic: z.string().max(200) });
const outputSchema = z.object({ title: z.string(), summary: z.string(), risks: z.array(z.string()).max(3) });
const tools = { ...pick(userInterfaceTools, ["ui_prompt"]) };

async function run(input: z.infer<typeof inputSchema>, task: Task<typeof tools>) {
  const research = await task.llm.generateText({
    system: "You are a senior data researcher.",
    prompt: `Research the latest developments in ${input.topic}. Return ten bullet points.`,
  });
  const brief = await task.llm.generateText({
    system: "You are an executive writer. Write briefs that are short and concrete.",
    prompt: `Turn this research into a brief with a title, a summary, and three risks:\n\n${research.text}`,
    schema: outputSchema,
  });
  const answer = await task.tools.ui_prompt({
    type: "text",
    text: `Draft brief:\n\n${brief.title}\n\n${brief.summary}\n\nReply "ok" to accept.`,
  });
  if (!/^\s*ok\b/i.test(answer.text)) throw new Error("Brief not accepted");
  return brief;
}

export default agent({ description: "Researches a topic and writes an executive brief, with a human review.", inputSchema, outputSchema, tools, run });

research_task:
  description: Research the latest developments in {topic}.
  expected_output: Ten bullet points about {topic}.
  agent: researcher
brief_task:
  description: Turn the research into a one-page brief for executives.
  expected_output: A brief with a title, a summary, and three risks.
  agent: writer
  human_input: true

class Brief(BaseModel):
    title: str
    summary: str
    risks: list[str]

@CrewBase
class BriefCrew():
    @task
    def brief_task(self) -> Task:
        return Task(config=self.tasks_config["brief_task"], output_pydantic=Brief)

    @crew
    def crew(self) -> Crew:
        return Crew(agents=self.agents, tasks=self.tasks, process=Process.sequential)