@@ -12,6 +12,7 @@ import (
12 "github.com/charmbracelet/crush/internal/session"
12 "github.com/charmbracelet/crush/internal/session"
13 "github.com/charmbracelet/fantasy/ai"
13 "github.com/charmbracelet/fantasy/ai"
14 "github.com/charmbracelet/fantasy/anthropic"
14 "github.com/charmbracelet/fantasy/anthropic"
15+ "github.com/stretchr/testify/assert"
15 "github.com/stretchr/testify/require"
16 "github.com/stretchr/testify/require"
16 "gopkg.in/dnaeon/go-vcr.v4/pkg/recorder"
17 "gopkg.in/dnaeon/go-vcr.v4/pkg/recorder"
@@ -23,12 +24,12 @@ type builderFunc func(r *recorder.Recorder) (ai.LanguageModel, error)
23func TestSessionSimpleAgent(t *testing.T) {
24func TestSessionSimpleAgent(t *testing.T) {
24 r := newRecorder(t)
25 r := newRecorder(t)
25 sonnet, err := anthropicBuilder("claude-sonnet-4-5-20250929")(r)
26 sonnet, err := anthropicBuilder("claude-sonnet-4-5-20250929")(r)
26− require.Nil(t, err)
27+ require.NoError(t, err)
27 haiku, err := anthropicBuilder("claude-3-5-haiku-20241022")(r)
28 haiku, err := anthropicBuilder("claude-3-5-haiku-20241022")(r)
28− require.Nil(t, err)
29+ require.NoError(t, err)
29 agent, sessions, messages := testSessionAgent(t, sonnet, haiku, "You are a helpful assistant")
30 agent, sessions, messages := testSessionAgent(t, sonnet, haiku, "You are a helpful assistant")
30 session, err := sessions.Create(t.Context(), "New Session")
31 session, err := sessions.Create(t.Context(), "New Session")
31− require.Nil(t, err)
32+ require.NoError(t, err)
33 res, err := agent.Run(t.Context(), SessionAgentCall{
34 res, err := agent.Run(t.Context(), SessionAgentCall{
34 Prompt: "Hello",
35 Prompt: "Hello",
@@ -36,14 +37,14 @@ func TestSessionSimpleAgent(t *testing.T) {
36 MaxOutputTokens: 10000,
37 MaxOutputTokens: 10000,
39− require.Nil(t, err)
40+ require.NoError(t, err)
40− require.NotNil(t, res)
41+ assert.NotNil(t, res)
42 t.Run("should create session messages", func(t *testing.T) {
43 t.Run("should create session messages", func(t *testing.T) {
43 msgs, err := messages.List(t.Context(), session.ID)
44 msgs, err := messages.List(t.Context(), session.ID)
44− require.Nil(t, err)
45+ require.NoError(t, err)
45 // Should have the agent and user message
46 // Should have the agent and user message
46− require.Equal(t, len(msgs), 2)
47+ assert.Equal(t, len(msgs), 2)