Handle two-factor authentication

import os

from hai_agents import Client
from hai_agents_tools import OtpRequest, imap_otp_handler, otp_tool

inbox = imap_otp_handler(
    host="imap.gmail.com",
    username="agent-inbox@gmail.com",
    password=os.environ["GMAIL_APP_PASSWORD"],
    sender="no-reply@example.com",  # optional: only mail from this address
)

def from_slack(request: OtpRequest) -> str:
    return slack.ask_user(request.prompt)

client = Client()
result = client.run_session(
    agent="h/web-surfer-flash",
    messages="Log in to example.com and check for new notifications.",
    tools=[otp_tool(inbox)],  # or otp_tool() to prompt, otp_tool(from_slack) for your own handler
)
print(result.answer)

import { HaiAgentsClient, imapOtpHandler, otpTool, type OtpRequest } from "hai-agents";

// The IMAP handler needs: npm install imapflow mailparser
const inbox = imapOtpHandler({
  host: "imap.gmail.com",
  username: "agent-inbox@gmail.com",
  password: process.env.GMAIL_APP_PASSWORD!,
  sender: "no-reply@example.com", // optional: only mail from this address
});

async function fromSlack(request: OtpRequest): Promise<string> {
  return slack.askUser(request.prompt);
}

const client = new HaiAgentsClient();
const result = await client.runSession({
  agent: "h/web-surfer-flash",
  messages: "Log in to example.com and check for new notifications.",
  tools: [otpTool({ handler: inbox })], // or otpTool() to prompt, otpTool({ handler: fromSlack })
});
console.log(result.answer);