diff --git a/agent.go b/ai/agent.go
similarity index 100%
rename from agent.go
rename to ai/agent.go
diff --git a/agent_stream_test.go b/ai/agent_stream_test.go
similarity index 100%
rename from agent_stream_test.go
rename to ai/agent_stream_test.go
diff --git a/agent_test.go b/ai/agent_test.go
similarity index 100%
rename from agent_test.go
rename to ai/agent_test.go
diff --git a/content.go b/ai/content.go
similarity index 100%
rename from content.go
rename to ai/content.go
diff --git a/errors.go b/ai/errors.go
similarity index 100%
rename from errors.go
rename to ai/errors.go
diff --git a/model.go b/ai/model.go
similarity index 100%
rename from model.go
rename to ai/model.go
diff --git a/provider.go b/ai/provider.go
similarity index 100%
rename from provider.go
rename to ai/provider.go
diff --git a/retry.go b/ai/retry.go
similarity index 100%
rename from retry.go
rename to ai/retry.go
diff --git a/tool.go b/ai/tool.go
similarity index 100%
rename from tool.go
rename to ai/tool.go
diff --git a/tool_test.go b/ai/tool_test.go
similarity index 100%
rename from tool_test.go
rename to ai/tool_test.go
diff --git a/util.go b/ai/util.go
similarity index 100%
rename from util.go
rename to ai/util.go
diff --git a/examples/agent/main.go b/examples/agent/main.go
index 97364783f632a5f8623abd3e550142aa70c3f8fe..c91ba5868a90e719464eb33c723d11466bfaa045 100644
--- a/examples/agent/main.go
+++ b/examples/agent/main.go
@@ -5,7 +5,7 @@ import (
 	"fmt"
 	"os"
 
-	"github.com/charmbracelet/ai"
+	"github.com/charmbracelet/ai/ai"
 	"github.com/charmbracelet/ai/providers/openai"
 )
 
diff --git a/examples/simple/main.go b/examples/simple/main.go
index 5aea35047fda21a9266c6f17b613b9d75ba90ccc..b3f950e1c910fa50229b0ff61d7f7ca921df49d4 100644
--- a/examples/simple/main.go
+++ b/examples/simple/main.go
@@ -5,7 +5,7 @@ import (
 	"fmt"
 	"os"
 
-	"github.com/charmbracelet/ai"
+	"github.com/charmbracelet/ai/ai"
 	"github.com/charmbracelet/ai/providers/anthropic"
 )
 
diff --git a/examples/stream/main.go b/examples/stream/main.go
index 1eb3c9ee18f31f03229a34ed194cd703c899f2ef..f141594d59ee57c28a773b063eb7c0e7578ad25f 100644
--- a/examples/stream/main.go
+++ b/examples/stream/main.go
@@ -6,7 +6,7 @@ import (
 	"fmt"
 	"os"
 
-	"github.com/charmbracelet/ai"
+	"github.com/charmbracelet/ai/ai"
 	"github.com/charmbracelet/ai/providers/openai"
 )
 
diff --git a/examples/streaming-agent-simple/main.go b/examples/streaming-agent-simple/main.go
index 5f94d0a30cdfd518de1afbb05b79448ab34e2842..46dbb041e2f83b3e67a195e23c68bc92be51ab69 100644
--- a/examples/streaming-agent-simple/main.go
+++ b/examples/streaming-agent-simple/main.go
@@ -5,7 +5,7 @@ import (
 	"fmt"
 	"os"
 
-	"github.com/charmbracelet/ai"
+	"github.com/charmbracelet/ai/ai"
 	"github.com/charmbracelet/ai/providers/openai"
 )
 
diff --git a/examples/streaming-agent/main.go b/examples/streaming-agent/main.go
index fb4fa400599e3814bd8eccf117467b0bebcaeb32..f009753e49d4d817162eb5df8818c41e70c95c15 100644
--- a/examples/streaming-agent/main.go
+++ b/examples/streaming-agent/main.go
@@ -6,7 +6,7 @@ import (
 	"os"
 	"strings"
 
-	"github.com/charmbracelet/ai"
+	"github.com/charmbracelet/ai/ai"
 	"github.com/charmbracelet/ai/providers/anthropic"
 )
 
diff --git a/providers/anthropic/anthropic.go b/providers/anthropic/anthropic.go
index b2d089da17de66b35048e474162d488075fdcd63..88a322e9724250fe3c43ed142043bb81e5226cd6 100644
--- a/providers/anthropic/anthropic.go
+++ b/providers/anthropic/anthropic.go
@@ -14,7 +14,7 @@ import (
 	"github.com/anthropics/anthropic-sdk-go"
 	"github.com/anthropics/anthropic-sdk-go/option"
 	"github.com/anthropics/anthropic-sdk-go/packages/param"
-	"github.com/charmbracelet/ai"
+	"github.com/charmbracelet/ai/ai"
 )
 
 type options struct {
diff --git a/providers/openai/openai.go b/providers/openai/openai.go
index 4627defdb65bbd902ab74893300f9453c452230e..c62a8da3c539a1f49740fa8ae38e8968312819d5 100644
--- a/providers/openai/openai.go
+++ b/providers/openai/openai.go
@@ -11,7 +11,7 @@ import (
 	"maps"
 	"strings"
 
-	"github.com/charmbracelet/ai"
+	"github.com/charmbracelet/ai/ai"
 	"github.com/google/uuid"
 	"github.com/openai/openai-go/v2"
 	"github.com/openai/openai-go/v2/option"
diff --git a/providers/openai/openai_test.go b/providers/openai/openai_test.go
index a0c9ae9e9af178f336166337439eeea974a99caf..5c05141d45624eb7c75216e0f7c6b4afa518dfea 100644
--- a/providers/openai/openai_test.go
+++ b/providers/openai/openai_test.go
@@ -10,7 +10,7 @@ import (
 	"strings"
 	"testing"
 
-	"github.com/charmbracelet/ai"
+	"github.com/charmbracelet/ai/ai"
 	"github.com/openai/openai-go/v2/packages/param"
 	"github.com/stretchr/testify/require"
 )