lucidicai 3.3.1__tar.gz → 3.4.1__tar.gz

{lucidicai-3.3.1 → lucidicai-3.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lucidicai
-Version: 3.3.1
+Version: 3.4.1
 Summary: Lucidic AI Python SDK
 Author: Andy Liang
 Author-email: andy@lucidic.ai

{lucidicai-3.3.1 → lucidicai-3.4.1}/lucidicai/__init__.py

@@ -34,11 +34,14 @@ from .core.errors import (
     FeatureFlagError,
 )
 
+# Prompt object
+from .api.resources.prompt import Prompt
+
 # Integrations
 from .integrations.livekit import setup_livekit
 
 # Version
-__version__ = "3.3.1"
+__version__ = "3.4.1"
 
 # All exports
 __all__ = [
@@ -53,6 +56,8 @@ __all__ = [
     "InvalidOperationError",
     "PromptError",
     "FeatureFlagError",
+    # Prompt object
+    "Prompt",
     # Integrations
     "setup_livekit",
     # Version

{lucidicai-3.3.1 → lucidicai-3.4.1}/lucidicai/api/resources/prompt.py

@@ -1,6 +1,7 @@
 """Prompt resource API operations."""
 import logging
 import time
+from dataclasses import dataclass
 from typing import Any, Dict, Optional, Tuple, TYPE_CHECKING
 
 from ..client import HttpClient
@@ -11,6 +12,36 @@ if TYPE_CHECKING:
 logger = logging.getLogger("Lucidic")
 
 
+@dataclass
+class Prompt:
+    """Represents a prompt retrieved from the Lucidic prompt database."""
+
+    raw_content: str
+    content: str
+    metadata: Dict[str, Any]
+
+    def __str__(self) -> str:
+        return self.content
+
+    def replace_variables(self, variables: Dict[str, Any]) -> "Prompt":
+        """Replace template variables in the prompt content.
+
+        Replaces {{key}} placeholders in raw_content with the provided
+        variable values and updates content.
+
+        Args:
+            variables: Dictionary mapping variable names to their values.
+
+        Returns:
+            self, for method chaining.
+        """
+        content = self.raw_content
+        for key, value in variables.items():
+            content = content.replace(f"{{{{{key}}}}}", str(value))
+        self.content = content
+        return self
+
+
 class PromptResource:
     """Handle prompt-related API operations."""
 
@@ -52,7 +83,7 @@ class PromptResource:
         variables: Optional[Dict[str, Any]] = None,
         label: str = "production",
         cache_ttl: int = 0,
-    ) -> str:
+    ) -> Prompt:
         """Get a prompt from the prompt database.
 
         Args:
@@ -63,38 +94,40 @@ class PromptResource:
                        positive value = seconds before refetching.
 
         Returns:
-            The prompt content with variables interpolated.
+            A Prompt object with raw_content, content (with variables replaced),
+            and metadata. Use str(prompt) for backward-compatible string access.
         """
         try:
             cache_key = (prompt_name, label)
 
             # Check cache
             if self._is_cache_valid(cache_key, cache_ttl):
-                prompt = self._cache[cache_key]["content"]
+                raw_content = self._cache[cache_key]["content"]
+                metadata = self._cache[cache_key]["metadata"]
             else:
                 response = self.http.get(
                     "getprompt",
                     {"prompt_name": prompt_name, "label": label, "agent_id": self._config.agent_id},
                 )
-                prompt = response.get("prompt_content", "")
+                raw_content = response.get("prompt_content", "")
+                metadata = response.get("metadata", {})
 
                 # Store in cache if caching is enabled
                 if cache_ttl != 0:
                     self._cache[cache_key] = {
-                        "content": prompt,
+                        "content": raw_content,
+                        "metadata": metadata,
                         "timestamp": time.time(),
                     }
 
-            # Replace variables
+            prompt = Prompt(raw_content=raw_content, content=raw_content, metadata=metadata)
             if variables:
-                for key, value in variables.items():
-                    prompt = prompt.replace(f"{{{{{key}}}}}", str(value))
-
+                prompt.replace_variables(variables)
             return prompt
         except Exception as e:
             if self._production:
                 logger.error(f"[PromptResource] Failed to get prompt: {e}")
-                return ""
+                return Prompt(raw_content="", content="", metadata={})
             raise
 
     async def aget(
@@ -103,7 +136,7 @@ class PromptResource:
         variables: Optional[Dict[str, Any]] = None,
         label: str = "production",
         cache_ttl: int = 0,
-    ) -> str:
+    ) -> Prompt:
         """Get a prompt from the prompt database (asynchronous).
 
         See get() for full documentation.
@@ -113,28 +146,30 @@ class PromptResource:
 
             # Check cache
             if self._is_cache_valid(cache_key, cache_ttl):
-                prompt = self._cache[cache_key]["content"]
+                raw_content = self._cache[cache_key]["content"]
+                metadata = self._cache[cache_key]["metadata"]
             else:
                 response = await self.http.aget(
                     "getprompt",
                     {"prompt_name": prompt_name, "label": label, "agent_id": self._config.agent_id},
                 )
-                prompt = response.get("prompt_content", "")
+                raw_content = response.get("prompt_content", "")
+                metadata = response.get("metadata", {})
 
                 # Store in cache if caching is enabled
                 if cache_ttl != 0:
                     self._cache[cache_key] = {
-                        "content": prompt,
+                        "content": raw_content,
+                        "metadata": metadata,
                         "timestamp": time.time(),
                     }
 
+            prompt = Prompt(raw_content=raw_content, content=raw_content, metadata=metadata)
             if variables:
-                for key, value in variables.items():
-                    prompt = prompt.replace(f"{{{{{key}}}}}", str(value))
-
+                prompt.replace_variables(variables)
             return prompt
         except Exception as e:
             if self._production:
                 logger.error(f"[PromptResource] Failed to get prompt: {e}")
-                return ""
+                return Prompt(raw_content="", content="", metadata={})
             raise

{lucidicai-3.3.1 → lucidicai-3.4.1}/lucidicai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lucidicai
-Version: 3.3.1
+Version: 3.4.1
 Summary: Lucidic AI Python SDK
 Author: Andy Liang
 Author-email: andy@lucidic.ai

{lucidicai-3.3.1 → lucidicai-3.4.1}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="lucidicai",
-    version="3.3.1",
+    version="3.4.1",
     packages=find_packages(),
     install_requires=[
         "requests>=2.25.1",