prompt

pkg/main.go

@@ -16,7 +16,7 @@ func main() {
 		{Role: "user", Content: "こんにちは"},
 	}
 
-	response, err := llm.SendMessage(msg)
+	response, err := llm.Invoke(msg)
 	if err != nil {
 		fmt.Println(err)
 	}

pkg/openai/openai.go

@@ -29,7 +29,7 @@ func NewChatOpenAI(model string) (*ChatOpenAI, error) {
 	}, nil
 }
 
-func (c *ChatOpenAI) SendMessage(message []Message) (*Response, error) {
+func (c *ChatOpenAI) Invoke(message []Message) (*Response, error) {
 
 	requestBody := Request{
 		Model:    c.model,

pkg/openai/openai_test.go

@@ -8,7 +8,7 @@ import (
 	"github.com/jarcoal/httpmock"
 )
 
-func TestSendMessage(t *testing.T) {
+func TestInvoke(t *testing.T) {
 	tests := []struct {
 		name     string
 		message  []Message
@@ -50,7 +50,7 @@ func TestSendMessage(t *testing.T) {
 	}
 	for _, tc := range tests {
 		t.Run(tc.name, func(t *testing.T) {
-			have, err := client.SendMessage(tc.message)
+			have, err := client.Invoke(tc.message)
 			if err != nil {
 				t.Fatalf("Error happens: %v", err)
 			}

pkg/prompt/prompt.go

@@ -0,0 +1,30 @@
+package prompt
+
+import (
+	"bytes"
+	"fmt"
+	"text/template"
+)
+
+type PromptTemplate struct {
+	template *template.Template
+}
+
+func NewPromptTemplate(input string) (*PromptTemplate, error) {
+	tmpl, err := template.New("tmpl").Parse(input)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse template: %w", err)
+	}
+
+	return &PromptTemplate{
+		template: tmpl,
+	}, nil
+}
+
+func (t *PromptTemplate) Invoke(input any) (string, error) {
+	var buf bytes.Buffer
+	if err := t.template.Execute(&buf, input); err != nil {
+		return "", err
+	}
+	return buf.String(), nil
+}

pkg/prompt/prompt_test.go

@@ -0,0 +1,43 @@
+package prompt
+
+import (
+	"testing"
+)
+
+func TestInvoke(t *testing.T) {
+	tests := []struct {
+		name     string
+		template string
+		input    any
+		expected string
+	}{
+		{
+			name:     "template is string",
+			template: "template",
+			expected: "template",
+		},
+		{
+			name:     "template includes input",
+			template: "the meaning of {{.Word}}?",
+			input: map[string]string{
+				"Word": "satisfaction",
+			},
+			expected: "the meaning of satisfaction?",
+		},
+	}
+	for _, tc := range tests {
+		t.Run(tc.name, func(t *testing.T) {
+			prompt, err := NewPromptTemplate(tc.template)
+			if err != nil {
+				t.Fatalf("Error happens: %v", err)
+			}
+			have, err := prompt.Invoke(tc.input)
+			if err != nil {
+				t.Fatalf("Error happens: %v", err)
+			}
+			if have != tc.expected {
+				t.Fatalf("unexpected string: %v != %v", have, tc.expected)
+			}
+		})
+	}
+}