package ollama

import (
	"log/slog"
	"net/http"

	"git.warky.dev/wdevs/amcs/internal/ai/compat"
	"git.warky.dev/wdevs/amcs/internal/config"
)

func New(cfg config.AIConfig, httpClient *http.Client, log *slog.Logger) (*compat.Client, error) {
	return compat.New(compat.Config{
		Name:                  "ollama",
		BaseURL:               cfg.Ollama.BaseURL,
		APIKey:                cfg.Ollama.APIKey,
		EmbeddingModel:        cfg.Embeddings.Model,
		MetadataModel:         cfg.Metadata.Model,
		FallbackMetadataModel: cfg.Metadata.FallbackModel,
		Temperature:           cfg.Metadata.Temperature,
		Headers:               cfg.Ollama.RequestHeaders,
		HTTPClient:            httpClient,
		Log:                   log,
		Dimensions:            cfg.Embeddings.Dimensions,
	}), nil
}