kurious 0.8.4__py3-none-any.whl

kurious/__init__.py

@@ -0,0 +1,204 @@
+"""Kurious Python SDK — client library for the Kurious Engine API."""
+
+from kurious._exceptions import (
+    KuriousError,
+    AIntropyError,  # backward-compat alias
+    AuthenticationError,
+    ConflictError,
+    NotFoundError,
+    PermissionError,
+    RateLimitError,
+    ServerError,
+    ValidationError,
+)
+from kurious.client import Kurious, AIntropy  # AIntropy is backward-compat alias
+from kurious.entitlements import PlanType
+from kurious.jobs import Job, JobsResource, JobThrottled, TERMINAL_STATUSES
+from kurious.types import (
+    APIKey,
+    APIKeyListItem,
+    APIKeyPage,
+    AuditLogEntry,
+    AuditLogPage,
+    AutoIngestConfig,
+    BackfillFanoutResponse,
+    BlendedAnswer,
+    Bookmark,
+    BookmarkList,
+    BulkResult,
+    ChatResponse,
+    Conversation,
+    ConversationList,
+    DailyUsage,
+    DocumentResult,
+    EmailOtpRequestResult,
+    EmailOtpVerifyResult,
+    EmbeddingResponse,
+    Entitlement,
+    EntitlementListPage,
+    EvalResult,
+    EvalResultPage,
+    EvalRun,
+    EvalRunPage,
+    EvalSet,
+    EvalSetPage,
+    IndexStats,
+    IngestionJob,
+    IngestionJobPage,
+    IngestStage,
+    IngestStatus,
+    IntelligentSearchResult,
+    JobChildrenRollup,
+    JobStatus,
+    KGAutoChainConfig,
+    KGJobResponse,
+    KGJobStatus,
+    KGSearchResult,
+    LoginResponse,
+    Member,
+    MemberList,
+    Message,
+    MessageList,
+    NL2SQLDedupResult,
+    NL2SQLPipelineJob,
+    NL2SQLPipelineStatus,
+    NL2SQLResult,
+    OperationResult,
+    PresignedUpload,
+    Project,
+    ProjectConfig,
+    ProjectFile,
+    ProjectFileList,
+    ProjectList,
+    ProjectMetrics,
+    RAGSearchResult,
+    RawSearchResult,
+    ResumeIngestResult,
+    SearchLogEntry,
+    ShareInfo,
+    SharedConversation,
+    SharedMessage,
+    SharedMessageList,
+    SignupResponse,
+    SourceAttribution,
+    StepTiming,
+    StepTimingsResponse,
+    StreamEvent,
+    TokenResponse,
+    UsageBreakdown,
+    UsageSummary,
+    VideoInspectResult,
+    VideoPipelineJob,
+    VideoPipelineStatus,
+    VideoSearchHit,
+    VideoSearchResult,
+    VideoTimeline,
+    VideoTimelineEvent,
+    WhoAmIResponse,
+)
+
+__version__ = "0.8.4"
+
+__all__ = [
+    # Client
+    "Kurious",
+    "AIntropy",  # backward-compat
+    # Exceptions
+    "KuriousError",
+    "AIntropyError",  # backward-compat
+    "AuthenticationError",
+    "ConflictError",
+    "NotFoundError",
+    "PermissionError",
+    "RateLimitError",
+    "ServerError",
+    "ValidationError",
+    # Enum-like literal type aliases
+    "PlanType",
+    # Jobs
+    "Job",
+    "JobsResource",
+    "JobThrottled",
+    "TERMINAL_STATUSES",
+    # Types
+    "APIKey",
+    "APIKeyListItem",
+    "APIKeyPage",
+    "AuditLogEntry",
+    "AuditLogPage",
+    "AutoIngestConfig",
+    "BackfillFanoutResponse",
+    "BlendedAnswer",
+    "Bookmark",
+    "BookmarkList",
+    "BulkResult",
+    "ChatResponse",
+    "Conversation",
+    "ConversationList",
+    "DailyUsage",
+    "DocumentResult",
+    "EmailOtpRequestResult",
+    "EmailOtpVerifyResult",
+    "EmbeddingResponse",
+    "Entitlement",
+    "EntitlementListPage",
+    "EvalResult",
+    "EvalResultPage",
+    "EvalRun",
+    "EvalRunPage",
+    "EvalSet",
+    "EvalSetPage",
+    "IndexStats",
+    "IngestionJob",
+    "IngestionJobPage",
+    "IngestStage",
+    "IngestStatus",
+    "IntelligentSearchResult",
+    "JobChildrenRollup",
+    "JobStatus",
+    "KGAutoChainConfig",
+    "KGJobResponse",
+    "KGJobStatus",
+    "KGSearchResult",
+    "LoginResponse",
+    "Member",
+    "MemberList",
+    "Message",
+    "MessageList",
+    "NL2SQLDedupResult",
+    "NL2SQLPipelineJob",
+    "NL2SQLPipelineStatus",
+    "NL2SQLResult",
+    "OperationResult",
+    "PresignedUpload",
+    "Project",
+    "ProjectConfig",
+    "ProjectFile",
+    "ProjectFileList",
+    "ProjectList",
+    "ProjectMetrics",
+    "RAGSearchResult",
+    "RawSearchResult",
+    "ResumeIngestResult",
+    "SearchLogEntry",
+    "ShareInfo",
+    "SharedConversation",
+    "SharedMessage",
+    "SharedMessageList",
+    "SignupResponse",
+    "SourceAttribution",
+    "StepTiming",
+    "StepTimingsResponse",
+    "StreamEvent",
+    "TokenResponse",
+    "UsageBreakdown",
+    "UsageSummary",
+    "VideoInspectResult",
+    "VideoPipelineJob",
+    "VideoPipelineStatus",
+    "VideoSearchHit",
+    "VideoSearchResult",
+    "VideoTimeline",
+    "VideoTimelineEvent",
+    "WhoAmIResponse",
+]

kurious/_config.py

@@ -0,0 +1,106 @@
+"""Local credential config — read/write ``~/.kurious/config.toml``.
+
+Written by ``kurious init`` and read back by :meth:`AIntropy.from_config`.
+The file holds a live API key, so it is created with ``0600`` permissions.
+
+Kept dependency-free: TOML is read with the stdlib ``tomllib`` (Python 3.11+)
+and falls back to a minimal flat-table parser on 3.9 / 3.10. Writing uses a
+small serialiser since the schema is a flat table of string values.
+"""
+
+from __future__ import annotations
+
+import os
+from pathlib import Path
+
+# Field names — referenced from both the CLI and the client, so centralised.
+KEY_BASE_URL = "base_url"
+KEY_API_KEY = "api_key"
+KEY_COMPANY_ID = "company_id"
+KEY_USER_ID = "user_id"
+
+#: Override the config location with this env var (absolute path).
+ENV_CONFIG_PATH = "KURIOUS_CONFIG_PATH"
+
+_DEFAULT_CONFIG_PATH = Path.home() / ".kurious" / "config.toml"
+_OWNER_RW = 0o600
+
+
+def config_path(override: str | os.PathLike[str] | None = None) -> Path:
+    """Resolve the config path: explicit arg → ``$KURIOUS_CONFIG_PATH`` → default."""
+    if override:
+        return Path(override)
+    env_value = os.environ.get(ENV_CONFIG_PATH)
+    if env_value:
+        return Path(env_value)
+    return _DEFAULT_CONFIG_PATH
+
+
+def read_config(path: str | os.PathLike[str] | None = None) -> dict[str, str]:
+    """Load the config table. Returns ``{}`` if the file does not exist."""
+    target = config_path(path)
+    if not target.exists():
+        return {}
+    text = target.read_text(encoding="utf-8")
+    try:
+        import tomllib  # Python 3.11+
+
+        return dict(tomllib.loads(text))
+    except ModuleNotFoundError:
+        return _parse_flat_toml(text)
+
+
+def write_config(
+    data: dict[str, str | None], path: str | os.PathLike[str] | None = None
+) -> Path:
+    """Write the config table to disk with ``0600`` perms. Returns the path.
+
+    ``None`` values are skipped. The parent directory is created if absent.
+    """
+    target = config_path(path)
+    target.parent.mkdir(parents=True, exist_ok=True)
+    body = "".join(
+        f"{key} = {_quote(str(value))}\n"
+        for key, value in data.items()
+        if value is not None
+    )
+    # Create with 0600 from the start (O_CREAT honours the mode arg) so the
+    # file holding a live API key is never briefly world-readable. chmod after
+    # to fix perms if the file already existed.
+    fd = os.open(target, os.O_WRONLY | os.O_CREAT | os.O_TRUNC, _OWNER_RW)
+    try:
+        os.write(fd, body.encode("utf-8"))
+    finally:
+        os.close(fd)
+    os.chmod(target, _OWNER_RW)
+    return target
+
+
+def _quote(value: str) -> str:
+    """Serialise a string as a basic TOML string literal."""
+    escaped = value.replace("\\", "\\\\").replace('"', '\\"')
+    return f'"{escaped}"'
+
+
+def _unquote(value: str) -> str:
+    """Inverse of :func:`_quote` for the fallback parser."""
+    inner = value.strip()
+    if len(inner) >= 2 and inner[0] == '"' and inner[-1] == '"':
+        inner = inner[1:-1]
+    return inner.replace('\\"', '"').replace("\\\\", "\\")
+
+
+def _parse_flat_toml(text: str) -> dict[str, str]:
+    """Minimal ``key = "value"`` parser for the 3.9 / 3.10 fallback path.
+
+    Only supports the flat string table this module writes — comments and
+    blank lines are ignored; tables / arrays are not expected.
+    """
+    result: dict[str, str] = {}
+    for raw in text.splitlines():
+        line = raw.strip()
+        if not line or line.startswith("#") or "=" not in line:
+            continue
+        key, _, value = line.partition("=")
+        result[key.strip()] = _unquote(value)
+    return result

kurious/_exceptions.py

@@ -0,0 +1,54 @@
+"""AIntropy SDK exceptions."""
+
+from __future__ import annotations
+
+
+class KuriousError(Exception):
+    """Base exception for all SDK errors."""
+
+    def __init__(
+        self, message: str, status_code: int | None = None, body: dict | None = None
+    ):
+        super().__init__(message)
+        self.status_code = status_code
+        self.body = body or {}
+
+
+class AuthenticationError(KuriousError):
+    """Raised on 401 — missing or invalid credentials."""
+
+
+class PermissionError(KuriousError):
+    """Raised on 403 — insufficient permissions."""
+
+
+class NotFoundError(KuriousError):
+    """Raised on 404 — resource does not exist."""
+
+
+class ConflictError(KuriousError):
+    """Raised on 409 — duplicate resource."""
+
+
+class ValidationError(KuriousError):
+    """Raised on 422 — request validation failed."""
+
+
+class RateLimitError(KuriousError):
+    """Raised on 429 — rate limit exceeded."""
+
+    def __init__(
+        self,
+        message: str,
+        retry_after: float | None = None,
+        **kwargs,
+    ):
+        super().__init__(message, **kwargs)
+        self.retry_after = retry_after
+
+
+class ServerError(KuriousError):
+    """Raised on 5xx — server-side failure."""
+
+# Backward-compatibility alias — prefer KuriousError going forward
+AIntropyError = KuriousError