diff --git a/internal/agent/agent_test.go b/internal/agent/agent_test.go index 0682083ae20daaa1c17747afdfd6b4db358b88fa..527552e28292320a3486cfabd68ce63003322aec 100644 --- a/internal/agent/agent_test.go +++ b/internal/agent/agent_test.go @@ -33,7 +33,7 @@ func getModels(t *testing.T, r *recorder.Recorder, pair modelPair) (fantasy.Lang return large, small } -func setupAgent(t *testing.T, pair modelPair) (SessionAgent, env) { +func setupAgent(t *testing.T, pair modelPair) (SessionAgent, fakeEnv) { r := newRecorder(t) large, small := getModels(t, r, pair) env := testEnv(t) diff --git a/internal/agent/common_test.go b/internal/agent/common_test.go index f6f564109a32c278f6e127809b9b2ef550c239bd..b668b5dba6d42f24a4ca2eea788f7c8a608452f8 100644 --- a/internal/agent/common_test.go +++ b/internal/agent/common_test.go @@ -30,7 +30,8 @@ import ( _ "github.com/joho/godotenv/autoload" ) -type env struct { +// fakeEnv is an environment for testing. +type fakeEnv struct { workingDir string sessions session.Service messages message.Service @@ -100,7 +101,7 @@ func zAIBuilder(model string) builderFunc { } } -func testEnv(t *testing.T) env { +func testEnv(t *testing.T) fakeEnv { workingDir := filepath.Join("/tmp/crush-test/", t.Name()) os.RemoveAll(workingDir) @@ -123,7 +124,7 @@ func testEnv(t *testing.T) env { os.RemoveAll(workingDir) }) - return env{ + return fakeEnv{ workingDir, sessions, messages, @@ -133,7 +134,7 @@ func testEnv(t *testing.T) env { } } -func testSessionAgent(env env, large, small fantasy.LanguageModel, systemPrompt string, tools ...fantasy.AgentTool) SessionAgent { +func testSessionAgent(env fakeEnv, large, small fantasy.LanguageModel, systemPrompt string, tools ...fantasy.AgentTool) SessionAgent { largeModel := Model{ Model: large, CatwalkCfg: catwalk.Model{ @@ -152,7 +153,7 @@ func testSessionAgent(env env, large, small fantasy.LanguageModel, systemPrompt return agent } -func coderAgent(r *recorder.Recorder, env env, large, small fantasy.LanguageModel) (SessionAgent, error) { +func coderAgent(r *recorder.Recorder, env fakeEnv, large, small fantasy.LanguageModel) (SessionAgent, error) { fixedTime := func() time.Time { t, _ := time.Parse("1/2/2006", "1/1/2025") return t