seekrai 0.5.8__tar.gz → 0.5.11__tar.gz

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

@@ -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 → seekrai-0.5.11}/pyproject.toml

@@ -14,7 +14,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.5.8"
+version = "0.5.11"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]

{seekrai-0.5.8 → seekrai-0.5.11}/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.8 → seekrai-0.5.11}/src/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