feat(anthropic): tool handling and improve summary prefix logic (#211)

- Add `.codegpt.yaml` to `.gitignore`
- Import `encoding/json` in `anthropic.go`
- Replace `MessageContentToolResult` with `MessageContentToolUse` in `GetSummaryPrefix` function
- Add JSON unmarshalling for `toolUse.Input` in `GetSummaryPrefix` function
- Change content assignment from `toolResult.Content[0].GetText()` to `result.Prefix` in `GetSummaryPrefix` function
- Define a new `tool` struct with a `Prefix` field in `func.go`
- Rename tool definition key from `unit` to `prefix` in `tools` slice

Signed-off-by: Bo-Yi Wu <[email protected]>

Author: appleboy

.gitignore

@@ -18,3 +18,4 @@ dist
 **/.DS_Store
 bin
 release
+.codegpt.yaml

provider/anthropic/anthropic.go

@@ -2,6 +2,7 @@ package anthropic
 
 import (
 	"context"
+	"encoding/json"
 	"errors"
 	"fmt"
 
@@ -68,20 +69,25 @@ func (c *Client) GetSummaryPrefix(ctx context.Context, content string) (*core.Re
 		return nil, err
 	}
 
-	var toolResult *anthropic.MessageContentToolResult
+	var toolUse *anthropic.MessageContentToolUse
 
 	for _, c := range resp.Content {
-		if c.Type == anthropic.MessagesContentTypeToolResult {
-			toolResult = c.MessageContentToolResult
+		if c.Type == anthropic.MessagesContentTypeToolUse {
+			toolUse = c.MessageContentToolUse
 		}
 	}
 
-	if toolResult == nil {
+	if toolUse == nil {
 		return nil, errors.New("no tool use found in response")
 	}
 
+	var result tool
+	if err := json.Unmarshal(toolUse.Input, &result); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal tool use input: %w", err)
+	}
+
 	return &core.Response{
-		Content: toolResult.Content[0].GetText(),
+		Content: result.Prefix,
 		Usage: core.Usage{
 			PromptTokens:     resp.Usage.InputTokens,
 			CompletionTokens: resp.Usage.OutputTokens,

provider/anthropic/func.go

@@ -5,14 +5,30 @@ import (
 	"github.com/sashabaranov/go-openai/jsonschema"
 )
 
+// tool represents a structure with a single field Prefix.
+// Prefix is a string that is serialized to JSON with the key "prefix".
+type tool struct {
+	Prefix string `json:"prefix"`
+}
+
+// tools is a slice of ToolDefinition that contains the definition for various tools.
+// Each ToolDefinition includes the following fields:
+// - Name: The name of the tool.
+// - Description: A brief description of what the tool does.
+// - InputSchema: The schema for the input that the tool expects, defined using jsonschema.Definition.
+//   - Type: The type of the input, which is an object.
+//   - Properties: A map of property names to their definitions. In this case, it includes:
+//   - "prefix": A string that must be one of the specified values ("build", "chore", "ci", "docs", "feat", "fix", "perf", "refactor", "style", "test").
+//     This property also has a description indicating that it is the prefix to use for the summary.
+//   - Required: A list of required properties, which includes "prefix".
 var tools = []anthropic.ToolDefinition{
 	{
 		Name:        "get_summary_prefix",
 		Description: "Get a summary prefix using function call",
 		InputSchema: jsonschema.Definition{
 			Type: jsonschema.Object,
 			Properties: map[string]jsonschema.Definition{
-				"unit": {
+				"prefix": {
 					Type: jsonschema.String,
 					Enum: []string{
 						"build", "chore", "ci",