diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index a13c940..5439fc2 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -68,7 +68,7 @@ jobs:
         with:
           fetch-depth: 1
       - uses: ./.github/actions/setup-python-uv
-      - run: uv build
+      - run: uv build --all-packages
       - run: |
           uv pip install twine
           uv run twine check dist/*
diff --git a/packages/text-generation/pyproject.toml b/packages/text-generation/pyproject.toml
index 5b2a665..61f4b55 100644
--- a/packages/text-generation/pyproject.toml
+++ b/packages/text-generation/pyproject.toml
@@ -1,5 +1,5 @@
 [project]
-name = "text-generation"
+name = "celeste-text-generation"
 version = "0.1.0"
 description = "Type-safe text generation for Celeste AI. Unified interface for OpenAI, Anthropic, Google, Mistral, Cohere, and more"
 authors = [{name = "Kamilbenkirane", email = "kamil@withceleste.ai"}]
diff --git a/pyproject.toml b/pyproject.toml
index c2e5ae3..46bfa70 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -32,6 +32,10 @@ Documentation = "https://withceleste.ai/docs"
 Repository = "https://github.com/withceleste/celeste-python"
 Issues = "https://github.com/withceleste/celeste-python/issues"
 
+[project.optional-dependencies]
+text-generation = ["celeste-text-generation>=0.1.0"]
+all = ["celeste-text-generation>=0.1.0"]
+
 [dependency-groups]
 dev = [
     "pytest>=8.0",
@@ -49,6 +53,9 @@ dev = [
 [tool.uv.workspace]
 members = ["packages/*"]
 
+[tool.uv.sources]
+celeste-text-generation = { workspace = true }
+
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"