auto-tweet.ts

import path from "node:path";

import type { ModelMessage } from "@hyv/core";
import { Agent, sequence } from "@hyv/core";
import { createInstruction, GPTModelAdapter } from "@hyv/openai";
import type { ImageMessage } from "@hyv/stable-diffusion";
import { Automatic1111ModelAdapter } from "@hyv/stable-diffusion";
import { minify, createFileWriter } from "@hyv/utils";

const dir = path.join(process.cwd(), `examples/output/auto-tweet/${Date.now()}`);
const fileWriter = createFileWriter(dir);
const imageWriter = createFileWriter(dir, "base64");

const termAgent = new Agent(
	new GPTModelAdapter({
		model: "gpt-4",
		maxTokens: 1024,
		temperature: 0.8,
		systemInstruction: createInstruction(
			"mastermind, random term creator, very funny, hilarious",
			minify`
			Think deeply.
			Reason your thoughts.
			Reflect on your reasons.
			Provide several ideas based on your reasons.
			Decide on ONE of your your ideas.
			Provide two absolutely random, uncommon and unrelated terms and pair them.
			NEVER use terms from the {{example}}!
			`,
			{
				thought: "very detailed elaborative string",
				reasoning: "very detailed elaborative string",
				reflection: "very detailed elaborative string",
				ideas: ["{{comparison}}"],
				decision: "very thoughtful reflective and detailed elaborative string",
				instructions: {
					term1: "one word",
					term2: "one word",
					comparator: "is * than",
					comparison: "{{term1}} {{comparator}} {{term2}}",
				},
			}
		),
	}),
	{
		verbosity: 1,
	}
);

const tweeter = new Agent(
	new GPTModelAdapter({
		model: "gpt-4",
		maxTokens: 1024,
		systemInstruction: createInstruction(
			"Comedic Writer, Twitter trend expert",
			minify`\
				Follow instructions closely!
				Think deeply.
				Reason your thoughts.
				Reflect on your reasons.
				Make a decision based on your reflection.
				Write a tweet with hashtags and emojis.
				Provide meta-data for images, based on your tweet.
				**Acceptance Criteria**:
				1. Write a UNIQUE hilarious tweet WITH characters:length(~{{characterCount}}) AND hashtags:length(~{{hashtagCount}}), emojis:length(~{{emojiCount}}), images:length(={{imageCount}})!
				2. Tweet compares 2 terms and adds a valid reason!!
				3. Add a prompt(+{{illustrationStyle}}),  negativePrompt for each image!
				4. Add an alt text for each image!
				`,
			{
				thought: "very detailed elaborative string",
				reasoning: "very detailed elaborative string",
				reflection: "very detailed elaborative string",
				decision: "very detailed elaborative string",
				hashtags: ["string"],
				emojis: ["🤦‍"],
				tweet: "Did you know {(comparison)}, because {{term1}} … {{term2}} ?",
				images: [
					{
						path: "[filename].jpg",
						prompt: "description(~{{term1}} + ~{{term2}}) + keywords(comma separated) + {{illustrationStyle}}",
						negativePrompt: "keywords(comma separated), bad quality, blurry",
						alt: "concise string",
					},
				],
				files: [
					{
						path: "tweet.md",
						content:
							"markdown: ![image.alt](image.path)\n\n # {{tweet}} {{emojis}} {{hashtags}}",
					},
				],
			}
		),
	}),
	{
		verbosity: 1,
		sideEffects: [fileWriter],
		async before(message: ModelMessage & { instructions: Record<string, unknown> }) {
			return {
				// ...message,
				instructions: {
					...message.instructions,
					imageCount: 1,
					characterCount: 250,
					hashtagCount: "<=5",
					emojiCount: "<=3",
					style: "slapstick comedy, fun facts",
					illustrationStyle: "flat illustration",
				},
			};
		},
	}
);

const illustrator = new Agent(new Automatic1111ModelAdapter(), {
	sideEffects: [imageWriter],
	async before(message: ImageMessage): Promise<ImageMessage> {
		return {
			...message,
			images: message.images.map(image => ({
				...image,
				prompt:
					image.prompt +
					", vector art, simplified, vector illustration, absurdres, masterpiece, 4k, 8k, best quality",
				negativePrompt:
					image.negativePrompt +
					", blurry, lowres,  worst quality, bad quality, deformed",
			})),
		};
	},
});

try {
	await sequence(
		{
			ideaCount: "5-8",
			example: {
				term1: "running",
				term2: "tambourine",
				comparator: "is better than",
				comparison: "running is better than a tambourine",
			},
		},
		[termAgent, tweeter, illustrator]
	);
	console.log("Done");
} catch (error) {
	console.error("Error:", error);
}