import {
  type Actor,
  type ContextBundle,
  type GuildContext,
  type RetrievalResult,
  type SourceMessage,
  type StylePack,
} from "@voidbot/shared";

export interface BuildContextInput {
  prompt: string;
  actor: Actor;
  guildContext: GuildContext;
  recentMessages?: SourceMessage[];
  retrieval?: RetrievalResult[];
  stylePack?: StylePack;
}

export class ContextBuilder {
  public build(input: BuildContextInput): ContextBundle {
    return {
      prompt: input.prompt,
      actor: input.actor,
      guildContext: input.guildContext,
      recentMessages: input.recentMessages ?? [],
      retrieval: input.retrieval ?? [],
      stylePack: input.stylePack,
      createdAt: new Date().toISOString(),
    };
  }
}