rename apply template method

adrianliechti · Oct 11, 2024 · 4c7581b · 4c7581b
1 parent 7410c97
commit 4c7581b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/pkg/chain/agent/chain.go b/pkg/chain/agent/chain.go
@@ -62,7 +62,7 @@ func (c *Chain) Complete(ctx context.Context, messages []provider.Message, optio
 	}
 
 	if len(c.messages) > 0 {
-		values, err := template.ApplyMessages(c.messages, nil)
+		values, err := template.Messages(c.messages, nil)
 
 		if err != nil {
 			return nil, err

diff --git a/pkg/chain/assistant/chain.go b/pkg/chain/assistant/chain.go
@@ -64,7 +64,7 @@ func (c *Chain) Complete(ctx context.Context, messages []provider.Message, optio
 	}
 
 	if len(c.messages) > 0 {
-		values, err := template.ApplyMessages(c.messages, nil)
+		values, err := template.Messages(c.messages, nil)
 
 		if err != nil {
 			return nil, err

diff --git a/pkg/template/template_message.go b/pkg/template/template_message.go
@@ -6,7 +6,7 @@ import (
 	"github.com/adrianliechti/llama/pkg/provider"
 )
 
-func ApplyMessage(message provider.Message, data any) (provider.Message, error) {
+func Message(message provider.Message, data any) (provider.Message, error) {
 	t, err := NewTemplate(message.Content)
 
 	if err != nil {
@@ -24,11 +24,11 @@ func ApplyMessage(message provider.Message, data any) (provider.Message, error)
 	return message, nil
 }
 
-func ApplyMessages(messages []provider.Message, data any) ([]provider.Message, error) {
+func Messages(messages []provider.Message, data any) ([]provider.Message, error) {
 	result := slices.Clone(messages)
 
 	for i, m := range result {
-		message, err := ApplyMessage(m, data)
+		message, err := Message(m, data)
 
 		if err != nil {
 			return nil, err