puppeteer-browser-agent/src/agent/driver/handlers/content-handler.ts

Puppeteer browser agent for use in AI agent systems

import { BaseResponseHandler, type IResponseHandler } from "./base-handler";

export interface IContentResponseHandler extends IResponseHandler {}

export class ContentResponseHandler extends BaseResponseHandler {
	actionId?: string;

	public async handle(step: any) {
		super.handle(step);
		if (this.isFunctionCallStep()) {
			return;
		}
		this.printCurrentCost();
		let nextContent = this.step.content.trim();

		if (nextContent === "") {
			nextContent = "<empty response>";
		}
		await this.createMessageForController(nextContent);
	}

	createMessageForController(content: string) {
		this.driver.createMessageForController(content);
	}

	printCurrentCost() {
		this.driver.printCurrentCost();
	}
}