seekrai 0.5.6__tar.gz → 0.5.8__tar.gz

{seekrai-0.5.6 → seekrai-0.5.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: seekrai
-Version: 0.5.6
+Version: 0.5.8
 Summary: Python client for SeekrAI
 License: Apache-2.0
 Author: SeekrFlow
@@ -17,7 +17,7 @@ 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.3.0,<11.0.0)

{seekrai-0.5.6 → seekrai-0.5.8}/pyproject.toml

@@ -14,7 +14,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.5.6"
+version = "0.5.8"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]
@@ -46,7 +46,7 @@ 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.6 → seekrai-0.5.8}/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.6 → seekrai-0.5.8}/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