celeste-ai 0.2.4__tar.gz → 0.3.3__tar.gz

{celeste_ai-0.2.4 → celeste_ai-0.3.3}/.env.example

@@ -7,6 +7,9 @@ OPENAI_API_KEY=your-openai-api-key-here
 # Anthropic
 ANTHROPIC_API_KEY=your-anthropic-api-key-here
 
+# BFL (Black Forest Labs)
+BFL_API_KEY=your-bfl-api-key-here
+
 # Google
 GOOGLE_API_KEY=your-google-api-key-here
 
@@ -36,3 +39,6 @@ TOPAZLABS_API_KEY=your-topazlabs-api-key-here
 
 # Perplexity
 PERPLEXITY_API_KEY=your-perplexity-api-key-here
+
+# Gradium
+GRADIUM_API_KEY=your-gradium-api-key-here

{celeste_ai-0.2.4 → celeste_ai-0.3.3}/.github/workflows/ci.yml

@@ -64,7 +64,7 @@ jobs:
           python-version: ${{ inputs.python-version || '3.12' }}
       - run: |
           if [ -d "packages" ]; then
-            uv run mypy -p celeste && uv run mypy tests/ && uv run mypy packages/
+            uv run mypy -p celeste && uv run mypy tests/ && uv run mypy packages/capabilities/image-generation packages/capabilities/text-generation packages/capabilities/video-generation packages/capabilities/speech-generation
           else
             uv run mypy -p celeste && uv run mypy tests/
           fi
@@ -105,7 +105,7 @@ jobs:
       - uses: ./.github/actions/setup-python-uv
         with:
           python-version: ${{ matrix.python-version }}
-      - run: uv run pytest tests/unit_tests -v --cov=celeste --cov-report=term-missing --cov-report=xml --cov-report=html --cov-fail-under=90
+      - run: uv run pytest tests/unit_tests -v --cov=celeste --cov-report=term-missing --cov-report=xml --cov-report=html --cov-fail-under=80
       - uses: codecov/codecov-action@v4
         if: matrix.os == 'ubuntu-latest' && matrix.python-version == '3.12'
         with:

{celeste_ai-0.2.4 → celeste_ai-0.3.3}/.github/workflows/publish.yml

@@ -41,28 +41,124 @@ jobs:
     uses: ./.github/workflows/ci.yml
     secrets: inherit
 
-  integration-test:
-    needs: [validate-release, run-ci]
+  detect-changes:
+    needs: run-ci
     runs-on: ubuntu-latest
+    outputs:
+      packages: ${{ steps.build-matrix.outputs.packages }}
+      has-changes: ${{ steps.build-matrix.outputs.has-changes }}
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+      - id: get-base
+        run: |
+          # Get previous tag to compare against (tag pushes don't have github.event.before)
+          PREVIOUS_TAG=$(git describe --tags --abbrev=0 HEAD^ 2>/dev/null || echo "")
+          if [ -z "$PREVIOUS_TAG" ]; then
+            echo "base=" >> $GITHUB_OUTPUT
+            echo "run-all=true" >> $GITHUB_OUTPUT
+          else
+            echo "base=$PREVIOUS_TAG" >> $GITHUB_OUTPUT
+            echo "run-all=false" >> $GITHUB_OUTPUT
+          fi
+      - uses: dorny/paths-filter@v3
+        id: filter
+        if: steps.get-base.outputs.run-all != 'true'
+        with:
+          base: ${{ steps.get-base.outputs.base }}
+          filters: |
+            text-generation:
+              - 'packages/capabilities/text-generation/**'
+              - '!**/pyproject.toml'
+            image-generation:
+              - 'packages/capabilities/image-generation/**'
+              - '!**/pyproject.toml'
+            video-generation:
+              - 'packages/capabilities/video-generation/**'
+              - '!**/pyproject.toml'
+            speech-generation:
+              - 'packages/capabilities/speech-generation/**'
+              - '!**/pyproject.toml'
+            core:
+              - 'src/celeste/**'
+              - '!**/pyproject.toml'
+      - id: build-matrix
+        run: |
+          PACKAGES=()
+
+          # If first release or paths-filter was skipped, run all
+          if [[ "${{ steps.get-base.outputs.run-all }}" == "true" ]]; then
+            PACKAGES=("text-generation" "image-generation" "video-generation" "speech-generation")
+          else
+            CORE="${{ steps.filter.outputs.core }}"
+            if [[ "$CORE" == "true" || "${{ steps.filter.outputs.text-generation }}" == "true" ]]; then
+              PACKAGES+=("text-generation")
+            fi
+            if [[ "$CORE" == "true" || "${{ steps.filter.outputs.image-generation }}" == "true" ]]; then
+              PACKAGES+=("image-generation")
+            fi
+            if [[ "$CORE" == "true" || "${{ steps.filter.outputs.video-generation }}" == "true" ]]; then
+              PACKAGES+=("video-generation")
+            fi
+            if [[ "$CORE" == "true" || "${{ steps.filter.outputs.speech-generation }}" == "true" ]]; then
+              PACKAGES+=("speech-generation")
+            fi
+          fi
+
+          # Convert to JSON array
+          JSON=$(printf '%s\n' "${PACKAGES[@]}" | jq -R . | jq -s -c .)
+          echo "packages=$JSON" >> $GITHUB_OUTPUT
+
+          if [[ ${#PACKAGES[@]} -gt 0 ]]; then
+            echo "has-changes=true" >> $GITHUB_OUTPUT
+          else
+            echo "has-changes=false" >> $GITHUB_OUTPUT
+          fi
+
+  integration-tests:
+    needs: [validate-release, run-ci, detect-changes]
+    if: needs.detect-changes.outputs.has-changes == 'true'
+    runs-on: ubuntu-latest
+    timeout-minutes: 30
     environment:
       name: integration-tests
+    permissions:
+      contents: read
+      id-token: write
+    strategy:
+      fail-fast: false
+      matrix:
+        package: ${{ fromJSON(needs.detect-changes.outputs.packages) }}
     steps:
       - uses: actions/checkout@v4
+      - id: auth
+        uses: google-github-actions/auth@v2
         with:
-          fetch-depth: 1
+          workload_identity_provider: ${{ secrets.GCP_WORKLOAD_IDENTITY_PROVIDER }}
+          service_account: ${{ secrets.GCP_SERVICE_ACCOUNT }}
       - uses: ./.github/actions/setup-python-uv
-      - name: Run integration tests
+      - name: Run ${{ matrix.package }} integration tests
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
           ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
+          BFL_API_KEY: ${{ secrets.BFL_API_KEY }}
           GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
           MISTRAL_API_KEY: ${{ secrets.MISTRAL_API_KEY }}
           COHERE_API_KEY: ${{ secrets.COHERE_API_KEY }}
-          BYTEDANCE_API_KEY: ${{ secrets.BYTEDANCE_API_KEY }}
-        run: make integration-test
+          BYTEPLUS_API_KEY: ${{ secrets.BYTEPLUS_API_KEY }}
+          XAI_API_KEY: ${{ secrets.XAI_API_KEY }}
+          ELEVENLABS_API_KEY: ${{ secrets.ELEVENLABS_API_KEY }}
+          GRADIUM_API_KEY: ${{ secrets.GRADIUM_API_KEY }}
+        run: uv run pytest packages/capabilities/${{ matrix.package }}/tests/integration_tests/ -m integration -v
 
   build:
-    needs: [validate-release, run-ci, integration-test]
+    needs: [validate-release, run-ci, detect-changes, integration-tests]
+    if: |
+      always() &&
+      needs.validate-release.result == 'success' &&
+      needs.run-ci.result == 'success' &&
+      (needs.integration-tests.result == 'success' || needs.integration-tests.result == 'skipped')
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4

{celeste_ai-0.2.4 → celeste_ai-0.3.3}/.gitignore

@@ -157,3 +157,4 @@ uv.lock
 
 # Security reports
 bandit-report.json
+mureka.md

celeste_ai-0.3.3/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Kamil Benkirane
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

{celeste_ai-0.2.4 → celeste_ai-0.3.3}/Makefile

@@ -8,7 +8,8 @@ help:
 	@echo "  make format     - Apply Ruff formatting"
 	@echo "  make typecheck  - Run mypy type checking"
 	@echo "  make test       - Run all tests (core + packages) with coverage"
-	@echo "  make integration-test - Run integration tests (requires API keys)"
+	@echo "  make integration-test [capability]  - Run integration tests (all or specific)"
+	@echo "                                        (e.g., make integration-test image-intelligence)"
 	@echo "  make security   - Run Bandit security scan"
 	@echo "  make ci       - Run full CI/CD pipeline"
 	@echo "  make clean      - Clean cache directories"
@@ -33,15 +34,22 @@ format:
 
 # Type checking (fail fast on any error)
 typecheck:
-	@uv run mypy -p celeste && uv run mypy tests/ && uv run mypy packages/
+	@uv run mypy -p celeste && uv run mypy tests/ && uv run mypy packages/*/*/src/
 
 # Testing
 test:
-	uv run pytest tests/unit_tests packages/*/tests/unit_tests --cov=celeste --cov-report=term-missing --cov-fail-under=80 -v
+	uv run pytest tests/unit_tests --cov=celeste --cov-report=term-missing --cov-fail-under=80 -v
 
 # Integration testing (requires API keys)
+# Usage: make integration-test [capability]
 integration-test:
-	uv run pytest packages/*/tests/integration_tests/ -m integration -v --dist=worksteal -n auto
+	@cap="$(filter-out $@,$(MAKECMDGOALS))"; \
+	if [ -z "$$cap" ]; then cap="*"; fi; \
+	uv run pytest packages/capabilities/$$cap/tests/integration_tests/ -m integration -v --dist=worksteal -n auto
+
+# Catch capability names as no-op targets
+%:
+	@:
 
 # Security scanning (config reads from pyproject.toml)
 security:

{celeste_ai-0.2.4 → celeste_ai-0.3.3}/PKG-INFO

@@ -1,18 +1,18 @@
 Metadata-Version: 2.4
 Name: celeste-ai
-Version: 0.2.4
+Version: 0.3.3
 Summary: Open source, type-safe primitives for multi-modal AI. All capabilities, all providers, one interface
 Project-URL: Homepage, https://withceleste.ai
 Project-URL: Documentation, https://withceleste.ai/docs
 Project-URL: Repository, https://github.com/withceleste/celeste-python
 Project-URL: Issues, https://github.com/withceleste/celeste-python/issues
 Author-email: Kamilbenkirane <kamil@withceleste.ai>
-License: Apache-2.0
+License: MIT
 License-File: LICENSE
-Keywords: ai,anthropic,claude,embeddings,gemini,image-generation,multimodal,openai,sdk,text-generation,video-generation
+Keywords: ai,anthropic,claude,embeddings,gemini,image-generation,multimodal,openai,sdk,speech-generation,text-generation,video-generation
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
-Classifier: License :: OSI Approved :: Apache Software License
+Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
@@ -25,19 +25,20 @@ Requires-Dist: httpx>=0.27.0
 Requires-Dist: pydantic-settings>=2.0
 Requires-Dist: pydantic>=2.0
 Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: websockets>=13.0
 Provides-Extra: all
-Requires-Dist: celeste-image-generation>=0.2.1; extra == 'all'
-Requires-Dist: celeste-image-intelligence>=0.2.1; extra == 'all'
-Requires-Dist: celeste-speech-generation>=0.2.3; extra == 'all'
-Requires-Dist: celeste-text-generation>=0.2.1; extra == 'all'
+Requires-Dist: celeste-image-generation>=0.3.3; extra == 'all'
+Requires-Dist: celeste-speech-generation>=0.3.3; extra == 'all'
+Requires-Dist: celeste-text-generation>=0.3.3; extra == 'all'
+Requires-Dist: celeste-video-generation>=0.3.3; extra == 'all'
 Provides-Extra: image-generation
-Requires-Dist: celeste-image-generation>=0.2.1; extra == 'image-generation'
-Provides-Extra: image-intelligence
-Requires-Dist: celeste-image-intelligence>=0.2.1; extra == 'image-intelligence'
+Requires-Dist: celeste-image-generation>=0.3.3; extra == 'image-generation'
 Provides-Extra: speech-generation
-Requires-Dist: celeste-speech-generation>=0.2.3; extra == 'speech-generation'
+Requires-Dist: celeste-speech-generation>=0.3.3; extra == 'speech-generation'
 Provides-Extra: text-generation
-Requires-Dist: celeste-text-generation>=0.2.1; extra == 'text-generation'
+Requires-Dist: celeste-text-generation>=0.3.3; extra == 'text-generation'
+Provides-Extra: video-generation
+Requires-Dist: celeste-video-generation>=0.3.3; extra == 'video-generation'
 Description-Content-Type: text/markdown
 
 <div align="center">
@@ -53,61 +54,100 @@ All capabilities. All providers. One interface.
 Primitives, not frameworks.
 
 [![Python](https://img.shields.io/badge/Python-3.12+-blue?style=for-the-badge)](https://www.python.org/)
-[![License](https://img.shields.io/badge/License-Apache_2.0-red?style=for-the-badge)](LICENSE)
+[![License](https://img.shields.io/badge/License-MIT-yellow?style=for-the-badge)](LICENSE)
 [![PyPI](https://img.shields.io/badge/PyPI-celeste--ai-green?style=for-the-badge)](https://pypi.org/project/celeste-ai/)
 
+<a href="https://github.com/withceleste/celeste-python" target="_parent">
+  <img alt="" src="https://img.shields.io/github/stars/withceleste/celeste-python.svg?style=social&label=Star" alt="GitHub stars" />
+</a>
+<a href="https://pypi.org/project/celeste-ai" target="_parent">
+  <img alt="" src="https://img.shields.io/pypi/dm/celeste-ai.svg" />
+</a>
+<a href="https://www.linkedin.com/company/withceleste" target="_parent">
+  <img src="https://img.shields.io/badge/LinkedIn-Follow%20%40withceleste-0077B5?style=social&logo=linkedin&logoColor=white" alt="Follow @withceleste on LinkedIn"/>
+</a>
+
 [Quick Start](#-quick-start) • [Request Provider](https://github.com/withceleste/celeste-python/issues/new)
 
 </div>
 
----
+# Celeste AI
 
-## 🚀 Quick Start
 
+Type-safe, capability-provider-agnostic primitives .
+
+- **Unified Interface:** One API for OpenAI, Anthropic, Gemini, Mistral, and 14+ others.
+- **True Multi-Modal:** Text, Image, Audio, Video, Embeddings, Search —all first-class citizens.
+- **Type-Safe by Design:** Full Pydantic validation and IDE autocomplete.
+- **Zero Lock-In:** Switch providers instantly by changing a single config string.
+- **Primitives, Not Frameworks:** No agents, no chains, no magic. Just clean I/O.
+- **Lightweight Architecture:** No vendor SDKs. Pure, fast HTTP.
+
+## 🚀 Quick Start
 ```python
-from celeste import create_client, Capability, Provider
+from celeste import create_client
+
 
-# Create client
+# "We need a catchy slogan for our new eco-friendly sneaker."
 client = create_client(
-    capability=Capability.TEXT_GENERATION,
-    provider=Provider.ANTHROPIC,
-    api_key="your-api-key",  # Or loads automatically from environment
+    capability="text-generation",
+    model="gpt-5"
 )
-
-# Generate
-response = await client.generate(prompt="Explain quantum computing")
-print(response.content)
-```
-
-**Install:**
-```bash
-uv add "celeste-ai[text-generation]"  # Text only
-uv add "celeste-ai[image-generation]" # Image generation
-uv add "celeste-ai[all]"              # Everything
+slogan = await client.generate("Write a slogan for an eco-friendly sneaker.")
+print(slogan.content)
 ```
 
----
-
-## 🎨 Multi-Modal Example
 
+## 🎨 Multimodal example
 ```python
-# Same API, different modalities
-text_client = create_client(Capability.TEXT_GENERATION, Provider.ANTHROPIC)
-image_client = create_client(Capability.IMAGE_GENERATION, Provider.OPENAI)
-video_client = create_client(Capability.VIDEO_GENERATION, Provider.GOOGLE)
-
-text = await text_client.generate(prompt="Write a haiku about AI")
-image = await image_client.generate(prompt="A sunset over mountains")
-video = await video_client.generate(prompt="Waves crashing on a beach")
+from pydantic import BaseModel, Field
+
+class ProductCampaign(BaseModel):
+    visual_prompt: str
+    audio_script: str
+
+# 2. Extract Campaign Assets (Anthropic)
+# -----------------------------------------------------
+extract_client = create_client(Capability.TEXT_GENERATION, model="claude-opus-4-1")
+campaign_output = await extract_client.generate(
+    f"Create campaign assets for slogan: {slogan.content}",
+    output_schema=ProductCampaign
+)
+campaign = campaign_output.content
+
+# 3. Generate Ad Visual (Flux)
+# -----------------------------------------------------
+image_client = create_client(Capability.IMAGE_GENERATION, model="flux-2-flex")
+image_output = await image_client.generate(
+    campaign.visual_prompt,
+    aspect_ratio="1:1"
+)
+image = image_output.content
+
+# 4. Generate Radio Spot (ElevenLabs)
+# -----------------------------------------------------
+speech_client = create_client(Capability.SPEECH_GENERATION, model="eleven_v3")
+speech_output = await speech_client.generate(
+    campaign.audio_script,
+    voice="adam"
+)
+speech = speech_output.content
 ```
 
 No special cases. No separate libraries. **One consistent interface.**
 
----
 
 
+---
+
 <div align="center">
 
+## 15+ providers. Zero lock-in.
+
+
+
+
+
 <img src="https://www.google.com/favicon.ico" width="32" height="32" alt="Google" title="Google">
 <img src="https://www.anthropic.com/favicon.ico" width="32" height="32" alt="Anthropic" title="Anthropic">
 <img src="https://www.google.com/s2/favicons?domain=openai.com&sz=64" width="32" height="32" alt="OpenAI" title="OpenAI">
@@ -132,6 +172,84 @@ No special cases. No separate libraries. **One consistent interface.**
 
 ---
 
+## 🔄 Switch providers in one line
+
+
+```python
+from pydantic import BaseModel
+
+class User(BaseModel):
+    name: str
+    age: int
+
+# Model IDs
+anthropic_model_id = "claude-4-5-sonnet"
+google_model_id = "gemini-2.5-flash"
+```
+
+```python
+# ❌ Anthropic Way
+from anthropic import Anthropic
+import json
+
+client = Anthropic()
+response = client.messages.create(
+    model=anthropic_model_id,
+    messages=[
+        {"role": "user",
+         "content": "Extract user info: John is 30"}
+    ],
+    output_format={
+        "type": "json_schema",
+        "schema": User.model_json_schema()
+    }
+)
+user_data = json.loads(response.content[0].text)
+```
+
+```python
+# ❌ Google Gemini Way
+from google import genai
+from google.genai import types
+
+client = genai.Client()
+response = await client.aio.models.generate_content(
+    model=gemini_model_id,
+    contents="Extract user info: John is 30",
+    config=types.GenerateContentConfig(
+        response_mime_type="application/json",
+        response_schema=User
+    )
+)
+user = response.parsed
+```
+
+```python
+# ✅ Celeste Way
+from celeste import create_client, Capability
+
+
+client = create_client(
+    Capability.TEXT_GENERATION,
+    model=google_model_id  # <--- Choose any model from any provider
+)
+
+response = await client.generate(
+    prompt="Extract user info: John is 30",
+    output_schema=User  # <--- Unified parameter working across all providers
+)
+user = response.content  # Already parsed as User instance
+```
+
+---
+## 🪶 Install what you need
+```bash
+uv add "celeste-ai[text-generation]"  # Text only
+uv add "celeste-ai[image-generation]" # Image generation
+uv add "celeste-ai[all]"              # Everything
+```
+---
+
 ## 🔧 Type-Safe by Design
 
 ```python
@@ -163,7 +281,7 @@ We welcome contributions! See [CONTRIBUTING.md](CONTRIBUTING.md).
 
 ## 📄 License
 
-Apache 2.0 license – see [LICENSE](LICENSE) for details.
+MIT license – see [LICENSE](LICENSE) for details.
 
 ---