lucidicai 2.1.3__py3-none-any.whl → 3.1.0__py3-none-any.whl

lucidicai/api/resources/experiment.py

@@ -0,0 +1,108 @@
+"""Experiment resource API operations."""
+import logging
+from typing import Any, Dict, List, Optional
+
+from ..client import HttpClient
+
+logger = logging.getLogger("Lucidic")
+
+
+class ExperimentResource:
+    """Handle experiment-related API operations."""
+
+    def __init__(
+        self,
+        http: HttpClient,
+        agent_id: Optional[str] = None,
+        production: bool = False,
+    ):
+        """Initialize experiment resource.
+
+        Args:
+            http: HTTP client instance
+            agent_id: Default agent ID for experiments
+            production: Whether to suppress errors in production mode
+        """
+        self.http = http
+        self._agent_id = agent_id
+        self._production = production
+
+    def create(
+        self,
+        experiment_name: str,
+        description: Optional[str] = None,
+        tags: Optional[List[str]] = None,
+        LLM_boolean_evaluators: Optional[List[str]] = None,
+        LLM_numeric_evaluators: Optional[List[str]] = None,
+    ) -> Optional[str]:
+        """Create a new experiment.
+
+        Args:
+            experiment_name: Name of the experiment.
+            description: Optional description.
+            tags: Optional tags for filtering.
+            LLM_boolean_evaluators: Boolean evaluator names.
+            LLM_numeric_evaluators: Numeric evaluator names.
+
+        Returns:
+            The experiment ID if created successfully, None otherwise.
+        """
+        evaluator_names = []
+        if LLM_boolean_evaluators:
+            evaluator_names.extend(LLM_boolean_evaluators)
+        if LLM_numeric_evaluators:
+            evaluator_names.extend(LLM_numeric_evaluators)
+
+        try:
+            response = self.http.post(
+                "createexperiment",
+                {
+                    "agent_id": self._agent_id,
+                    "experiment_name": experiment_name,
+                    "description": description or "",
+                    "tags": tags or [],
+                    "evaluator_names": evaluator_names,
+                },
+            )
+            return response.get("experiment_id")
+        except Exception as e:
+            if self._production:
+                logger.error(f"[ExperimentResource] Failed to create experiment: {e}")
+                return None
+            raise
+
+    async def acreate(
+        self,
+        experiment_name: str,
+        description: Optional[str] = None,
+        tags: Optional[List[str]] = None,
+        LLM_boolean_evaluators: Optional[List[str]] = None,
+        LLM_numeric_evaluators: Optional[List[str]] = None,
+    ) -> Optional[str]:
+        """Create a new experiment (asynchronous).
+
+        See create() for full documentation.
+        """
+        evaluator_names = []
+        if LLM_boolean_evaluators:
+            evaluator_names.extend(LLM_boolean_evaluators)
+        if LLM_numeric_evaluators:
+            evaluator_names.extend(LLM_numeric_evaluators)
+
+        try:
+            response = await self.http.apost(
+                "createexperiment",
+                {
+                    "agent_id": self._agent_id,
+                    "experiment_name": experiment_name,
+                    "description": description or "",
+                    "tags": tags or [],
+                    "evaluator_names": evaluator_names,
+                },
+            )
+            return response.get("experiment_id")
+        except Exception as e:
+            if self._production:
+                logger.error(f"[ExperimentResource] Failed to create experiment: {e}")
+                return None
+            raise

lucidicai/api/resources/feature_flag.py

@@ -0,0 +1,78 @@
+"""Feature flag resource API operations."""
+import logging
+from typing import Any, Dict, Optional
+
+from ..client import HttpClient
+
+logger = logging.getLogger("Lucidic")
+
+
+class FeatureFlagResource:
+    """Handle feature flag-related API operations."""
+
+    def __init__(
+        self,
+        http: HttpClient,
+        agent_id: Optional[str] = None,
+        production: bool = False,
+    ):
+        """Initialize feature flag resource.
+
+        Args:
+            http: HTTP client instance
+            agent_id: Default agent ID for feature flags
+            production: Whether to suppress errors in production mode
+        """
+        self.http = http
+        self._agent_id = agent_id
+        self._production = production
+
+    def get(
+        self,
+        flag_name: str,
+        default: Any = None,
+        context: Optional[Dict[str, Any]] = None,
+    ) -> Any:
+        """Get a feature flag value.
+
+        Args:
+            flag_name: Name of the feature flag.
+            default: Default value if flag is not found.
+            context: Optional context for flag evaluation.
+
+        Returns:
+            The flag value or default.
+        """
+        try:
+            response = self.http.get(
+                "featureflags",
+                {"flag_name": flag_name, "agent_id": self._agent_id},
+            )
+            return response.get("value", default)
+        except Exception as e:
+            if self._production:
+                logger.error(f"[FeatureFlagResource] Failed to get feature flag: {e}")
+                return default
+            raise
+
+    async def aget(
+        self,
+        flag_name: str,
+        default: Any = None,
+        context: Optional[Dict[str, Any]] = None,
+    ) -> Any:
+        """Get a feature flag value (asynchronous).
+
+        See get() for full documentation.
+        """
+        try:
+            response = await self.http.aget(
+                "featureflags",
+                {"flag_name": flag_name, "agent_id": self._agent_id},
+            )
+            return response.get("value", default)
+        except Exception as e:
+            if self._production:
+                logger.error(f"[FeatureFlagResource] Failed to get feature flag: {e}")
+                return default
+            raise

lucidicai/api/resources/prompt.py

@@ -0,0 +1,84 @@
+"""Prompt resource API operations."""
+import logging
+from typing import Any, Dict, Optional
+
+from ..client import HttpClient
+
+logger = logging.getLogger("Lucidic")
+
+
+class PromptResource:
+    """Handle prompt-related API operations."""
+
+    def __init__(self, http: HttpClient, production: bool = False):
+        """Initialize prompt resource.
+
+        Args:
+            http: HTTP client instance
+            production: Whether to suppress errors in production mode
+        """
+        self.http = http
+        self._production = production
+
+    def get(
+        self,
+        prompt_name: str,
+        variables: Optional[Dict[str, Any]] = None,
+        label: str = "production",
+    ) -> str:
+        """Get a prompt from the prompt database.
+
+        Args:
+            prompt_name: Name of the prompt.
+            variables: Variables to interpolate into the prompt.
+            label: Prompt version label (default: "production").
+
+        Returns:
+            The prompt content with variables interpolated.
+        """
+        try:
+            response = self.http.get(
+                "getprompt",
+                {"prompt_name": prompt_name, "label": label},
+            )
+            prompt = response.get("prompt_content", "")
+
+            # Replace variables
+            if variables:
+                for key, value in variables.items():
+                    prompt = prompt.replace(f"{{{key}}}", str(value))
+
+            return prompt
+        except Exception as e:
+            if self._production:
+                logger.error(f"[PromptResource] Failed to get prompt: {e}")
+                return ""
+            raise
+
+    async def aget(
+        self,
+        prompt_name: str,
+        variables: Optional[Dict[str, Any]] = None,
+        label: str = "production",
+    ) -> str:
+        """Get a prompt from the prompt database (asynchronous).
+
+        See get() for full documentation.
+        """
+        try:
+            response = await self.http.aget(
+                "getprompt",
+                {"prompt_name": prompt_name, "label": label},
+            )
+            prompt = response.get("prompt_content", "")
+
+            if variables:
+                for key, value in variables.items():
+                    prompt = prompt.replace(f"{{{key}}}", str(value))
+
+            return prompt
+        except Exception as e:
+            if self._production:
+                logger.error(f"[PromptResource] Failed to get prompt: {e}")
+                return ""
+            raise