main.go to test openAI chat generation with langchaingo

main.go

package main

import (
	"fmt"
	"log"

	"github.com/tmc/langchaingo/chatmodels/openai"
	"github.com/tmc/langchaingo/exp/chains"
	"github.com/tmc/langchaingo/exp/prompts"
)

func main() {
	chat, err := openai.New() // FIXME: should probably not use the same API as the openAI LLM
	if err != nil {
		log.Fatal(err)
	}

	systemTemplate, err := prompts.NewPromptTemplate(
		"You are a helpful assistant that translates {input_language} to {output_language}.",
		[]string{"input_language", "output_language"},
	)
	if err != nil {
		log.Fatal(err)
	}

	userTemplate, err := prompts.NewPromptTemplate("{text}", []string{"text"})
	if err != nil {
		log.Fatal(err)
	}

	chatPrompt, err := prompts.NewChatTemplate([]prompts.Message{
		prompts.NewSystemMessage(systemTemplate),
		prompts.NewHumanMessage(userTemplate),
	}, []string{"input_language", "output_language", "text"})

	chainB := chains.NewLLMChain(chat, chatPrompt)

	resB, err := chains.Call(chainB, map[string]any{
		"input_language":  "English",
		"output_language": "French",
		"text":            "I love programming",
	})
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println(resB)
}