"linter fix"

Author: Hendrixx-RE

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

@@ -5,6 +5,7 @@
 import os
 import uuid
 from typing import Any, Dict, List, Optional
+from typing import Sequence
 
 import httpx
 from moss_core import (
@@ -23,10 +24,10 @@
 
 logger = logging.getLogger(__name__)
 
-from typing import Sequence
 
 class QueryOptions:
     """Options for search queries."""
+
     def __init__(
         self,
         embedding: Optional[Sequence[float]] = None,
@@ -46,7 +47,6 @@ def __init__(
         self.rerank_model = rerank_model
 
 
-
 def _get_manage_url() -> str:
     """Manage URL, overridable via env for local development."""
     return os.getenv("MOSS_CLOUD_API_MANAGE_URL", CLOUD_API_MANAGE_URL)
@@ -228,10 +228,10 @@ async def query(
                     name,
                 )
             result = await self._query_cloud(name, query, options, override_top_k)
-            
+
         if rerank:
             result = await self._rerank_results(query, result, options)
-            
+
         return result
 
     # -- Internal ---------------------------------------------------
@@ -243,7 +243,11 @@ async def _query_local(
         options: Optional[QueryOptions],
         override_top_k: Optional[int] = None,
     ) -> SearchResult:
-        top_k = override_top_k if override_top_k is not None else (getattr(options, "top_k", None) or 5)
+        top_k = (
+            override_top_k
+            if override_top_k is not None
+            else (getattr(options, "top_k", None) or 5)
+        )
         alpha = getattr(options, "alpha", None)
         if alpha is None:
             alpha = 0.8
@@ -286,7 +290,11 @@ async def _query_cloud(
         override_top_k: Optional[int] = None,
     ) -> SearchResult:
         """Fallback: query via the cloud API when the index is not loaded locally."""
-        top_k = override_top_k if override_top_k is not None else (getattr(options, "top_k", None) or 10)
+        top_k = (
+            override_top_k
+            if override_top_k is not None
+            else (getattr(options, "top_k", None) or 10)
+        )
         query_embedding = getattr(options, "embedding", None)
 
         request_body: Dict[str, Any] = {
@@ -346,14 +354,19 @@ async def _rerank_results(
                 "Install it with: pip install 'moss[rerank]'"
             )
 
-        model_name = getattr(options, "rerank_model", None) or "cross-encoder/ms-marco-MiniLM-L-6-v2"
+        model_name = (
+            getattr(options, "rerank_model", None)
+            or "cross-encoder/ms-marco-MiniLM-L-6-v2"
+        )
 
         def do_rerank():
             if not hasattr(self.__class__, "_cross_encoder_cache"):
                 self.__class__._cross_encoder_cache = {}
 
             if model_name not in self.__class__._cross_encoder_cache:
-                self.__class__._cross_encoder_cache[model_name] = CrossEncoder(model_name)
+                self.__class__._cross_encoder_cache[model_name] = CrossEncoder(
+                    model_name
+                )
 
             model = self.__class__._cross_encoder_cache[model_name]
 
@@ -371,7 +384,6 @@ def do_rerank():
 
         return await asyncio.to_thread(do_rerank)
 
-
     def _resolve_model_id(
         self,
         docs: List[DocumentInfo],