seekrai 0.5.7__py3-none-any.whl → 0.5.9__py3-none-any.whl

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/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.7.dist-info → seekrai-0.5.9.dist-info}/METADATA

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

@@ -13,10 +13,10 @@ seekrai/resources/agents/agents.py,sha256=d3C4VaXq4q5WlUr0f8txdRD3hvEKKvz4UQA-a_
 seekrai/resources/agents/threads.py,sha256=BwZ2_6wlezsb12PQjEw1fgdJh5S83SPgD6qZQoGvyIM,14544
 seekrai/resources/alignment.py,sha256=IOKlKK2I9_NhS9pwcrsd9-5OO7lVT8Uw0y_wuGHOnyA,5839
 seekrai/resources/chat/__init__.py,sha256=KmtPupgECtEN80NyvcnSmieTAFXhwmVxhMHP0qhspA4,618
-seekrai/resources/chat/completions.py,sha256=QcNsZqpJjkbiD_dN9fGBw8hgb_hLiHUqoMzHV_vtkqQ,12258
+seekrai/resources/chat/completions.py,sha256=-nFk5aL1ejZ9WUi_ksDHqCnEnT2CjPX-RaigT3N35gA,12317
 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.7.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-seekrai-0.5.7.dist-info/METADATA,sha256=GJcD7w1B_U65lyKQ8CaxD9uYUi-55jXT2iiInLKxaac,4780
-seekrai-0.5.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-seekrai-0.5.7.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
-seekrai-0.5.7.dist-info/RECORD,,
+seekrai-0.5.9.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+seekrai-0.5.9.dist-info/METADATA,sha256=ZOnhU4QoIs5dAPLVX_zic81Zc5OZE0-EpkXad0iWJLc,4780
+seekrai-0.5.9.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+seekrai-0.5.9.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
+seekrai-0.5.9.dist-info/RECORD,,