agentapi

type InitTavilyResearchReq struct {
	Topic            string               `json:"topic"`
	PreviousResearch string               `json:"previousResearch"`
	CitationFormat   string               `json:"citation_format"` // numbered, mla, apa, chicago
	Model            string               `json:"model"`           // mini, pro, auto
	OutputSchema     *tavily.OutputSchema `json:"output_schema"`
}

type Source struct {
	Favicon string `json:"favicon"`
	Title   string `json:"title"`
	URL     string `json:"url"`
}

type TavilyBackgroundCheckAspect struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	Example     string `json:"example"`
}

type TavilyBackgroundCheckReq struct {
	Language string                        `json:"language"`
	Entity   string                        `json:"entity"`  // entity name
	Context  string                        `json:"context"` // additional context about the entity or the research
	Asepcts  []TavilyBackgroundCheckAspect `json:"asepcts"` // aspects of the entity that should be answered
	Model    string                        `json:"model"`   // mini, pro, auto
}

type TavilyBackgroundCheckRes struct {
	Report  string   `json:"report"`
	Sources []Source `json:"sources"`
}

type TavilyDeepResearchRes struct {
	Report  string   `json:"report"`
	Sources []Source `json:"sources"`
}