import { Agent, unstable_callable as callable } from 'agents';
import { searchMenusByAgent, chooseWinners } from '../utils';

export class LunchAgent extends Agent<Env, LunchState> {
  onStart() {
    this.schedule('weekdays at 11:30pm', 'chooseLunch');
    this.schedule('daily at 5pm', 'resetLunch');
  }

  @callable()
  async nominateRestaurant(restaurantName: string) {
    // Uses a Browser Search tool to find restaurant info
    // Finds menu and stores it into Vectorize
    // On success updates Agent state with available restaurants
    await this.env.RESTAURANT_RESEARCHER_WORKFLOW.create({
      restaurantName,
      agent: this.name,
      near: this.state.officeAddress,
    });
  }

  @callable()
  async searchRestaurants(query: string) {
    // Uses Vector store results filtered by Metadata limited
    // To this agent
    const results = await searchMenusByAgent(query, this.name);
    return results.map((result) => result.metadata.restaurantName);
  }

  @callable()
  async vote(username: string, restaurantName: string) {
    const votes = this.state.todaysVotes;
    votes.push({
      username,
      restaurantName,
    });
    // Send update to all connected eaters
    this.setState({
      ...this.state,
      todaysVotes: votes,
    });
  }

  async resetLunch() {
    const state = this.state;
    state.todaysVotes = [];
    state.todaysRuling = undefined;
    this.setState(state);
  }

  async chooseLunch() {
    const restaurantWinners = chooseWinners(this.state.todaysVotes);
    const { response } = await this.env.AI.run("@cf/meta/llama-3.3-70b-instruct-fp8-fast", {
      messages: [
        {role: "system", content: `
          You help deliver results to a bunch of co-workers who are choosing lunch together.
          The user is going to provide you with the options.
          Your task is to make the choice sound exciting so people who voted for something
                    else feel validated.
        `},
        {role: "user", content: restaurantWinners?.join(", ") as string}
      ],
    });
    this.setState({
      ...this.state,
      todaysRuling: response
    })
  }
}

export type Restaurant = {
  cuisine: string;
  name: string;
  address: string;
};

export type Vote = {
  username: string;
  restaurantName: string;
};

export type LunchState = {
  officeAddress: string;
  todaysVotes: Vote[];
  todaysRuling?: string;
  restaurants: Restaurant[];
};