composer

type ComposedMeta struct {
	Tickers  []string `json:"tickers"`
	Markets  []string `json:"markets"`
	Hashtags []string `json:"hashtags"`
}

type ComposedNews struct {
	ID       string   `json:"id"`
	Text     string   `json:"text"`
	Tickers  []string `json:"tickers"`  // tickers mentioned or/and related to the news
	Markets  []string `json:"markets"`  // US/EU/Asia stocks, bonds, commodities, housing, etc.
	Hashtags []string `json:"hashtags"` // hashtags related to the news (#inflation, #fed, #buybacks, etc.)
}

type Composer struct {
	OpenAiClient       openAiClientInterface
	TogetherAIClient   togetherAIClientInterface
	GoogleGeminiClient GoogleGeminiClientInterface
	Config             *promptConfig
}

type Error struct {
	// contains filtered or unexported fields
}

type GoogleGemini struct {
	APIKey string
}

type GoogleGeminiClientInterface interface {
	CreateChatCompletion(ctx context.Context, req GoogleGeminiRequest) (response *genai.GenerateContentResponse, err error)
}

type GoogleGeminiRequest struct {
	Prompt      string  `json:"prompt"`
	MaxTokens   int32   `json:"max_tokens"`
	Temperature float32 `json:"temperature"`
	TopP        float32 `json:"top_p"`
	TopK        int32   `json:"top_k"`
}

type Headline struct {
	ID   string `json:"id"`
	Text string `json:"text"`
	Link string `json:"link"`
}

type SummarisedHeadline struct {
	ID      string `json:"id"`      // ID of the news or event
	Verb    string `json:"verb"`    // Main verb of the news or event to be marked in summary
	Summary string `json:"summary"` // Summary of the news or event
	Link    string `json:"link"`    // Link to the publication to use in string Markdown
}

type TogetherAI struct {
	APIKey string
	URL    string
}

type TogetherAIResponse struct {
	ID      string `json:"id"`
	Choices []struct {
		Text string `json:"text"`
	} `json:"choices"`
	Usage struct {
		PromptTokens     int `json:"prompt_tokens"`
		CompletionTokens int `json:"completion_tokens"`
		TotalTokens      int `json:"total_tokens"`
	}
	Created int64  `json:"created"`
	Model   string `json:"model"`
	Object  string `json:"object"`
}