seekrai 0.5.6__py3-none-any.whl → 0.5.8__py3-none-any.whl

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/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

{seekrai-0.5.6.dist-info → seekrai-0.5.8.dist-info}/METADATA

@@ -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.dist-info → seekrai-0.5.8.dist-info}/RECORD

@@ -16,7 +16,7 @@ seekrai/resources/chat/__init__.py,sha256=KmtPupgECtEN80NyvcnSmieTAFXhwmVxhMHP0q
 seekrai/resources/chat/completions.py,sha256=QcNsZqpJjkbiD_dN9fGBw8hgb_hLiHUqoMzHV_vtkqQ,12258
 seekrai/resources/completions.py,sha256=JhTN_lW2mblfHHONFmPC7QZei3wo5vx6GliMs9FkbOY,8452
 seekrai/resources/deployments.py,sha256=DY7IN7QgqDduCHGNuHENSVwrE5PXFL88jWgh8SES7Qk,5970
-seekrai/resources/embeddings.py,sha256=UMVb6LIJn8KFACMajVRL3SY88bbZl1aQ2Sx5MdT1sQQ,2593
+seekrai/resources/embeddings.py,sha256=7G-VisYrT9J35-hcKB8cXhs8BSi93IfveQKfVSC7diA,2585
 seekrai/resources/explainability.py,sha256=HAFxrPIvaYQxexGk8VPO1R67yDbSB6CqVly0ZY_dQcQ,3495
 seekrai/resources/files.py,sha256=bEn4jfYWfsI2OqKRGCUpnefIN-udNnafItgT2A7m-e4,15329
 seekrai/resources/finetune.py,sha256=Aw8lU9TohZdJGOstex12gg2t-RDppOponnWg203Bn-Q,11304
@@ -53,7 +53,7 @@ seekrai/types/embeddings.py,sha256=OANoLNOs0aceS8NppVvvcNYQbF7-pAOAmcr30pw64OU,7
 seekrai/types/error.py,sha256=uTKISs9aRC4_6zwirtNkanxepN8KY-SqCq0kNbfZylQ,370
 seekrai/types/explainability.py,sha256=l9dp9DJ_GPkHzNw_3zwiGkpAWDETMDN8NSoymhKvdgc,1420
 seekrai/types/files.py,sha256=yjJYT8twY-cNh9AY9qlcN_moTeCfR0tJSSCQsOVB02Y,2708
-seekrai/types/finetune.py,sha256=n65WGm7mXqmJSyo1cb8IAbqpoygYCwDo7yGWPF39ayw,6393
+seekrai/types/finetune.py,sha256=IcIaeaMVoITLApCB_h_TFYtQe2kOGDqZFCh9j77Vlgk,6435
 seekrai/types/images.py,sha256=Fusj8OhVYFsT8kz636lRGGivLbPXo_ZNgakKwmzJi3U,914
 seekrai/types/ingestion.py,sha256=uUdKOR4xqSfAXWQOR1UOltSlOnuyAwKVA1Q2a6Yslk8,919
 seekrai/types/models.py,sha256=9Z0nvLdlAfpF8mNRW5-IqBdDHoE-3qQ5przmIDJgwLo,1345
@@ -65,8 +65,8 @@ seekrai/utils/api_helpers.py,sha256=0Y8BblNIr9h_R12zdmhkxgTlxgoRkbq84QNi4nNWGu8,
 seekrai/utils/files.py,sha256=7ixn_hgV-6pEhYqLyOp-EN0o8c1CzUwJzX9n3PQ5oqo,7164
 seekrai/utils/tools.py,sha256=jgJTL-dOIouDbEJLdQpQfpXhqaz_poQYS52adyUtBjo,1781
 seekrai/version.py,sha256=q6iGQVFor8zXiPP5F-3vy9TndOxKv5JXbaNJ2kdOQws,125
-seekrai-0.5.6.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-seekrai-0.5.6.dist-info/METADATA,sha256=QptNWacnlAuFAUABIfoCqR-XyBvNS0L-zqjKTqWwObA,4780
-seekrai-0.5.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-seekrai-0.5.6.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
-seekrai-0.5.6.dist-info/RECORD,,
+seekrai-0.5.8.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+seekrai-0.5.8.dist-info/METADATA,sha256=aayyfKC7AZUcN4afB112qyaccnW9yWdblbki6N3Z5IY,4780
+seekrai-0.5.8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+seekrai-0.5.8.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
+seekrai-0.5.8.dist-info/RECORD,,