import "dotenv/config";
import { GraphAgent } from "@reagentai/reagent/agent";
import {
  ChatCompletion,
  ChatInput,
  User,
} from "@reagentai/reagent/agent/nodes";
import { Groq } from "@reagentai/reagent/llm/integrations/models";
import { DummyModel } from "@reagentai/reagent/llm/models/dummy";

// create a new agent
const agent = new GraphAgent({
  name: "Simple AI Chat",
  description: "A simple AI chat agent.",
});

// add an input node; each agent must have an input node and user node for final output
const input = agent.addNode("input", new ChatInput());

// add a chat completion node
const chat1 = agent.addNode("chat-1", new ChatCompletion(), {
  config: {
    systemPrompt: "You are an amazing AI assistant called Jarvis",
    temperature: 0.9,
    stream: true,
  },
});

// add user node; this is a mandatory node for output
const user = agent.addNode("user", new User());

// bind chat completion node's inputs
chat1.bind({
  // TODO: replace model with an actual model
  model: new Groq({ model: "llama3-8b-8192" }),
  query: input.output.query,
});

// bind user node's inputs
user.bind({
  markdown: chat1.output.markdown,
  markdownStream: chat1.output.stream,
});

// export agent as default to run this agent with reagentai cli
export default agent;
export const nodes = [];
export const __reagentai_exports__ = true;