git.iko.soy/crush
on main help sign in
chore: small fixes
commit99ef38cca63dd064f8b0368467c4ba3980edf548 authorKujtim Hoxha <kujtimii.h@gmail.com> dateSat 5 Jul 2025, 16:41 (1 year ago) parentc8d9a2b
2 files changed+11−63
compare with main
unifiedsplit
internal/config/load.go+11−7 internal/llm/prompt/prompt_test.go−56
internal/config/load.go+11−7
@@ -317,9 +317,10 @@ func (cfg *Config) defaultModelSelection(knownProviders []provider.Provider) (la
317317 return
318318 }
319319 largeModel = SelectedModel{
320− Provider: string(p.ID),
321− Model: defaultLargeModel.ID,
322− MaxTokens: defaultLargeModel.DefaultMaxTokens,
320+ Provider: string(p.ID),
321+ Model: defaultLargeModel.ID,
322+ MaxTokens: defaultLargeModel.DefaultMaxTokens,
323+ ReasoningEffort: defaultLargeModel.DefaultReasoningEffort,
323324 }
324325
325326 defaultSmallModel := cfg.GetModel(string(p.ID), p.DefaultSmallModelID)
@@ -327,9 +328,10 @@ func (cfg *Config) defaultModelSelection(knownProviders []provider.Provider) (la
327328 err = fmt.Errorf("default small model %s not found for provider %s", p.DefaultSmallModelID, p.ID)
328329 }
329330 smallModel = SelectedModel{
330− Provider: string(p.ID),
331− Model: defaultSmallModel.ID,
332− MaxTokens: defaultSmallModel.DefaultMaxTokens,
331+ Provider: string(p.ID),
332+ Model: defaultSmallModel.ID,
333+ MaxTokens: defaultSmallModel.DefaultMaxTokens,
334+ ReasoningEffort: defaultSmallModel.DefaultReasoningEffort,
333335 }
334336 return
335337 }
@@ -387,7 +389,9 @@ func (cfg *Config) configureSelectedModels(knownProviders []provider.Provider) e
387389 } else {
388390 large.MaxTokens = model.DefaultMaxTokens
389391 }
390− large.ReasoningEffort = largeModelSelected.ReasoningEffort
392+ if largeModelSelected.ReasoningEffort != "" {
393+ large.ReasoningEffort = largeModelSelected.ReasoningEffort
394+ }
391395 large.Think = largeModelSelected.Think
392396 }
393397 smallModelSelected, smallModelConfigured := cfg.Models[SelectedModelTypeSmall]
internal/llm/prompt/prompt_test.godeleted−56
@@ -1,56 +0,0 @@
1−package prompt
2−
3−import (
4− "fmt"
5− "os"
6− "path/filepath"
7− "testing"
8−
9− "github.com/charmbracelet/crush/internal/config"
10− "github.com/stretchr/testify/assert"
11− "github.com/stretchr/testify/require"
12−)
13−
14−func TestGetContextFromPaths(t *testing.T) {
15− t.Parallel()
16−
17− tmpDir := t.TempDir()
18− _, err := config.Init(tmpDir, false)
19− if err != nil {
20− t.Fatalf("Failed to load config: %v", err)
21− }
22− testFiles := []string{
23− "file.txt",
24− "directory/file_a.txt",
25− "directory/file_b.txt",
26− "directory/file_c.txt",
27− }
28−
29− createTestFiles(t, tmpDir, testFiles)
30−
31− context := getContextFromPaths(
32− []string{
33− "file.txt",
34− "directory/",
35− },
36− )
37− expectedContext := fmt.Sprintf("# From:%s/file.txt\nfile.txt: test content\n# From:%s/directory/file_a.txt\ndirectory/file_a.txt: test content\n# From:%s/directory/file_b.txt\ndirectory/file_b.txt: test content\n# From:%s/directory/file_c.txt\ndirectory/file_c.txt: test content", tmpDir, tmpDir, tmpDir, tmpDir)
38− assert.Equal(t, expectedContext, context)
39−}
40−
41−func createTestFiles(t *testing.T, tmpDir string, testFiles []string) {
42− t.Helper()
43− for _, path := range testFiles {
44− fullPath := filepath.Join(tmpDir, path)
45− if path[len(path)-1] == '/' {
46− err := os.MkdirAll(fullPath, 0o755)
47− require.NoError(t, err)
48− } else {
49− dir := filepath.Dir(fullPath)
50− err := os.MkdirAll(dir, 0o755)
51− require.NoError(t, err)
52− err = os.WriteFile(fullPath, []byte(path+": test content"), 0o644)
53− require.NoError(t, err)
54− }
55− }
56−}