types

type Channel struct {
	Key       string      `json:"key"`
	TestModel []string    `json:"test_model"`
	URL       string      `json:"url"`
	Type      ChannelType `json:"type"`
}

type ChannelType int

type ChatMessage struct {
	Role    string          `json:"role"`
	Content json.RawMessage `json:"content"`
}

type Message struct {
	Type     MessageType
	Content  string
	Headers  *RequestHeaders
	Error    error
	Request  string
	Response string
}

type MessageType int

type Node struct {
	IP           string
	Country      string
	Time         time.Time
	UserAgent    string
	RequestInfo  string
	ForwardedFor string
	RequestCount int // Track number of requests for this node
	IsNew        bool
	NodeIndex    int
	RegionName   string
	Org          string
	ServerName   string
}

type RequestHeaders struct {
	UserAgent    string
	ForwardedFor string
	Time         time.Time
	IP           string
}

type TestResult struct {
	Key      string
	Model    string
	Success  bool
	Latency  float64
	ErrorMsg string
}

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}