fix: fix typo

instill-ai · Jul 8, 2024 · 4f0a13a · 4f0a13a
1 parent 5b0a01b
commit 4f0a13a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/ai/cohere/v0/component_test.go b/ai/cohere/v0/component_test.go
@@ -63,7 +63,7 @@ func TestComponent_Tasks(t *testing.T) {
 		wantResp textGenerationOutput
 	}{
 		input:    map[string]any{"model-name": "command-r-plus"},
-		wantResp: textGenerationOutput{Text: "Hi! My name is command-r-plus.", Ciatations: []ciatation{}, Usage: commandUsage{InputTokens: 20, OutputTokens: 30}},
+		wantResp: textGenerationOutput{Text: "Hi! My name is command-r-plus.", Citations: []citation{}, Usage: commandUsage{InputTokens: 20, OutputTokens: 30}},
 	}
 
 	c.Run("ok - task command", func(c *qt.C) {

diff --git a/ai/cohere/v0/text_generation.go b/ai/cohere/v0/text_generation.go
@@ -35,7 +35,7 @@ type textGenerationInput struct {
 	Documents    []string      `json:"documents"`
 }
 
-type ciatation struct {
+type citation struct {
 	Start int    `json:"start"`
 	End   int    `json:"end"`
 	Text  string `json:"text"`
@@ -47,9 +47,9 @@ type commandUsage struct {
 }
 
 type textGenerationOutput struct {
-	Text       string       `json:"text"`
-	Ciatations []ciatation  `json:"citations"`
-	Usage      commandUsage `json:"usage"`
+	Text      string       `json:"text"`
+	Citations []citation   `json:"citations"`
+	Usage     commandUsage `json:"usage"`
 }
 
 func (e *execution) taskTextGeneration(in *structpb.Struct) (*structpb.Struct, error) {
@@ -109,10 +109,10 @@ func (e *execution) taskTextGeneration(in *structpb.Struct) (*structpb.Struct, e
 		return nil, err
 	}
 
-	citations := []ciatation{}
+	citations := []citation{}
 
 	for _, c := range resp.Citations {
-		citation := ciatation{
+		citation := citation{
 			Start: c.Start,
 			End:   c.End,
 			Text:  c.Text,
@@ -124,8 +124,8 @@ func (e *execution) taskTextGeneration(in *structpb.Struct) (*structpb.Struct, e
 	outputTokens := *bills.OutputTokens
 
 	outputStruct := textGenerationOutput{
-		Text:       resp.Text,
-		Ciatations: citations,
+		Text:      resp.Text,
+		Citations: citations,
 		Usage: commandUsage{
 			InputTokens:  int(inputTokens),
 			OutputTokens: int(outputTokens),