celeste-ai 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

celeste/__init__.py

@@ -1,15 +1,16 @@
-"""
-Celeste AI Framework
-
-A unified multi-modal AI framework for text, image, video, and audio generation.
-
-This is a placeholder package to reserve the name. The full framework is coming soon!
-"""
+"""Celeste AI Framework"""
 
 __version__ = "0.0.1"
 __author__ = "agent-kai"
 
-def create_client(*args, **kwargs):
+from celeste.core import Capability, Provider
+
+
+def create_client(
+    capability: Capability,
+    provider: Provider,
+    model: str,  # TODO: Use real Model type
+) -> None:
     """
     Placeholder for the universal client factory.
 
@@ -18,8 +19,9 @@ def create_client(*args, **kwargs):
     raise NotImplementedError(
         "Celeste AI is in development. "
         "This is a placeholder package to reserve the name. "
-        "Follow updates at: https://github.com/agent-kai/celeste-ai"
+        "Follow updates at: https://github.com/celeste-kai/celeste-ai"
     )
 
+
 # Placeholder exports
-__all__ = ["create_client"]
+__all__ = ["create_client"]

celeste/artifacts.py

@@ -0,0 +1,56 @@
+"""Unified artifact types for the Celeste AI framework."""
+
+from typing import Any, Optional, Union
+
+from pydantic import BaseModel, Field
+
+from celeste.mime_types import AudioMimeType, ImageMimeType, VideoMimeType
+
+
+class Artifact(BaseModel):
+    """Base class for all media artifacts.
+
+    Artifacts can be represented in three ways:
+    - url: Remote HTTP/HTTPS URL (may expire, e.g., DALL-E URLs last 1 hour)
+    - data: In-memory bytes (for immediate use without download)
+    - path: Local filesystem path (for local providers or saved files)
+
+    Providers typically populate only one of these fields.
+    """
+
+    url: Optional[str] = None
+    data: Optional[bytes] = None
+    path: Optional[str] = None
+    mime_type: Optional[str] = None  # Standard MIME type for the artifact
+    metadata: dict[str, Any] = Field(default_factory=dict)
+
+    @property
+    def has_content(self) -> bool:
+        """Check if artifact has any content."""
+        return bool(self.url or self.data or self.path)
+
+
+class ImageArtifact(Artifact):
+    """Image artifact from generation/edit operations."""
+
+    mime_type: Optional[Union[ImageMimeType, str]] = None
+
+
+class VideoArtifact(Artifact):
+    """Video artifact from generation operations."""
+
+    mime_type: Optional[Union[VideoMimeType, str]] = None
+
+
+class AudioArtifact(Artifact):
+    """Audio artifact from TTS/transcription operations."""
+
+    mime_type: Optional[Union[AudioMimeType, str]] = None
+
+
+__all__ = [
+    "Artifact",
+    "AudioArtifact",
+    "ImageArtifact",
+    "VideoArtifact",
+]

celeste/core.py

@@ -0,0 +1,35 @@
+"""Core enumerations for the Celeste AI Framework."""
+
+from enum import Enum, Flag, auto
+
+
+class Provider(str, Enum):
+    """Supported AI providers."""
+
+    OPENAI = "openai"
+    ANTHROPIC = "anthropic"
+    GOOGLE = "google"
+    MISTRAL = "mistral"
+    COHERE = "cohere"
+    XAI = "xai"
+    HUGGINGFACE = "huggingface"
+    REPLICATE = "replicate"
+    STABILITYAI = "stabilityai"
+    LUMA = "luma"
+    TOPAZLABS = "topazlabs"
+    OLLAMA = "ollama"
+    TRANSFORMERS = "transformers"
+    LOCAL = "local"
+
+
+class Capability(Flag):
+    """Supported AI capabilities."""
+
+    # Text
+    TEXT_GENERATION = auto()
+
+    # Image
+    IMAGE_GENERATION = auto()
+
+
+__all__ = ["Capability", "Provider"]

celeste/credentials.py

@@ -0,0 +1,79 @@
+"""Provider API credentials management for Celeste AI Framework."""
+
+from typing import Optional
+
+from pydantic import Field, SecretStr
+from pydantic_settings import BaseSettings
+
+from celeste.core import Provider
+
+# Provider to credential field mapping
+PROVIDER_CREDENTIAL_MAP = {
+    Provider.OPENAI: "openai_api_key",
+    Provider.ANTHROPIC: "anthropic_api_key",
+    Provider.GOOGLE: "google_api_key",
+    Provider.MISTRAL: "mistral_api_key",
+    Provider.HUGGINGFACE: "huggingface_token",
+    Provider.STABILITYAI: "stabilityai_api_key",
+    Provider.REPLICATE: "replicate_api_token",
+    Provider.COHERE: "cohere_api_key",
+    Provider.XAI: "xai_api_key",
+    Provider.LUMA: "luma_api_key",
+    Provider.TOPAZLABS: "topazlabs_api_key",
+    # LOCAL, OLLAMA, TRANSFORMERS have no credentials
+}
+
+
+class Credentials(BaseSettings):
+    """API credentials for all supported providers.
+
+    Credentials are loaded from environment variables or .env file.
+    All credentials are optional - only configure the providers you use.
+    Uses SecretStr for API keys to prevent accidental exposure in logs.
+    """
+
+    openai_api_key: Optional[SecretStr] = Field(None, alias="OPENAI_API_KEY")
+    anthropic_api_key: Optional[SecretStr] = Field(None, alias="ANTHROPIC_API_KEY")
+    google_api_key: Optional[SecretStr] = Field(None, alias="GOOGLE_API_KEY")
+    mistral_api_key: Optional[SecretStr] = Field(None, alias="MISTRAL_API_KEY")
+    huggingface_token: Optional[SecretStr] = Field(None, alias="HUGGINGFACE_TOKEN")
+    stabilityai_api_key: Optional[SecretStr] = Field(None, alias="STABILITYAI_API_KEY")
+    replicate_api_token: Optional[SecretStr] = Field(None, alias="REPLICATE_API_TOKEN")
+    cohere_api_key: Optional[SecretStr] = Field(None, alias="COHERE_API_KEY")
+    xai_api_key: Optional[SecretStr] = Field(None, alias="XAI_API_KEY")
+    luma_api_key: Optional[SecretStr] = Field(None, alias="LUMA_API_KEY")
+    topazlabs_api_key: Optional[SecretStr] = Field(None, alias="TOPAZLABS_API_KEY")
+
+    model_config = {
+        "env_file": ".env",
+        "env_file_encoding": "utf-8",
+        "case_sensitive": False,
+    }
+
+    def get_credentials(self, provider: Provider) -> SecretStr:
+        """Get credentials for a specific provider."""
+        if not self.has_credential(provider):
+            raise ValueError(f"Provider {provider} has no credentials configured.")
+
+        credential: SecretStr = getattr(self, PROVIDER_CREDENTIAL_MAP[provider])
+        return credential
+
+    def list_available_providers(self) -> list[Provider]:
+        """List all providers that have credentials configured."""
+        return [
+            provider
+            for provider in PROVIDER_CREDENTIAL_MAP
+            if self.has_credential(provider)
+        ]
+
+    def has_credential(self, provider: Provider) -> bool:
+        """Check if a specific provider has credentials configured."""
+        credential_field = PROVIDER_CREDENTIAL_MAP.get(provider)
+        if not credential_field:
+            raise ValueError(f"Provider {provider} has no credential mapping")
+        return getattr(self, credential_field, None) is not None
+
+
+credentials = Credentials()  # type: ignore[call-arg]
+
+__all__ = ["Credentials", "credentials"]

celeste/mime_types.py

@@ -0,0 +1,46 @@
+"""MIME type enumerations for the Celeste AI framework."""
+
+from enum import Enum
+
+
+class MimeType(str, Enum):
+    """Base class for all MIME types."""
+
+    pass
+
+
+class ImageMimeType(MimeType):
+    """Standard MIME types for images."""
+
+    PNG = "image/png"
+    JPEG = "image/jpeg"
+    # WEBP = "image/webp"
+
+
+class VideoMimeType(MimeType):
+    """Standard MIME types for videos."""
+
+    MP4 = "video/mp4"
+    # AVI = "video/x-msvideo"
+    # MOV = "video/quicktime"
+
+
+class AudioMimeType(MimeType):
+    """Standard MIME types for audio."""
+
+    MP3 = "audio/mpeg"
+    WAV = "audio/wav"
+    # OGG = "audio/ogg"
+    # WEBM = "audio/webm"
+    # AAC = "audio/aac"
+    # FLAC = "audio/flac"
+    # M4A = "audio/mp4"
+    # WMA = "audio/x-ms-wma"
+
+
+__all__ = [
+    "AudioMimeType",
+    "ImageMimeType",
+    "MimeType",
+    "VideoMimeType",
+]

{celeste_ai-0.0.1.dist-info → celeste_ai-0.0.2.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: celeste-ai
-Version: 0.0.1
+Version: 0.0.2
 Summary: Celeste AI Framework - Multi-modal AI interface (placeholder)
-Project-URL: Homepage, https://github.com/agent-kai/celeste-ai
-Project-URL: Repository, https://github.com/agent-kai/celeste-ai
-Author-email: agent-kai <contact@example.com>
+Project-URL: Homepage, https://celeste-ai.co
+Project-URL: Repository, https://github.com/celeste-kai/celeste-ai
+Author-email: agent-kai <kai@celeste-ai.co>
 Keywords: ai,anthropic,google,ml,multimodal,openai
 Classifier: Development Status :: 1 - Planning
 Classifier: Intended Audience :: Developers
@@ -16,10 +16,24 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.9
+Requires-Dist: pydantic-settings>=2.0
+Requires-Dist: pydantic>=2.0
+Provides-Extra: dev
+Requires-Dist: bandit[toml]>=1.7.5; extra == 'dev'
+Requires-Dist: mypy>=1.13.0; extra == 'dev'
+Requires-Dist: pre-commit>=3.5.0; extra == 'dev'
+Requires-Dist: pytest-cov>=7.0; extra == 'dev'
+Requires-Dist: pytest-randomly>=4.0; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Requires-Dist: ruff>=0.8.0; extra == 'dev'
+Requires-Dist: types-requests>=2.31.0; extra == 'dev'
 Description-Content-Type: text/markdown
 
 # Celeste AI Framework
 
+[![semantic-release: conventionalcommits](https://img.shields.io/badge/semantic--release-conventionalcommits-e10079?logo=semantic-release)](https://github.com/semantic-release/semantic-release)
+[![PyPI version](https://badge.fury.io/py/celeste-ai.svg)](https://pypi.org/project/celeste-ai/)
+
 > **Note: This is a placeholder package to reserve the name. The full framework is coming soon!**
 
 Celeste AI will be a unified multi-modal AI framework providing a single interface for:
@@ -56,4 +70,4 @@ pip install "celeste-ai[vision]"       # Image/video generation
 
 **Status**: Package name reserved. Framework in active development.
 
-**Contact**: [GitHub Issues](https://github.com/agent-kai/celeste-ai)
+**Contact**: [GitHub Issues](https://github.com/agent-kai/celeste-ai)

celeste_ai-0.0.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+celeste/__init__.py,sha256=HVfwFSxpiqF9OfbvR4nDee4jUNJbrE9hKXeXlZ2bsls,667
+celeste/artifacts.py,sha256=qYu3o0KZIp9WeEp9Xm8kNWjsiQlHRu0jDChmdzKER2s,1543
+celeste/core.py,sha256=udKfbCppwQsCH3mqhFJrc94tVYeyNDJ8YfF-lgDfulA,687
+celeste/credentials.py,sha256=YrN9ynwmGUWqgrNORTUlvpW4sd5TCsgRDghLpUDxSjM,3221
+celeste/mime_types.py,sha256=KhfqExpv771-wHuaueB_kO6Re08zzDt8ilPwLFe8pm8,836
+celeste/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+celeste_ai-0.0.2.dist-info/METADATA,sha256=edV4J0BXduxSsV6yaf-X906Kra8It3lUhgqKfYTpJo0,2786
+celeste_ai-0.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+celeste_ai-0.0.2.dist-info/RECORD,,

celeste_ai-0.0.1.dist-info/RECORD

@@ -1,4 +0,0 @@
-celeste/__init__.py,sha256=kN3rq_XOYDIJTXt7XZXlkQfFfrS28GILeQUaNIivXMw,695
-celeste_ai-0.0.1.dist-info/METADATA,sha256=7BIAzAFXLiRrVyDdsuPdNLBu3C0sgMcLd4hnawlKXMg,2041
-celeste_ai-0.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-celeste_ai-0.0.1.dist-info/RECORD,,