sdk update

Author: CoderOMaster

sdks/python/sdk/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "moss"
-version = "1.0.0b19"
-description = "Python SDK for semantic search with on-device AI capabilities"
+version = "1.4.1"
+description = "Python SDK for semantic search with on-device AI capabilities, including local-first session indexing"
 readme = "README.md"
 license-files = ["LICENSE"]
 authors = [
@@ -28,7 +28,7 @@ requires-python = ">=3.10"
 dependencies = [
     "typing-extensions>=4.0.0",
     "httpx>=0.25.0",
-    "inferedge-moss-core==0.8.7",
+    "inferedge-moss-core==0.17.0",
 ]
 
 [project.optional-dependencies]
@@ -78,6 +78,13 @@ target-version = ['py310']
 profile = "black"
 line_length = 88
 
+[tool.pytest.ini_options]
+asyncio_mode = "strict"
+markers = [
+    "e2e: integration tests that require real cloud credentials (MOSS_TEST_PROJECT_ID / MOSS_TEST_PROJECT_KEY)",
+    "slow: tests that take a long time to run (large indexes, etc.)",
+]
+
 [tool.mypy]
 python_version = "3.10"
 warn_return_any = false

sdks/python/sdk/src/moss/__init__.py

@@ -16,6 +16,16 @@
     await client.load_index('my-index')
     results = await client.query('my-index', 'search query')
     ```
+
+Local-first session indexing:
+    ```python
+    session = await client.session("call-abc123")
+
+    await session.add_docs([DocumentInfo(id="1", text="Customer asked about billing")])
+    results = await session.query("billing issue")
+
+    result = await session.push_index()
+    ```
 """
 
 from moss_core import (
@@ -24,24 +34,31 @@
     IndexInfo,
     IndexStatus,
     IndexStatusValues,
+    LoadIndexesResult,
     ModelRef,
     MutationOptions,
     MutationResult,
     JobStatus,
     JobPhase,
     JobProgress,
     JobStatusResponse,
+    PushIndexResult,
     QueryOptions,
     QueryResultDocumentInfo,
     SearchResult,
 )
 
-from .client.moss_client import MossClient
+from .client.moss_client import MossClient, ParseFileInput
+from .client.session_index import SessionIndex
 
-__version__ = "1.0.0b19"
+__version__ = "1.4.1"
 
 __all__ = [
     "MossClient",
+    "ParseFileInput",
+    "SessionIndex",
+    "PushIndexResult",
+    "LoadIndexesResult",
     # Core data types
     "DocumentInfo",
     "GetDocumentsOptions",

sdks/python/sdk/src/moss/__init__.pyi

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import ClassVar, Dict, List, Optional, Sequence
+from typing import ClassVar, Dict, List, Optional, Sequence, Tuple
 
 
 class MossClient:
@@ -10,13 +10,26 @@ class MossClient:
 
     def __init__(self, project_id: str, project_key: str) -> None: ...
 
+    async def session(
+        self,
+        index_name: str,
+        model_id: Optional[str] = None,
+    ) -> SessionIndex: ...
+
     async def create_index(
         self,
         name: str,
         docs: List[DocumentInfo],
         model_id: Optional[str] = ...,
     ) -> MutationResult: ...
 
+    async def create_index_from_files(
+        self,
+        name: str,
+        files: List[ParseFileInput],
+        model_id: Optional[str] = None,
+    ) -> MutationResult: ...
+
     async def add_docs(
         self,
         name: str,
@@ -53,16 +66,99 @@ class MossClient:
 
     async def unload_index(self, name: str) -> None: ...
 
+    async def load_indexes(
+        self,
+        names: List[str],
+        auto_refresh: bool = False,
+        polling_interval_in_seconds: int = 600,
+    ) -> LoadIndexesResult: ...
+
+    async def unload_indexes(self, names: List[str]) -> None: ...
+
     async def query(
         self,
         name: str,
         query: str,
         options: Optional[QueryOptions] = None,
     ) -> SearchResult: ...
 
+    async def query_multi_index(
+        self,
+        names: List[str],
+        query: str,
+        options: Optional[QueryOptions] = None,
+    ) -> SearchResult: ...
+
+
+class SessionIndex:
+    """Local in-session index for real-time indexing and querying."""
+
+    name: str
+    doc_count: int
+
+    async def add_docs(
+        self,
+        docs: List[DocumentInfo],
+        options: Optional[MutationOptions] = None,
+    ) -> Tuple[int, int]: ...
+
+    async def delete_docs(self, doc_ids: List[str]) -> int: ...
+
+    async def get_docs(
+        self,
+        options: Optional[GetDocumentsOptions] = None,
+    ) -> List[DocumentInfo]: ...
+
+    async def query(
+        self,
+        query: str,
+        options: Optional[QueryOptions] = None,
+    ) -> SearchResult: ...
+
+    async def push_index(self) -> PushIndexResult: ...
+
+
+class ParseFileInput:
+    """Input descriptor for a single file in the parse pipeline."""
+
+    name: str
+    content_type: str
+    path: Optional[str]
+    data: Optional[bytes]
+
+    def __init__(
+        self,
+        name: str,
+        content_type: str,
+        path: Optional[str] = None,
+        data: Optional[bytes] = None,
+    ) -> None: ...
+
+
+class PushIndexResult:
+    """Result from SessionIndex.push_index()."""
+
+    job_id: str
+    index_name: str
+    doc_count: int
+    status: str
+
+
+class LoadIndexesResult:
+    """Outcome of a load_indexes call. Best-effort across the batch."""
+
+    loaded: List[str]
+    failed: Dict[str, str]
+
+    def __init__(
+        self,
+        loaded: Optional[List[str]] = None,
+        failed: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
 
 class MutationResult:
-    """Return value from create_index/add_docs/delete_docs."""
+    """Return value from create_index / add_docs / delete_docs."""
 
     job_id: str
     index_name: str
@@ -86,8 +182,6 @@ class GetDocumentsOptions:
 
 
 class JobStatus:
-    """Enum-like class for job status values."""
-
     PENDING_UPLOAD: ClassVar[str]
     UPLOADING: ClassVar[str]
     BUILDING: ClassVar[str]
@@ -98,8 +192,6 @@ class JobStatus:
 
 
 class JobPhase:
-    """Enum-like class for job phase values."""
-
     DOWNLOADING: ClassVar[str]
     DESERIALIZING: ClassVar[str]
     GENERATING_EMBEDDINGS: ClassVar[str]
@@ -111,17 +203,13 @@ class JobPhase:
 
 
 class JobProgress:
-    """Progress update for a job."""
-
     job_id: str
     status: JobStatus
     progress: float
     current_phase: Optional[JobPhase]
 
 
 class JobStatusResponse:
-    """Full status response from get_job_status."""
-
     job_id: str
     status: JobStatus
     progress: float
@@ -143,12 +231,14 @@ class QueryResultDocumentInfo:
     text: str
     metadata: Optional[Dict[str, str]]
     score: float
+    index_name: Optional[str]
     def __init__(
         self,
         id: str,
         text: str,
         metadata: Optional[Dict[str, str]] = ...,
         score: float = ...,
+        index_name: Optional[str] = ...,
     ) -> None: ...
 
 
@@ -230,6 +320,10 @@ __version__: str
 
 __all__ = [
     "MossClient",
+    "SessionIndex",
+    "ParseFileInput",
+    "PushIndexResult",
+    "LoadIndexesResult",
     "DocumentInfo",
     "GetDocumentsOptions",
     "IndexInfo",