package models

type Question struct {
	ID       string `json:"id"`
	Topic    string `json:"topic"`
	Question string `json:"question"`
}

type Answer struct {
	Q      Question
	Answer string `json:"answer"`
	Model  string `json:"model"`
	// resp time?
}

// type OpenRouterResp struct {
// 	ID       string `json:"id"`
// 	Provider string `json:"provider"`
// 	Model    string `json:"model"`
// 	Object   string `json:"object"`
// 	Created  int    `json:"created"`
// 	Choices  []struct {
// 		Logprobs           any    `json:"logprobs"`
// 		FinishReason       string `json:"finish_reason"`
// 		NativeFinishReason string `json:"native_finish_reason"`
// 		Index              int    `json:"index"`
// 		Message            struct {
// 			Role      string `json:"role"`
// 			Content   string `json:"content"`
// 			Refusal   any    `json:"refusal"`
// 			Reasoning any    `json:"reasoning"`
// 		} `json:"message"`
// 	} `json:"choices"`
// 	Usage struct {
// 		PromptTokens     int `json:"prompt_tokens"`
// 		CompletionTokens int `json:"completion_tokens"`
// 		TotalTokens      int `json:"total_tokens"`
// 	} `json:"usage"`
// }

type DSResp struct {
	ID      string `json:"id"`
	Choices []struct {
		Text         string `json:"text"`
		Index        int    `json:"index"`
		FinishReason string `json:"finish_reason"`
	} `json:"choices"`
	Created           int    `json:"created"`
	Model             string `json:"model"`
	SystemFingerprint string `json:"system_fingerprint"`
	Object            string `json:"object"`
}

type LLMResp struct {
	Index           int    `json:"index"`
	Content         string `json:"content"`
	Tokens          []any  `json:"tokens"`
	IDSlot          int    `json:"id_slot"`
	Stop            bool   `json:"stop"`
	Model           string `json:"model"`
	TokensPredicted int    `json:"tokens_predicted"`
	TokensEvaluated int    `json:"tokens_evaluated"`
	Prompt          string `json:"prompt"`
	HasNewLine      bool   `json:"has_new_line"`
	Truncated       bool   `json:"truncated"`
	StopType        string `json:"stop_type"`
	StoppingWord    string `json:"stopping_word"`
	TokensCached    int    `json:"tokens_cached"`
}