ai_generator

type AiAttachment struct {
	MimeType string
	Data     []byte
}

type AiGenerator interface {
	Generate(prompt string, attachment *AiAttachment, histories []AiMessage) (*AiMessage, error)
	SetApiKey(apiKey string)
	SetSystemInstruction(instruction string)
	SetModel(model string)
	SetContentConfig(config *ContentConfig)
	SetHost(host string)
}

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

type AiMessage struct {
	Role       string // Contoh: "user", "model", "system"
	Content    string
	Attachment *AiAttachment
}

type ContentConfig struct {
	Temperature      *float32 `json:"temperature,omitempty"`
	TopP             *float32 `json:"topP,omitempty"`
	TopK             *float32 `json:"topK,omitempty"`
	CandidateCount   int32    `json:"candidateCount,omitempty"`
	MaxOutputTokens  int32    `json:"maxOutputTokens,omitempty"`
	StopSequences    []string `json:"stopSequences,omitempty"`
	ResponseLogprobs bool     `json:"responseLogprobs,omitempty"`
	Logprobs         *int32   `json:"logprobs,omitempty"`
	PresencePenalty  *float32 `json:"presencePenalty,omitempty"`
	FrequencyPenalty *float32 `json:"frequencyPenalty,omitempty"`
	Seed             *int32   `json:"seed,omitempty"`
	ResponseMIMEType string   `json:"responseMimeType,omitempty"`
}

type DeepSeekService struct {
	ApiKey string
	// contains filtered or unexported fields
}

type GeminiV2SService struct {
	ApiKey string
	// contains filtered or unexported fields
}

type GeneratorConfig struct {
	Ctx               *context.Context
	APIKey            string
	SystemInstruction string
	Model             string
	Host              string
}

type GeneratorFactory func(config GeneratorConfig) (AiGenerator, error)

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

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

type OpenAiService struct {
	ApiKey string
	// contains filtered or unexported fields
}

type OtherService struct {
	ApiKey string
	// contains filtered or unexported fields
}

type Request struct {
	Model    string    `json:"model"`
	Messages []Message `json:"messages"`
	Stream   bool      `json:"stream"`
	Format   string    `json:"format"`
	Options  struct {
		NumCtx int `json:"num_ctx"`
	}
}

type Response struct {
	Model              string    `json:"model"`
	CreatedAt          time.Time `json:"created_at"`
	Message            Message   `json:"message"`
	Done               bool      `json:"done"`
	TotalDuration      int64     `json:"total_duration"`
	LoadDuration       int       `json:"load_duration"`
	PromptEvalCount    int       `json:"prompt_eval_count"`
	PromptEvalDuration int       `json:"prompt_eval_duration"`
	EvalCount          int       `json:"eval_count"`
	EvalDuration       int64     `json:"eval_duration"`
}