diff --git a/pkg/agent/instance_test.go b/pkg/agent/instance_test.go index 7f0406836..7c043d88f 100644 --- a/pkg/agent/instance_test.go +++ b/pkg/agent/instance_test.go @@ -275,8 +275,14 @@ func TestNewAgentInstance_ReadFileModeSelectsSchema(t *testing.T) { params := readTool.Parameters() props, _ := params["properties"].(map[string]any) - if _, ok := props["limit"]; !ok { - t.Fatalf("expected line-mode schema to expose limit, got %#v", props) + if _, ok := props["start_line"]; !ok { + t.Fatalf("expected line-mode schema to expose start_line, got %#v", props) + } + if _, ok := props["max_lines"]; !ok { + t.Fatalf("expected line-mode schema to expose max_lines, got %#v", props) + } + if _, ok := props["offset"]; ok { + t.Fatalf("did not expect line-mode schema to expose offset, got %#v", props) } if _, ok := props["length"]; ok { t.Fatalf("did not expect line-mode schema to expose length, got %#v", props) diff --git a/pkg/tools/filesystem.go b/pkg/tools/filesystem.go index 38d54919e..cb5921f4b 100644 --- a/pkg/tools/filesystem.go +++ b/pkg/tools/filesystem.go @@ -320,7 +320,7 @@ func (t *ReadFileTool) Name() string { } func (t *ReadFileLinesTool) Name() string { - return "read_file_lines" + return "read_file" } func (t *ReadFileTool) Description() string {