seekrai 0.5.7__tar.gz → 0.5.9__tar.gz

{seekrai-0.5.7 → seekrai-0.5.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: seekrai
-Version: 0.5.7
+Version: 0.5.9
 Summary: Python client for SeekrAI
 License: Apache-2.0
 Author: SeekrFlow
@@ -17,10 +17,10 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: click (>=8.1.7,<9.0.0)
 Requires-Dist: eval-type-backport (>=0.1.3,<0.3.0)
 Requires-Dist: filelock (>=3.13.1,<4.0.0)
-Requires-Dist: httpx[http2] (>=0.28.0,<0.29.0)
+Requires-Dist: httpx[http2] (>=0.27.0,<0.28.0)
 Requires-Dist: numpy (>=1.23.5) ; python_version < "3.12"
 Requires-Dist: numpy (>=1.26.0) ; python_version >= "3.12"
-Requires-Dist: pillow (>=10.2.0,<11.0.0)
+Requires-Dist: pillow (>=10.3.0,<11.0.0)
 Requires-Dist: pyarrow (>=10.0.1)
 Requires-Dist: pydantic (>=2.7,<3.0)
 Requires-Dist: requests (>=2.31.0,<3.0.0)

{seekrai-0.5.7 → seekrai-0.5.9}/pyproject.toml

@@ -14,7 +14,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.5.7"
+version = "0.5.9"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]
@@ -40,13 +40,13 @@ pydantic = "^2.7"
 filelock = "^3.13.1"
 eval-type-backport = ">=0.1.3,<0.3.0"
 click = "^8.1.7"
-pillow = "^10.2.0"
+pillow = "^10.3.0"
 pyarrow = ">=10.0.1"
 numpy = [
     { version = ">=1.23.5", python = "<3.12" },
     { version = ">=1.26.0", python = ">=3.12" },
 ]
-httpx = {extras = ["http2"], version = "^0.28.0"}
+httpx = {extras = ["http2"], version = "^0.27.0"}
 
 [tool.poetry.group.quality]
 optional = true

{seekrai-0.5.7 → seekrai-0.5.9}/src/seekrai/resources/chat/completions.py

@@ -112,11 +112,10 @@ class ChatCompletions:
             n=n,
             safety_model=safety_model,
             response_format=response_format,
-            tools=tools or [],
+            tools=tools or None,
         ).model_dump()
         if tool_choice is not None:
             parameter_payload["tool_choice"] = tool_choice
-
         response, _, _ = requestor.request(
             options=SeekrFlowRequest(
                 method="POST",
@@ -158,8 +157,8 @@ class AsyncChatCompletions:
         n: int = 1,
         safety_model: str | None = None,
         response_format: Dict[str, str | Dict[str, Any]] | None = None,
-        # tools: Dict[str, str | Dict[str, Any]] | None = None,
-        # tool_choice: str | Dict[str, str | Dict[str, str]] | None = None,
+        tools: Dict[str, str | Dict[str, Any]] | None = None,
+        tool_choice: str | Dict[str, str | Dict[str, str]] | None = None,
     ) -> AsyncGenerator[ChatCompletionChunk, None] | ChatCompletionResponse:
         """
         Async method to generate completions based on a given prompt using a specified model.
@@ -235,9 +234,10 @@ class AsyncChatCompletions:
             n=n,
             safety_model=safety_model,
             response_format=response_format,
-            # tools=tools,
-            # tool_choice=tool_choice,
+            tools=tools or None,
         ).model_dump()
+        if tool_choice is not None:
+            parameter_payload["tool_choice"] = tool_choice
 
         response, _, _ = await requestor.arequest(
             options=SeekrFlowRequest(

{seekrai-0.5.7 → seekrai-0.5.9}/src/seekrai/resources/embeddings.py

@@ -29,7 +29,6 @@ class Embeddings(ResourceBase):
         Returns:
             EmbeddingResponse: Object containing embeddings
         """
-        raise NotImplementedError("Function not implemented yet")
 
         requestor = api_requestor.APIRequestor(
             client=self._client,
@@ -45,6 +44,7 @@ class Embeddings(ResourceBase):
                 method="POST",
                 url="inference/embeddings",
                 params=parameter_payload,
+                headers={"content-type": "application/json"},
             ),
             stream=False,
         )
@@ -71,7 +71,6 @@ class AsyncEmbeddings(ResourceBase):
         Returns:
             EmbeddingResponse: Object containing embeddings
         """
-        raise NotImplementedError("Function not implemented yet")
 
         requestor = api_requestor.APIRequestor(
             client=self._client,
@@ -87,6 +86,7 @@ class AsyncEmbeddings(ResourceBase):
                 method="POST",
                 url="inference/embeddings",
                 params=parameter_payload,
+                headers={"content-type": "application/json"},
             ),
             stream=False,
         )

{seekrai-0.5.7 → seekrai-0.5.9}/src/seekrai/types/finetune.py

@@ -4,6 +4,8 @@ from datetime import datetime
 from enum import Enum
 from typing import List, Literal, Optional
 
+from pydantic import Field
+
 from seekrai.types.abstract import BaseModel
 from seekrai.types.common import (
     ObjectType,
@@ -103,7 +105,7 @@ class TrainingConfig(BaseModel):
     # number of checkpoints to save
     n_checkpoints: int | None = None
     # training batch size
-    batch_size: int | None = None
+    batch_size: int = Field(..., ge=1, le=1024)
     # up to 40 character suffix for output model name
     experiment_name: str | None = None
     # # weights & biases api key