pixie-prompts 0.0.0__tar.gz → 0.1.0__tar.gz

{pixie_prompts-0.0.0 → pixie_prompts-0.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pixie-prompts
-Version: 0.0.0
+Version: 0.1.0
 Summary: Code-first, type-safe prompt management
 License: MIT
 License-File: LICENSE
@@ -23,6 +23,7 @@ Requires-Dist: jsonsubschema (>=0.0.7,<0.0.8)
 Requires-Dist: pydantic (>=2.12.5,<3.0.0)
 Requires-Dist: strawberry-graphql (>=0.288.1) ; extra == "server"
 Requires-Dist: uvicorn (>=0.40.0) ; extra == "server"
+Requires-Dist: watchdog (>=6.0.0) ; extra == "server"
 Project-URL: Changelog, https://github.com/yiouli/pixie-prompts/commits/main/
 Project-URL: Documentation, https://yiouli.github.io/pixie-prompts/
 Project-URL: Homepage, https://gopixie.ai

pixie_prompts-0.1.0/pixie/prompts/__init__.py

@@ -0,0 +1,19 @@
+from . import graphql
+from .prompt import Prompt, Variables
+from .prompt_management import (
+    create_prompt,
+    get_prompt,
+    list_prompts,
+)
+from .storage import initialize_prompt_storage, StorageBackedPrompt
+
+__all__ = [
+    "Prompt",
+    "Variables",
+    "StorageBackedPrompt",
+    "create_prompt",
+    "get_prompt",
+    "graphql",
+    "initialize_prompt_storage",
+    "list_prompts",
+]

pixie_prompts-0.1.0/pixie/prompts/file_watcher.py

@@ -0,0 +1,320 @@
+from contextlib import asynccontextmanager
+import os
+import re
+import sys
+import importlib.util
+from pathlib import Path
+from fastapi import FastAPI
+from watchdog.events import FileSystemEventHandler, FileSystemEvent
+from watchdog.observers import Observer
+import asyncio
+import logging
+
+from pixie.prompts.storage import initialize_prompt_storage
+
+logger = logging.getLogger(__name__)
+
+_storage_observer: Observer | None = None  # type: ignore
+_storage_reload_task: asyncio.Task | None = None
+
+
+def discover_and_load_modules():
+    """Discover and load all Python files that use pixie.prompts.create_prompt, or pixie.create_prompt.
+
+    This function recursively searches the current working directory for Python files
+    """
+    cwd = Path.cwd()
+    # Recursively find all Python files
+    python_files = list(cwd.rglob("*.py"))
+
+    if not python_files:
+        return
+
+    # Add current directory to Python path if not already there
+    if str(cwd) not in sys.path:
+        sys.path.insert(0, str(cwd))
+
+    loaded_count = 0
+    for py_file in python_files:
+        # Skip __init__.py, private files, and anything in site-packages/venv
+        if py_file.name.startswith("_") or any(
+            part in py_file.parts
+            for part in ["site-packages", ".venv", "venv", "__pycache__"]
+        ):
+            continue
+
+        # Quick check if file imports register_application
+        content = py_file.read_text()
+        if not re.search(r"(?:^|\s)pixie(?=[\s.])", content):
+            continue
+
+        # Load the module with a unique name based on path
+        relative_path = py_file.relative_to(cwd)
+        module_name = str(relative_path.with_suffix("")).replace("/", ".")
+        spec = importlib.util.spec_from_file_location(module_name, py_file)
+        if spec and spec.loader:
+            try:
+                module = importlib.util.module_from_spec(spec)
+                sys.modules[module_name] = module
+                spec.loader.exec_module(module)
+                loaded_count += 1
+            except Exception as e:
+                logger.error("Failed to load module %s: %s", module_name, e)
+
+
+def _reload_prompt_storage() -> None:
+    """Reload prompts from disk and refresh any actualized prompts."""
+    from pixie.prompts import storage as storage_module
+
+    storage_instance = storage_module._storage_instance
+    if storage_instance is None:
+        logger.warning("Prompt storage is not initialized; skip reload.")
+        return
+
+    storage_instance.load()
+
+
+async def _debounced_reload(delay_seconds: float) -> None:
+    """Reload storage after a debounce window to collapse bursty events."""
+    try:
+        await asyncio.sleep(delay_seconds)
+        logger.info("Detected prompt storage change; reloading prompts...")
+        _reload_prompt_storage()
+    except asyncio.CancelledError:
+        raise
+    except Exception:
+        logger.exception("Failed to reload prompts after storage change.")
+
+
+class _StorageChangeHandler(FileSystemEventHandler):
+    """Watchdog handler that schedules prompt reloads on any change."""
+
+    # Patterns to ignore
+    IGNORE_PATTERNS = {
+        ".vscode",
+        ".git",
+        "__pycache__",
+        ".pytest_cache",
+        ".mypy_cache",
+        ".ruff_cache",
+        ".ipynb_checkpoints",
+        "~",  # Vim swap files
+        ".swp",
+        ".swo",
+        ".swx",
+        ".tmp",
+        ".temp",
+        ".DS_Store",
+        "thumbs.db",
+        ".lock",
+    }
+
+    ALLOWED_EVENT_TYPES = {"created", "modified", "deleted", "moved"}
+
+    def __init__(
+        self,
+        loop: asyncio.AbstractEventLoop,
+        debounce_seconds: float,
+        watch_extensions: set[str] | None = None,
+    ) -> None:
+        self._loop = loop
+        self._debounce_seconds = debounce_seconds
+        # Default to common prompt file formats if not specified
+        self._watch_extensions = watch_extensions or {
+            ".json",
+            ".jinja",
+        }
+        self._last_seen: dict[Path, tuple[float | None, int | None]] = {}
+
+    def _should_ignore_event(self, event: FileSystemEvent) -> bool:
+        """Check if event should be ignored."""
+        src = event.src_path
+        if not isinstance(src, str):
+            return True
+
+        path = Path(src)
+
+        # Check primary path
+        if self._is_ignored_path(path):
+            return True
+
+        # For move events, also check destination
+        if hasattr(event, "dest_path") and event.dest_path:
+            dest = event.dest_path
+            if not isinstance(dest, str):
+                return True
+            dest_path = Path(dest)
+            # Ignore if either source or dest should be ignored
+            if self._is_ignored_path(dest_path):
+                return True
+
+        return False
+
+    def _is_ignored_path(self, path: Path) -> bool:
+        """Check if a path should be ignored."""
+        path_str = str(path)
+
+        # Ignore if path contains any ignore pattern
+        if any(pattern in path_str for pattern in self.IGNORE_PATTERNS):
+            return True
+
+        # Ignore temporary files (various editor patterns)
+        filename = path.name
+
+        # VSCode creates files like: .file.txt.12345~
+        if filename.startswith(".") and "~" in filename:
+            return True
+
+        # Hidden files starting with .
+        if filename.startswith("."):
+            return True
+
+        # Backup files
+        if filename.startswith("~") or filename.endswith("~"):
+            return True
+
+        # Temporary suffixes
+        if any(
+            filename.endswith(suffix)
+            for suffix in (".tmp", ".temp", ".swp", ".swo", ".swx", ".bak", ".lock")
+        ):
+            return True
+
+        # For file events (not directories), check extension
+        if path.is_file() or (not path.exists() and path.suffix):
+            if path.suffix not in self._watch_extensions:
+                return True
+
+        return False
+
+    def on_any_event(self, event: FileSystemEvent) -> None:
+        # Filter out unwanted events
+        if event.event_type not in self.ALLOWED_EVENT_TYPES:
+            return
+        if self._should_ignore_event(event):
+            return
+
+        # Skip duplicate events where the file's mtime/size did not change
+        if not event.is_directory and event.event_type in {"created", "modified"}:
+            src = event.src_path
+            if isinstance(src, str):
+                path = Path(src)
+                if self._is_ignored_path(path):
+                    return
+                if self._is_duplicate_event(path):
+                    return
+
+        # Ignore pure directory events (unless it's a delete/move that could affect contents)
+        # We still want to know if a directory containing prompt files was deleted/moved
+        if event.is_directory and event.event_type not in ("deleted", "moved"):
+            return
+
+        logger.debug(
+            "Storage change detected: %s (%s)%s",
+            event.src_path,
+            event.event_type,
+            f" -> {event.dest_path}" if hasattr(event, "dest_path") else "",
+        )
+
+        # Avoid blocking the watchdog thread; schedule reload on the event loop.
+        self._loop.call_soon_threadsafe(self._trigger_reload)
+
+    def _trigger_reload(self) -> None:
+        global _storage_reload_task
+
+        # Cancel existing reload task to reset the debounce timer
+        if _storage_reload_task is not None and not _storage_reload_task.done():
+            _storage_reload_task.cancel()
+
+        _storage_reload_task = self._loop.create_task(
+            _debounced_reload(self._debounce_seconds)
+        )
+
+    def _is_duplicate_event(self, path: Path) -> bool:
+        """Return True if the event is a repeat with unchanged mtime/size."""
+        try:
+            stat = path.stat()
+            fingerprint = (stat.st_mtime, stat.st_size)
+        except FileNotFoundError:
+            self._last_seen.pop(path, None)
+            return False
+
+        last = self._last_seen.get(path)
+        if last == fingerprint:
+            return True
+
+        self._last_seen[path] = fingerprint
+        return False
+
+
+async def start_storage_watcher(
+    storage_directory: Path,
+    debounce_seconds: float,
+    watch_extensions: set[str] | None = None,
+) -> None:
+    """Start a watchdog observer on the prompt storage directory.
+
+    Args:
+        storage_directory: Directory to watch for changes
+        debounce_seconds: Delay before triggering reload (collapses rapid events)
+        watch_extensions: Set of file extensions to watch (e.g., {'.yaml', '.json'})
+    """
+    global _storage_observer
+
+    if _storage_observer is not None:
+        return
+
+    loop = asyncio.get_running_loop()
+    storage_directory.mkdir(parents=True, exist_ok=True)
+
+    handler = _StorageChangeHandler(loop, debounce_seconds, watch_extensions)
+    observer = Observer()
+    observer.schedule(handler, str(storage_directory), recursive=True)
+    observer.start()
+
+    _storage_observer = observer
+
+    extensions_str = ", ".join(sorted(handler._watch_extensions))
+    logger.info(
+        "Watching prompt storage at %s for %s files (debounce %.2fs)",
+        storage_directory,
+        extensions_str,
+        debounce_seconds,
+    )
+
+
+async def stop_storage_watcher() -> None:
+    """Stop the watchdog observer if running."""
+    global _storage_observer, _storage_reload_task
+
+    if _storage_reload_task is not None:
+        _storage_reload_task.cancel()
+        try:
+            await _storage_reload_task
+        except asyncio.CancelledError:
+            pass
+        _storage_reload_task = None
+
+    if _storage_observer is not None:
+        _storage_observer.stop()
+        await asyncio.to_thread(_storage_observer.join, 5.0)
+        _storage_observer = None
+
+
+def init_prompt_storage():
+
+    storage_directory = os.getenv("PIXIE_PROMPT_STORAGE_DIR", ".pixie/prompts")
+    initialize_prompt_storage(storage_directory)
+
+    @asynccontextmanager
+    async def lifespan(app: FastAPI):
+        try:
+            nonlocal storage_directory
+            storage_path = Path(storage_directory)
+            watch_interval = float(os.getenv("PIXIE_PROMPT_WATCH_INTERVAL", "1.0"))
+            await start_storage_watcher(storage_path, watch_interval)
+            yield
+        finally:
+            await stop_storage_watcher()
+
+    return lifespan

{pixie_prompts-0.0.0 → pixie_prompts-0.1.0}/pixie/prompts/graphql.py

@@ -9,7 +9,6 @@ from strawberry.scalars import JSON
 
 from pixie.prompts.prompt import variables_definition_to_schema
 from pixie.prompts.prompt_management import get_prompt, list_prompts
-from importlib.metadata import PackageNotFoundError, version
 
 logger = logging.getLogger(__name__)
 
@@ -41,13 +40,22 @@ class TKeyValue:
     value: str
 
 
+@strawberry.type
+class PromptVersion:
+    """Prompt version information."""
+
+    version_id: str
+    content: str
+    created_at: float
+
+
 @strawberry.type
 class Prompt:
     """Full prompt information including versions."""
 
     id: strawberry.ID
     variables_schema: JSON
-    versions: list[TKeyValue]
+    versions: list[PromptVersion]
     default_version_id: str | None
     """default version id can only be None if versions is empty"""
     description: Optional[str] = None
@@ -62,13 +70,7 @@ class Query:
     async def health_check(self) -> str:
         """Health check endpoint."""
         logger.debug("Health check endpoint called")
-        try:
-            version_str = version("pixie-sdk")
-            logger.debug("Pixie SDK version: %s", version_str)
-            return version_str
-        except PackageNotFoundError as e:
-            logger.warning("Failed to get Pixie SDK version: %s", str(e))
-            return "0.0.0"
+        return "0.0.0"
 
     @strawberry.field
     def list_prompts(self) -> list[PromptMetadata]:
@@ -123,7 +125,12 @@ class Query:
                 module=prompt_with_registration.module,
             )
         versions_dict = prompt.get_versions()
-        versions = [TKeyValue(key=k, value=v) for k, v in versions_dict.items()]
+        versions = [
+            PromptVersion(
+                version_id=k, content=v, created_at=prompt.get_version_creation_time(k)
+            )
+            for k, v in versions_dict.items()
+        ]
         default_version_id: str = prompt.get_default_version_id()
         variables_schema = prompt.get_variables_schema()
         return Prompt(

{pixie_prompts-0.0.0 → pixie_prompts-0.1.0}/pixie/prompts/prompt.py

@@ -11,12 +11,12 @@ from jsonsubschema import isSubschema
 from pydantic import BaseModel
 
 
-class PromptVariables(BaseModel):
+class Variables(BaseModel):
     # TODO add validation to prevent fields using reserved names
     pass
 
 
-TPromptVar = TypeVar("TPromptVar", bound=PromptVariables | None)
+TPromptVar = TypeVar("TPromptVar", bound=Variables | None)
 
 
 _prompt_registry: dict[str, "BasePrompt"] = {}
@@ -36,7 +36,7 @@ class _CompiledPrompt:
     value: str
     prompt: "BasePrompt | OutdatedPrompt"
     version_id: str
-    variables: PromptVariables | None
+    variables: Variables | None
 
 
 _compiled_prompt_registry: dict[int, _CompiledPrompt] = {}
@@ -186,7 +186,7 @@ def variables_definition_to_schema(definition: type[TPromptVar]) -> dict[str, An
     if definition is NoneType:
         return EMPTY_VARIABLES_SCHEMA
 
-    return cast(type[PromptVariables], definition).model_json_schema()
+    return cast(type[Variables], definition).model_json_schema()
 
 
 class BasePrompt(BaseUntypedPrompt, Generic[TPromptVar]):

{pixie_prompts-0.0.0 → pixie_prompts-0.1.0}/pixie/prompts/server.py

@@ -4,9 +4,6 @@ import argparse
 import os
 import colorlog
 import logging
-import sys
-import importlib.util
-from pathlib import Path
 from urllib.parse import quote
 
 import dotenv
@@ -15,7 +12,10 @@ from fastapi.middleware.cors import CORSMiddleware
 from strawberry.fastapi import GraphQLRouter
 import uvicorn
 
-from pixie.prompts.storage import initialize_prompt_storage
+from pixie.prompts.file_watcher import (
+    discover_and_load_modules,
+    init_prompt_storage,
+)
 from pixie.prompts.graphql import schema
 
 
@@ -25,42 +25,6 @@ logger = logging.getLogger(__name__)
 _logging_mode: str = "default"
 
 
-def discover_and_load_prompts():
-    """Discover and load all Python files that use pixie.prompts.create_prompt, or pixie.create_prompt.
-
-    This function recursively searches the current working directory for Python files
-    """
-    cwd = Path.cwd()
-    # Recursively find all Python files
-    python_files = list(cwd.rglob("*.py"))
-
-    if not python_files:
-        return
-
-    # Add current directory to Python path if not already there
-    if str(cwd) not in sys.path:
-        sys.path.insert(0, str(cwd))
-
-    loaded_count = 0
-    for py_file in python_files:
-        # Skip __init__.py, private files, and anything in site-packages/venv
-        if py_file.name.startswith("_") or any(
-            part in py_file.parts
-            for part in ["site-packages", ".venv", "venv", "__pycache__"]
-        ):
-            continue
-
-        # Load the module with a unique name based on path
-        relative_path = py_file.relative_to(cwd)
-        module_name = str(relative_path.with_suffix("")).replace("/", ".")
-        spec = importlib.util.spec_from_file_location(module_name, py_file)
-        if spec and spec.loader:
-            module = importlib.util.module_from_spec(spec)
-            sys.modules[module_name] = module
-            spec.loader.exec_module(module)
-            loaded_count += 1
-
-
 def setup_logging(mode: str = "default"):
     """Configure logging for the entire application.
 
@@ -123,22 +87,26 @@ def create_app() -> FastAPI:
     setup_logging(_logging_mode)
 
     # Discover and load applications on every app creation (including reloads)
-    discover_and_load_prompts()
+    discover_and_load_modules()
 
     dotenv.load_dotenv(os.getcwd() + "/.env")
-    storage_directory = os.getenv("PIXIE_PROMPT_STORAGE_DIR", ".pixie/prompts")
-    initialize_prompt_storage(storage_directory)
+    lifespan = init_prompt_storage()
 
     app = FastAPI(
-        title="Pixie SDK Server",
-        description="Server for running AI applications and agents",
+        title="Pixie Prompts Dev Server",
+        description="Server for managing prompts",
         version="0.1.0",
+        lifespan=lifespan,
     )
-
+    # Matches:
+    # 1. http://localhost followed by an optional port (:8080, :3000, etc.)
+    # 2. http://127.0.0.1 followed by an optional port
+    # 3. https://yourdomain.com (the production domain)
+    origins_regex = r"http://(localhost|127\.0\.0\.1)(:\d+)?|https://gopixie\.ai"
     # Add CORS middleware
     app.add_middleware(
         CORSMiddleware,
-        allow_origins=["*"],  # Allows all origins
+        allow_origin_regex=origins_regex,
         allow_credentials=True,
         allow_methods=["*"],  # Allows all methods
         allow_headers=["*"],  # Allows all headers
@@ -155,7 +123,7 @@ def create_app() -> FastAPI:
     @app.get("/")
     async def root():
         return {
-            "message": "Pixie SDK Server",
+            "message": "Pixie Prompts Dev Server",
             "graphiql": "/graphql",
             "version": "0.1.0",
         }