omnixon-lib 3.0.0__py3-none-any.whl

omnixon/__init__.py

@@ -0,0 +1,91 @@
+from .main import Client, MessageStream
+from .schemes import (
+    User,
+    RecentUser,
+    Message,
+    MessageResponse,
+    TraceStep,
+    Agent,
+    EmbeddingStatus,
+    AgentVersion,
+    Attachment,
+    VersionDiff,
+    Block,
+    BlockType,
+    ClientTool,
+    ToolCall,
+    ToolRecord,
+    ToolResult,
+    Defaults,
+    Token,
+    NewToken,
+    Role,
+    ROLES,
+    DailyUsage,
+    MonthlyUsage,
+    RAG,
+    Model,
+    Interrupted,
+    Chat,
+    MCPServer,
+    CoreBlock,
+    ArchivalPassage,
+    RecallHit,
+    MemoryUser,
+    MemfsEntry,
+    MemfsFile,
+    MemfsRevision,
+    MemfsHit,
+    MemfsUser,
+    StoreStatus,
+    EvalCreate,
+    EvalResult,
+    EvalRun,
+    EvalVersionScore,
+)
+
+__all__ = [
+    "EvalCreate",
+    "EvalResult",
+    "EvalRun",
+    "EvalVersionScore",
+    "Client",
+    "MessageStream",
+    # Schemes
+    "User",
+    "RecentUser",
+    "Message",
+    "MessageResponse",
+    "TraceStep",
+    "Agent",
+    "EmbeddingStatus",
+    "AgentVersion",
+    "Attachment",
+    "VersionDiff",
+    "Block",
+    "BlockType",
+    "ClientTool",
+    "ToolResult",
+    "Defaults",
+    "Token",
+    "NewToken",
+    "Role",
+    "ROLES",
+    "DailyUsage",
+    "MonthlyUsage",
+    "RAG",
+    "Model",
+    "Interrupted",
+    "Chat",
+    "MCPServer",
+    "CoreBlock",
+    "ArchivalPassage",
+    "RecallHit",
+    "MemoryUser",
+    "MemfsEntry",
+    "MemfsFile",
+    "MemfsRevision",
+    "MemfsHit",
+    "MemfsUser",
+    "StoreStatus",
+]

omnixon/exceptions.py

@@ -0,0 +1,68 @@
+class UnlinkError(Exception):
+    """Base exception for all Unlink client errors."""
+
+    def __init__(self, message: str, status_code: int = None):
+        super().__init__(message)
+        self.status_code = status_code
+
+
+class NotFound(UnlinkError):
+    """Raised when the requested resource does not exist (HTTP 404)."""
+
+    def __init__(self, message: str = "Resource not found"):
+        super().__init__(message, status_code=404)
+
+
+class NoAccess(UnlinkError):
+    """Raised when the token is invalid or lacks permission (HTTP 401/403)."""
+
+    def __init__(self, message: str = "Access denied: invalid or missing token"):
+        super().__init__(message, status_code=403)
+
+
+class Conflict(UnlinkError):
+    """Raised when the request conflicts with the current state (HTTP 409)."""
+
+    def __init__(self, message: str = "Conflict: resource already exists"):
+        super().__init__(message, status_code=409)
+
+
+class InvalidRequest(UnlinkError):
+    """Raised when the service rejects the data of a request (HTTP 422)."""
+
+    def __init__(self, message: str = "Invalid request"):
+        super().__init__(message, status_code=422)
+
+
+class UpstreamError(UnlinkError):
+    """Raised when what the service depends on failed: the model provider or an MCP
+    server returned an error (HTTP 502) or timed out (HTTP 504). The service itself
+    works; the message says what went wrong, and a retry may help."""
+
+    def __init__(
+        self, message: str = "Upstream service failed", status_code: int = 502
+    ):
+        super().__init__(message, status_code=status_code)
+
+
+class ConnectionError(UnlinkError):
+    """Raised when the client cannot reach the API."""
+
+    def __init__(self, message: str = "Cannot connect to the API"):
+        super().__init__(message)
+
+
+class StreamError(UnlinkError):
+    """Raised when the server reports an error in the middle of a streamed response."""
+
+    def __init__(self, message: str = "Streaming failed"):
+        super().__init__(message)
+
+
+class RateLimited(UnlinkError):
+    """Raised when the agent's Rate limit refuses a request (HTTP 429). `retry_after`: in how many seconds one is let through again
+    (None: the service did not say)."""
+
+    def __init__(self, message: str = "Too many requests", retry_after: int = None):
+        super().__init__(message, status_code=429)
+        self.retry_after = retry_after