ruff updates

Author: CoderOMaster

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

@@ -2,94 +2,76 @@ from __future__ import annotations
 
 from typing import ClassVar, Dict, List, Optional, Sequence, Tuple
 
-
 class MossClient:
     """Semantic search client for vector similarity operations."""
 
     DEFAULT_MODEL_ID: ClassVar[str]
 
     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,
         docs: List[DocumentInfo],
         options: Optional[MutationOptions] = None,
     ) -> MutationResult: ...
-
     async def delete_docs(
         self,
         name: str,
         doc_ids: List[str],
     ) -> MutationResult: ...
-
     async def get_job_status(self, job_id: str) -> JobStatusResponse: ...
-
     async def get_index(self, name: str) -> IndexInfo: ...
-
     async def list_indexes(self) -> List[IndexInfo]: ...
-
     async def delete_index(self, name: str) -> bool: ...
-
     async def get_docs(
         self,
         name: str,
         options: Optional[GetDocumentsOptions] = None,
     ) -> List[DocumentInfo]: ...
-
     async def load_index(
         self,
         name: str,
         auto_refresh: bool = False,
         polling_interval_in_seconds: int = 600,
     ) -> str: ...
-
     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."""
 
@@ -101,23 +83,18 @@ class SessionIndex:
         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."""
 
@@ -134,7 +111,6 @@ class ParseFileInput:
         data: Optional[bytes] = None,
     ) -> None: ...
 
-
 class PushIndexResult:
     """Result from SessionIndex.push_index()."""
 
@@ -143,7 +119,6 @@ class PushIndexResult:
     doc_count: int
     status: str
 
-
 class LoadIndexesResult:
     """Outcome of a load_indexes call. Best-effort across the batch."""
 
@@ -156,31 +131,27 @@ class LoadIndexesResult:
         failed: Optional[Dict[str, str]] = None,
     ) -> None: ...
 
-
 class MutationResult:
     """Return value from create_index / add_docs / delete_docs."""
 
     job_id: str
     index_name: str
     doc_count: int
 
-
 class MutationOptions:
     """Options for add_docs (e.g. upsert behavior)."""
 
     upsert: Optional[bool]
 
     def __init__(self, upsert: Optional[bool] = None) -> None: ...
 
-
 class GetDocumentsOptions:
     """Options for get_docs (e.g. filter by document IDs)."""
 
     doc_ids: Optional[List[str]]
 
     def __init__(self, doc_ids: Optional[List[str]] = None) -> None: ...
 
-
 class JobStatus:
     PENDING_UPLOAD: ClassVar[str]
     UPLOADING: ClassVar[str]
@@ -190,7 +161,6 @@ class JobStatus:
 
     value: str
 
-
 class JobPhase:
     DOWNLOADING: ClassVar[str]
     DESERIALIZING: ClassVar[str]
@@ -201,14 +171,12 @@ class JobPhase:
 
     value: str
 
-
 class JobProgress:
     job_id: str
     status: JobStatus
     progress: float
     current_phase: Optional[JobPhase]
 
-
 class JobStatusResponse:
     job_id: str
     status: JobStatus
@@ -219,13 +187,11 @@ class JobStatusResponse:
     updated_at: str
     completed_at: Optional[str]
 
-
 class ModelRef:
     id: str
     version: str
     def __init__(self, id: str, version: str) -> None: ...
 
-
 class QueryResultDocumentInfo:
     id: str
     text: str
@@ -241,7 +207,6 @@ class QueryResultDocumentInfo:
         index_name: Optional[str] = ...,
     ) -> None: ...
 
-
 class DocumentInfo:
     id: str
     text: str
@@ -255,7 +220,6 @@ class DocumentInfo:
         embedding: Optional[Sequence[float]] = ...,
     ) -> None: ...
 
-
 class QueryOptions:
     embedding: Optional[Sequence[float]]
     top_k: Optional[int]
@@ -269,7 +233,6 @@ class QueryOptions:
         filter: Optional[dict] = ...,
     ) -> None: ...
 
-
 class IndexInfo:
     id: str
     name: str
@@ -291,7 +254,6 @@ class IndexInfo:
         model: ModelRef,
     ) -> None: ...
 
-
 class SearchResult:
     docs: List[QueryResultDocumentInfo]
     query: str
@@ -305,15 +267,13 @@ class SearchResult:
         time_taken_ms: Optional[int] = None,
     ) -> None: ...
 
-
 class IndexStatus:
     NotStarted: ClassVar[str]
     Building: ClassVar[str]
     Ready: ClassVar[str]
     Failed: ClassVar[str]
     def __init__(self, value: str) -> None: ...
 
-
 IndexStatusValues: Dict[str, str]
 
 __version__: str

sdks/python/sdk/src/moss/client/moss_client.py

@@ -9,8 +9,6 @@
 
 import httpx
 
-_DEFAULT_MANAGE_URL = "https://service.usemoss.dev/v1/manage"
-
 from moss_core import (
     ManageClient,
     DocumentInfo,
@@ -30,6 +28,8 @@
 
 logger = logging.getLogger(__name__)
 
+_DEFAULT_MANAGE_URL = "https://service.usemoss.dev/v1/manage"
+
 
 def _get_manage_url() -> str:
     """Manage URL, overridable via env for staging/local/self-hosted setups."""
@@ -59,6 +59,7 @@ class ParseFileInput:
     Both ``name`` and ``content_type`` are required. Only ``"application/pdf"``
     is currently supported as ``content_type``.
     """
+
     name: str
     content_type: str
     path: Optional[str] = None
@@ -94,8 +95,12 @@ def __init__(self, project_id: str, project_key: str) -> None:
         self._project_key = project_key
         self._client_id = str(uuid.uuid4())
         manage_url = _get_manage_url()
-        self._manage = ManageClient(project_id, project_key, manage_url, self._client_id)
-        self._manager = IndexManager(project_id, project_key, manage_url, self._client_id)
+        self._manage = ManageClient(
+            project_id, project_key, manage_url, self._client_id
+        )
+        self._manager = IndexManager(
+            project_id, project_key, manage_url, self._client_id
+        )
 
     # -- Mutations --------------------------------------------------
 
@@ -108,7 +113,10 @@ async def create_index(
         """Create a new index and populate it with documents."""
         resolved_model_id = self._resolve_model_id(docs, model_id)
         return await asyncio.to_thread(
-            self._manage.create_index, name, docs, resolved_model_id,
+            self._manage.create_index,
+            name,
+            docs,
+            resolved_model_id,
         )
 
     async def create_index_from_files(
@@ -139,6 +147,7 @@ async def create_index_from_files(
                 "Use create_index() with pre-computed embeddings instead."
             )
         from moss_core import ParseFileInput as CoreParseFileInput
+
         core_files = [
             CoreParseFileInput(f.name, f.content_type, path=f.path, data=f.data)
             for f in files
@@ -155,7 +164,10 @@ async def add_docs(
     ) -> MutationResult:
         """Add or update documents in an index."""
         return await asyncio.to_thread(
-            self._manage.add_docs, name, docs, options,
+            self._manage.add_docs,
+            name,
+            docs,
+            options,
         )
 
     async def delete_docs(
@@ -165,7 +177,9 @@ async def delete_docs(
     ) -> MutationResult:
         """Delete documents from an index by their IDs."""
         return await asyncio.to_thread(
-            self._manage.delete_docs, name, doc_ids,
+            self._manage.delete_docs,
+            name,
+            doc_ids,
         )
 
     async def get_job_status(self, job_id: str) -> JobStatusResponse:
@@ -210,7 +224,10 @@ async def load_index(
         """
         try:
             await asyncio.to_thread(
-                self._manager.load_index, name, auto_refresh, polling_interval_in_seconds,
+                self._manager.load_index,
+                name,
+                auto_refresh,
+                polling_interval_in_seconds,
             )
             await asyncio.to_thread(self._manager.load_query_model, name)
             return name
@@ -429,7 +446,12 @@ async def _query_local(
         if query_embedding is None:
             try:
                 return await asyncio.to_thread(
-                    self._manager.query_text, name, query, top_k, alpha, filter_,
+                    self._manager.query_text,
+                    name,
+                    query,
+                    top_k,
+                    alpha,
+                    filter_,
                 )
             except RuntimeError as e:
                 if "requires explicit query embeddings" in str(e):
@@ -440,7 +462,13 @@ async def _query_local(
                 raise
 
         return await asyncio.to_thread(
-            self._manager.query, name, query, list(query_embedding), top_k, alpha, filter_,
+            self._manager.query,
+            name,
+            query,
+            list(query_embedding),
+            top_k,
+            alpha,
+            filter_,
         )
 
     async def _query_cloud(

sdks/python/sdk/tests/test_cloud_fallback.py

@@ -51,9 +51,7 @@ async def cloud_fallback_index(self, moss_client):
         index_name = generate_unique_index_name("test-cloud-fallback")
 
         # Create the index with documents
-        docs = [
-            DocumentInfo(id=doc["id"], text=doc["text"]) for doc in TEST_DOCUMENTS
-        ]
+        docs = [DocumentInfo(id=doc["id"], text=doc["text"]) for doc in TEST_DOCUMENTS]
         await moss_client.create_index(index_name, docs, TEST_MODEL_ID)
 
         yield index_name