import { z } from "zod";

export const SearchKGSchema = z.object({
  query: z.string().describe("The search query in third person perspective"),
  validAt: z.string().optional().describe("The valid at time in ISO format"),
  startTime: z.string().optional().describe("The start time in ISO format"),
  endTime: z.string().optional().describe("The end time in ISO format"),
});

export const IngestKGSchema = z.object({
  message: z.string().describe("The data to ingest in text format"),
});

export type SearchKG = z.infer<typeof SearchKGSchema>;
export type IngestKG = z.infer<typeof IngestKGSchema>;
