@@ -12,6 +12,7 @@ import (
1212 "github.com/charmbracelet/crush/internal/session"
1313 "github.com/charmbracelet/fantasy/ai"
1414 "github.com/charmbracelet/fantasy/anthropic"
15+ "github.com/stretchr/testify/assert"
1516 "github.com/stretchr/testify/require"
1617 "gopkg.in/dnaeon/go-vcr.v4/pkg/recorder"
1718
@@ -23,12 +24,12 @@ type builderFunc func(r *recorder.Recorder) (ai.LanguageModel, error)
2324func TestSessionSimpleAgent(t *testing.T) {
2425 r := newRecorder(t)
2526 sonnet, err := anthropicBuilder("claude-sonnet-4-5-20250929")(r)
26− require.Nil(t, err)
27+ require.NoError(t, err)
2728 haiku, err := anthropicBuilder("claude-3-5-haiku-20241022")(r)
28− require.Nil(t, err)
29+ require.NoError(t, err)
2930 agent, sessions, messages := testSessionAgent(t, sonnet, haiku, "You are a helpful assistant")
3031 session, err := sessions.Create(t.Context(), "New Session")
31− require.Nil(t, err)
32+ require.NoError(t, err)
3233
3334 res, err := agent.Run(t.Context(), SessionAgentCall{
3435 Prompt: "Hello",
@@ -36,14 +37,14 @@ func TestSessionSimpleAgent(t *testing.T) {
3637 MaxOutputTokens: 10000,
3738 })
3839
39− require.Nil(t, err)
40− require.NotNil(t, res)
40+ require.NoError(t, err)
41+ assert.NotNil(t, res)
4142
4243 t.Run("should create session messages", func(t *testing.T) {
4344 msgs, err := messages.List(t.Context(), session.ID)
44− require.Nil(t, err)
45+ require.NoError(t, err)
4546 // Should have the agent and user message
46− require.Equal(t, len(msgs), 2)
47+ assert.Equal(t, len(msgs), 2)
4748 })
4849}
4950