cocoindex-code 0.2.30__tar.gz → 0.2.31__tar.gz

{cocoindex_code-0.2.30 → cocoindex_code-0.2.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cocoindex-code
-Version: 0.2.30
+Version: 0.2.31
 Summary: MCP server for indexing and querying codebases using CocoIndex
 Project-URL: Homepage, https://github.com/cocoindex-io/cocoindex-code
 Project-URL: Repository, https://github.com/cocoindex-io/cocoindex-code

{cocoindex_code-0.2.30 → cocoindex_code-0.2.31}/src/cocoindex_code/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.2.30'
-__version_tuple__ = version_tuple = (0, 2, 30)
+__version__ = version = '0.2.31'
+__version_tuple__ = version_tuple = (0, 2, 31)
 
 __commit_id__ = commit_id = None

{cocoindex_code-0.2.30 → cocoindex_code-0.2.31}/src/cocoindex_code/litellm_embedder.py

@@ -13,8 +13,6 @@ import numpy as np
 from cocoindex.ops.litellm import LiteLLMEmbedder, litellm
 from numpy.typing import NDArray
 
-litellm.drop_params = True
-
 logger = logging.getLogger(__name__)
 
 _RATE_LIMIT_DELAY_RE = re.compile(r"Please try again in ([0-9.]+)(ms|s)", re.IGNORECASE)
@@ -84,10 +82,12 @@ class PacedLiteLLMEmbedder(LiteLLMEmbedder):
             if self._next_request_at > now:
                 await asyncio.sleep(self._next_request_at - now)
 
+            if not self._model.startswith(("voyage/", "bedrock/")):
+                kwargs["encoding_format"] = "float"
+                kwargs["drop_params"] = True
             response = await self._aembedding_with_rate_limit_retries(
                 model=self._model,
                 input=input,
-                encoding_format="float",
                 **kwargs,
             )