xiaoshiai-hub 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

xiaoshiai_hub/exceptions.py

@@ -25,8 +25,13 @@ class AuthenticationError(HubException):
 
 class HTTPError(HubException):
     """Raised when an HTTP error occurs."""
-    
+
     def __init__(self, message: str, status_code: int = None):
         super().__init__(message)
         self.status_code = status_code
 
+
+class EncryptionError(HubException):
+    """Raised when encryption is required but not provided or encryption fails."""
+    pass
+

xiaoshiai_hub/types.py

@@ -2,9 +2,9 @@
 Type definitions for XiaoShi AI Hub SDK
 """
 
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from datetime import datetime
-from typing import List, Optional, Literal
+from typing import List, Optional, Literal, Dict
 
 
 @dataclass
@@ -13,8 +13,26 @@ class Repository:
     name: str
     organization: str
     type: str  # "models" or "datasets"
-    default_branch: Optional[str] = None
     description: Optional[str] = None
+    metadata: Dict[str, List[str]] = field(default_factory=dict)  # Repository metadata
+    annotations: Dict[str, str] = field(default_factory=dict)  # Repository annotations/metadata
+
+
+@dataclass
+class FileEncryptionMetadata:
+    """Encryption metadata for a file."""
+    path: str
+    algorithm: str
+    encryptedHash: str
+    encryptedSize: int
+
+@dataclass
+class EncryptionMetadata:
+    """Encryption metadata for a repository."""
+    version: str
+    createAt: datetime
+    files: Optional[List[FileEncryptionMetadata]] = None
+
 
 
 @dataclass
@@ -42,7 +60,6 @@ class Ref:
     """Git reference (branch/tag)."""
     name: str
     ref: str
-    fully_name: str
     type: str  # "branch" or "tag"
     hash: str
     is_default: bool = False