package agent

import (
	"context"
	"fmt"
	"log"

	"github.com/tmc/langchaingo/llms"
)

func Call(llm llms.Model, ctx context.Context) (response string) {

	response, err := llms.GenerateFromSinglePrompt(
		ctx,
		llm,
		"Hello! My name is wonder!",
	)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println("AI:", response)

	return response

}
func CallWithMessage(llm llms.Model, ctx context.Context, prompt string) (response string) {

	response, err := llms.GenerateFromSinglePrompt(
		ctx,
		llm,
		prompt,
	)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println("AI:", response)

	return response

}