CPE: Process comment on PR #12

spachava753 · Jan 25, 2025 · d5387a7 · d5387a7
1 parent f0aaccd
commit d5387a7
Show file tree

Hide file tree

Showing 4 changed files with 82 additions and 0 deletions.
diff --git a/internal/agent/anthropic.go b/internal/agent/anthropic.go
@@ -96,6 +96,14 @@ func NewAnthropicExecutor(baseUrl string, apiKey string, logger Logger, ignorer
 					Properties: a.F[any](getRelatedFilesTool.InputSchema["properties"]),
 				}),
 			},
+			&a.BetaToolParam{
+				Name:        a.String(changeDirectoryTool.Name),
+				Description: a.String(changeDirectoryTool.Description),
+				InputSchema: a.F(a.BetaToolInputSchemaParam{
+					Type:       a.F(a.BetaToolInputSchemaTypeObject),
+					Properties: a.F[any](changeDirectoryTool.InputSchema["properties"]),
+				}),
+			},
 		}),
 	}
 
@@ -304,6 +312,19 @@ func (s *anthropicExecutor) Execute(input string) error {
 					}
 					s.logger.Printf("getting related files: %s", strings.Join(relatedFilesToolInput.InputFiles, ", "))
 					result, err = executeGetRelatedFilesTool(relatedFilesToolInput.InputFiles, s.ignorer)
+				case changeDirectoryTool.Name:
+					changeDirToolInput := struct {
+						Path string `json:"path"`
+					}{}
+					jsonInput, marshalErr := json.Marshal(block.Input)
+					if marshalErr != nil {
+						return fmt.Errorf("failed to marshal change directory tool input: %w", marshalErr)
+					}
+					if err := json.Unmarshal(jsonInput, &changeDirToolInput); err != nil {
+						return fmt.Errorf("failed to unmarshal change directory tool arguments: %w", err)
+					}
+					s.logger.Printf("changing directory to: %s", changeDirToolInput.Path)
+					result, err = executeChangeDirectoryTool(changeDirToolInput.Path)
 				default:
 					return fmt.Errorf("unexpected tool use block type: %s", block.Name)
 				}

diff --git a/internal/agent/deepseek.go b/internal/agent/deepseek.go
@@ -76,6 +76,14 @@ func NewDeepSeekExecutor(baseUrl string, apiKey string, logger Logger, ignorer *
 					Parameters:  oai.F(oai.FunctionParameters(getRelatedFilesTool.InputSchema)),
 				}),
 			},
+			{
+				Type: oai.F(oai.ChatCompletionToolTypeFunction),
+				Function: oai.F(oai.FunctionDefinitionParam{
+					Name:        oai.F(changeDirectoryTool.Name),
+					Description: oai.F(changeDirectoryTool.Description),
+					Parameters:  oai.F(oai.FunctionParameters(changeDirectoryTool.InputSchema)),
+				}),
+			},
 		}),
 	}
 
@@ -178,6 +186,15 @@ func (o *deepseekExecutor) Execute(input string) error {
 				}
 				o.logger.Printf("getting related files: %s", strings.Join(relatedFilesToolInput.InputFiles, ", "))
 				result, err = executeGetRelatedFilesTool(relatedFilesToolInput.InputFiles, o.ignorer)
+			case changeDirectoryTool.Name:
+				var changeDirToolInput struct {
+					Path string `json:"path"`
+				}
+				if err := json.Unmarshal([]byte(toolCall.Function.Arguments), &changeDirToolInput); err != nil {
+					return fmt.Errorf("failed to unmarshal change directory tool arguments: %w", err)
+				}
+				o.logger.Printf("changing directory to: %s", changeDirToolInput.Path)
+				result, err = executeChangeDirectoryTool(changeDirToolInput.Path)
 			default:
 				return fmt.Errorf("unexpected tool name: %s", toolCall.Function.Name)
 			}

diff --git a/internal/agent/gemini.go b/internal/agent/gemini.go
@@ -130,6 +130,20 @@ func NewGeminiExecutor(baseUrl string, apiKey string, logger Logger, ignorer *gi
 						Required: []string{"input_files"},
 					},
 				},
+				{
+					Name:        changeDirectoryTool.Name,
+					Description: changeDirectoryTool.Description,
+					Parameters: &genai.Schema{
+						Type: genai.TypeObject,
+						Properties: map[string]*genai.Schema{
+							"path": {
+								Type:        genai.TypeString,
+								Description: "The path to change to, can be relative or absolute",
+							},
+						},
+						Required: []string{"path"},
+					},
+				},
 			},
 		},
 	}
@@ -256,6 +270,19 @@ func (g *geminiExecutor) Execute(input string) error {
 					}
 					g.logger.Printf("getting related files: %s", strings.Join(relatedFilesToolInput.InputFiles, ", "))
 					result, err = executeGetRelatedFilesTool(relatedFilesToolInput.InputFiles, g.ignorer)
+				case changeDirectoryTool.Name:
+					var changeDirToolInput struct {
+						Path string `json:"path"`
+					}
+					jsonInput, marshalErr := json.Marshal(v.Args)
+					if marshalErr != nil {
+						return fmt.Errorf("failed to marshal change directory tool input: %w", marshalErr)
+					}
+					if err := json.Unmarshal(jsonInput, &changeDirToolInput); err != nil {
+						return fmt.Errorf("failed to unmarshal change directory tool arguments: %w", err)
+					}
+					g.logger.Printf("changing directory to: %s", changeDirToolInput.Path)
+					result, err = executeChangeDirectoryTool(changeDirToolInput.Path)
 				default:
 					return fmt.Errorf("unexpected tool name: %s", v.Name)
 				}

diff --git a/internal/agent/openai.go b/internal/agent/openai.go
@@ -95,6 +95,14 @@ func NewOpenAIExecutor(baseUrl string, apiKey string, logger Logger, ignorer *gi
 					Parameters:  oai.F(oai.FunctionParameters(getRelatedFilesTool.InputSchema)),
 				}),
 			},
+			{
+				Type: oai.F(oai.ChatCompletionToolTypeFunction),
+				Function: oai.F(oai.FunctionDefinitionParam{
+					Name:        oai.F(changeDirectoryTool.Name),
+					Description: oai.F(changeDirectoryTool.Description),
+					Parameters:  oai.F(oai.FunctionParameters(changeDirectoryTool.InputSchema)),
+				}),
+			},
 		}),
 	}
 
@@ -191,6 +199,15 @@ func (o *openaiExecutor) Execute(input string) error {
 				}
 				o.logger.Printf("getting related files: %s", strings.Join(relatedFilesToolInput.InputFiles, ", "))
 				result, err = executeGetRelatedFilesTool(relatedFilesToolInput.InputFiles, o.ignorer)
+			case changeDirectoryTool.Name:
+				var changeDirToolInput struct {
+					Path string `json:"path"`
+				}
+				if err := json.Unmarshal([]byte(toolCall.Function.Arguments), &changeDirToolInput); err != nil {
+					return fmt.Errorf("failed to unmarshal change directory tool arguments: %w", err)
+				}
+				o.logger.Printf("changing directory to: %s", changeDirToolInput.Path)
+				result, err = executeChangeDirectoryTool(changeDirToolInput.Path)
 			default:
 				return fmt.Errorf("unexpected tool name: %s", toolCall.Function.Name)
 			}