seekrai 0.5.8__py3-none-any.whl → 0.5.11__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/types/finetune.py

@@ -77,6 +77,7 @@ class FinetuneEventType(str, Enum):
 class FineTuneType(str, Enum):
     STANDARD = "STANDARD"
     DPO = "DPO"
+    GRPO = "GRPO"
 
 
 class FinetuneEvent(BaseModel):
@@ -108,6 +109,8 @@ class TrainingConfig(BaseModel):
     batch_size: int = Field(..., ge=1, le=1024)
     # up to 40 character suffix for output model name
     experiment_name: str | None = None
+    # sequence length
+    max_length: int = 2500
     # # weights & biases api key
     # wandb_key: str | None = None
     # IFT by default

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: seekrai
-Version: 0.5.8
+Version: 0.5.11
 Summary: Python client for SeekrAI
 License: Apache-2.0
 Author: SeekrFlow

{seekrai-0.5.8.dist-info → seekrai-0.5.11.dist-info}/RECORD

@@ -13,7 +13,7 @@ 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=7G-VisYrT9J35-hcKB8cXhs8BSi93IfveQKfVSC7diA,2585
@@ -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=IcIaeaMVoITLApCB_h_TFYtQe2kOGDqZFCh9j77Vlgk,6435
+seekrai/types/finetune.py,sha256=pt6lJYXnkHDgMo61zPWvgB04I4sdlijCb0velkrT7TY,6502
 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.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,,
+seekrai-0.5.11.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+seekrai-0.5.11.dist-info/METADATA,sha256=pPqdBDfPu-dJ7GozIqK1lWoxODbSl_tX-X0FKmsSbyg,4781
+seekrai-0.5.11.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+seekrai-0.5.11.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
+seekrai-0.5.11.dist-info/RECORD,,