schema

type BettaFishState struct {
	// User input
	Query string `json:"query"`

	// QueryEngine State
	ReportTitle string       `json:"report_title"`
	Paragraphs  []*Paragraph `json:"paragraphs"`
	NewsResults []string     `json:"news_results"` // The compiled news report(s)
	FinalReport string       `json:"final_report"` // The final combined report

	// MediaEngine State
	MediaResults []string `json:"media_results"`

	// InsightEngine State
	InsightResults []string `json:"insight_results"`

	// ForumEngine State
	Discussion []string `json:"discussion"`
}

type Paragraph struct {
	Title    string         `json:"title"`
	Content  string         `json:"content"` // Expected content description
	Research *ResearchState `json:"research"`
}

type ResearchState struct {
	SearchQueries []string                  `json:"search_queries"`
	SearchResults map[string][]SearchResult `json:"search_results"` // Query -> Results
	LatestSummary string                    `json:"latest_summary"`
	Completed     bool                      `json:"completed"`
	// contains filtered or unexported fields
}

type SearchResult struct {
	Title         string  `json:"title"`
	URL           string  `json:"url"`
	Content       string  `json:"content"`
	Score         float64 `json:"score"`
	RawContent    string  `json:"raw_content,omitempty"`
	PublishedDate string  `json:"published_date,omitempty"`
}