package reader

import (
	"context"
	"fmt"
	"net/url"

	"gitea.stevedudenhoeffer.com/steve/answer/pkg/agents/shared"

	"gitea.stevedudenhoeffer.com/steve/answer/pkg/cache"

	gollm "gitea.stevedudenhoeffer.com/steve/go-llm"
)

type Agent struct {
	// Model is the chat completion model to use
	Model gollm.ChatCompletion

	// OnNewFunction is called when a new function is created
	OnNewFunction func(ctx context.Context, funcName string, question string, parameter string) (any, error)

	// OnFunctionFinished is called when a function is finished
	OnFunctionFinished func(ctx context.Context, funcName string, question string, parameter string, result string, err error, newFunctionResult any) error

	Cache cache.Cache

	ContextualInformation []string
}

// Read will try to read the source and return the answer if possible.
func (a Agent) Read(ctx context.Context, question string, source *url.URL) (shared.Knowledge, error) {
	if a.Cache == nil {
		a.Cache = cache.Nop{}
	}

	ar, err := extractArticle(ctx, a.Cache, source)
	if err != nil {
		return shared.Knowledge{}, err
	}

	if ar.Body == "" {
		return shared.Knowledge{}, fmt.Errorf("could not extract body from page")
	}

	return doesTextAnswerQuestion(ctx, question, ar.Body, source.String(), a)
}