seekrai 0.2.2__tar.gz → 0.3.0__tar.gz

{seekrai-0.2.2 → seekrai-0.3.0}/PKG-INFO

@@ -1,7 +1,8 @@
 Metadata-Version: 2.1
 Name: seekrai
-Version: 0.2.2
+Version: 0.3.0
 Summary: Python client for SeekrAI
+Home-page: https://gitlab.cb.ntent.com/ml/seekr-py
 License: Apache-2.0
 Author: SeekrFlow
 Author-email: support@seekr.com
@@ -27,6 +28,7 @@ Requires-Dist: tabulate (>=0.9.0,<0.10.0)
 Requires-Dist: tqdm (>=4.66.2,<5.0.0)
 Requires-Dist: typer (>=0.9,<0.13)
 Project-URL: Homepage, https://www.seekr.com/
+Project-URL: Repository, https://gitlab.cb.ntent.com/ml/seekr-py
 Description-Content-Type: text/markdown
 
 The Seekr Python Library is the official Python client for SeekrFlow's API platform, providing a convenient way for interacting with the REST APIs and enables easy integrations with Python 3.8+ applications with easy to use synchronous and asynchronous clients.

{seekrai-0.2.2 → seekrai-0.3.0}/pyproject.toml

@@ -14,7 +14,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.2.2"
+version = "0.3.0"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]
@@ -26,8 +26,8 @@ classifiers = [
     "License :: OSI Approved :: Apache Software License",
     "Operating System :: POSIX :: Linux",
 ]
-#repository = "https://gitlab.cb.ntent.com/ml/seekr-py"
-#homepage = "https://gitlab.cb.ntent.com/ml/seekr-py"
+repository = "https://gitlab.cb.ntent.com/ml/seekr-py"
+homepage = "https://gitlab.cb.ntent.com/ml/seekr-py"
 
 
 [tool.poetry.dependencies]

{seekrai-0.2.2 → seekrai-0.3.0}/src/seekrai/client.py

@@ -19,6 +19,8 @@ class SeekrFlow:
     models: resources.Models
     fine_tuning: resources.FineTuning
     alignment: resources.Alignment
+    projects: resources.Projects
+    deployments: resources.Deployments
 
     # client options
     client: SeekrFlowClient
@@ -79,6 +81,8 @@ class SeekrFlow:
         self.models = resources.Models(self.client)
         self.fine_tuning = resources.FineTuning(self.client)
         self.alignment = resources.Alignment(self.client)
+        self.projects = resources.Projects(self.client)
+        self.deployments = resources.Deployments(self.client)
 
 
 class AsyncSeekrFlow:
@@ -90,6 +94,8 @@ class AsyncSeekrFlow:
     models: resources.AsyncModels
     fine_tuning: resources.AsyncFineTuning
     alignment: resources.AsyncAlignment
+    projects: resources.AsyncProjects
+    deployments: resources.AsyncDeployments
 
     # client options
     client: SeekrFlowClient
@@ -150,6 +156,8 @@ class AsyncSeekrFlow:
         self.models = resources.AsyncModels(self.client)
         self.fine_tuning = resources.AsyncFineTuning(self.client)
         self.alignment = resources.AsyncAlignment(self.client)
+        self.projects = resources.AsyncProjects(self.client)
+        self.deployments = resources.AsyncDeployments(self.client)
 
 
 Client = SeekrFlow

{seekrai-0.2.2 → seekrai-0.3.0}/src/seekrai/resources/__init__.py

@@ -1,11 +1,13 @@
 from seekrai.resources.alignment import Alignment, AsyncAlignment
 from seekrai.resources.chat import AsyncChat, Chat
 from seekrai.resources.completions import AsyncCompletions, Completions
+from seekrai.resources.deployments import AsyncDeployments, Deployments
 from seekrai.resources.embeddings import AsyncEmbeddings, Embeddings
 from seekrai.resources.files import AsyncFiles, Files
 from seekrai.resources.finetune import AsyncFineTuning, FineTuning
 from seekrai.resources.images import AsyncImages, Images
 from seekrai.resources.models import AsyncModels, Models
+from seekrai.resources.projects import AsyncProjects, Projects
 
 
 __all__ = [
@@ -25,4 +27,8 @@ __all__ = [
     "Images",
     "AsyncModels",
     "Models",
+    "AsyncProjects",
+    "Projects",
+    "AsyncDeployments",
+    "Deployments",
 ]

{seekrai-0.2.2 → seekrai-0.3.0}/src/seekrai/resources/chat/completions.py

@@ -29,7 +29,8 @@ class ChatCompletions:
         top_k: int = 5,
         repetition_penalty: float = 1,
         stream: bool = False,
-        logprobs: int = 0,
+        logprobs: bool | None = False,
+        top_logprobs: int | None = 0,
         echo: bool = False,
         n: int = 1,
         safety_model: str | None = None,
@@ -61,8 +62,10 @@ class ChatCompletions:
                 Defaults to None.
             stream (bool, optional): Flag indicating whether to stream the generated completions.
                 Defaults to False.
-            logprobs (int, optional): Number of top-k logprobs to return
-                Defaults to None.
+            logprobs (bool, optional): Whether to return log probabilities of the output tokens.
+                Defaults to False.
+            top_logprobs (int, optional): Number of most likely tokens to return the log probabilities for.
+                If greater than 0, it implies logprobs=True. Defaults to 0.
             echo (bool, optional): Echo prompt in output. Can be used with logprobs to return prompt logprobs.
                 Defaults to None.
             n (int, optional): Number of completions to generate. Setting to None will return a single generation.
@@ -102,6 +105,7 @@ class ChatCompletions:
             repetition_penalty=repetition_penalty,
             stream=stream,
             logprobs=logprobs,
+            top_logprobs=top_logprobs,
             echo=echo,
             n=n,
             safety_model=safety_model,
@@ -143,7 +147,8 @@ class AsyncChatCompletions:
         top_k: int = 5,
         repetition_penalty: float = 1,
         stream: bool = False,
-        logprobs: int = 0,
+        logprobs: bool | None = False,
+        top_logprobs: int | None = 0,
         echo: bool = False,
         n: int = 1,
         safety_model: str | None = None,
@@ -175,8 +180,10 @@ class AsyncChatCompletions:
                 Defaults to None.
             stream (bool, optional): Flag indicating whether to stream the generated completions.
                 Defaults to False.
-            logprobs (int, optional): Number of top-k logprobs to return
-                Defaults to None.
+            logprobs (bool, optional): Whether to return log probabilities of the output tokens.
+                Defaults to False.
+            top_logprobs (int, optional): Number of most likely tokens to return the log probabilities for.
+                If greater than 0, it implies logprobs=True. Defaults to 0.
             echo (bool, optional): Echo prompt in output. Can be used with logprobs to return prompt logprobs.
                 Defaults to None.
             n (int, optional): Number of completions to generate. Setting to None will return a single generation.
@@ -216,6 +223,7 @@ class AsyncChatCompletions:
             repetition_penalty=repetition_penalty,
             stream=stream,
             logprobs=logprobs,
+            top_logprobs=top_logprobs,
             echo=echo,
             n=n,
             safety_model=safety_model,

seekrai-0.3.0/src/seekrai/resources/deployments.py

@@ -0,0 +1,203 @@
+from seekrai.abstract import api_requestor
+from seekrai.seekrflow_response import SeekrFlowResponse
+from seekrai.types import SeekrFlowClient, SeekrFlowRequest
+from seekrai.types.deployments import Deployment as DeploymentSchema
+from seekrai.types.deployments import GetDeploymentsResponse
+
+
+class Deployments:
+    def __init__(self, client: SeekrFlowClient) -> None:
+        self._client = client
+
+    def list(self) -> GetDeploymentsResponse:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url="flow/deployments",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return GetDeploymentsResponse(**response.data)
+
+    def retrieve(self, deployment_id: str) -> DeploymentSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/deployments/{deployment_id}",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return DeploymentSchema(**response.data)
+
+    def create(
+        self,
+        name: str,
+        description: str,
+        model_type: str,
+        model_id: str,
+        n_instances: int,
+    ) -> DeploymentSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="POST",
+                url="flow/deployments",
+                params={
+                    "name": name,
+                    "description": description,
+                    "model_type": model_type,
+                    "model_id": model_id,
+                    "n_instances": n_instances,
+                },
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return DeploymentSchema(**response.data)
+
+    def promote(self, deployment_id: str) -> DeploymentSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="PUT",
+                url=f"flow/deployments/{deployment_id}/promote",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return DeploymentSchema(**response.data)
+
+    def demote(self, deployment_id: str) -> DeploymentSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="PUT",
+                url=f"flow/deployments/{deployment_id}/demote",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return DeploymentSchema(**response.data)
+
+
+class AsyncDeployments:
+    def __init__(self, client: SeekrFlowClient) -> None:
+        self._client = client
+
+    async def list(self) -> GetDeploymentsResponse:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url="flow/deployments",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return GetDeploymentsResponse(**response.data)
+
+    async def retrieve(self, deployment_id: str) -> DeploymentSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/deployments/{deployment_id}",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return DeploymentSchema(**response.data)
+
+    async def create(
+        self,
+        name: str,
+        description: str,
+        model_type: str,
+        model_id: str,
+        n_instances: int,
+    ) -> DeploymentSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="POST",
+                url="flow/deployments",
+                params={
+                    "name": name,
+                    "description": description,
+                    "model_type": model_type,
+                    "model_id": model_id,
+                    "n_instances": n_instances,
+                },
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return DeploymentSchema(**response.data)
+
+    async def promote(self, deployment_id: str) -> DeploymentSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="PUT",
+                url=f"flow/deployments/{deployment_id}/promote",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return DeploymentSchema(**response.data)
+
+    async def demote(self, deployment_id: str) -> DeploymentSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="PUT",
+                url=f"flow/deployments/{deployment_id}/demote",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return DeploymentSchema(**response.data)

{seekrai-0.2.2 → seekrai-0.3.0}/src/seekrai/resources/finetune.py

@@ -24,6 +24,7 @@ class FineTuning:
     def create(
         self,
         *,
+        project_id: int,
         training_config: TrainingConfig,
         infrastructure_config: InfrastructureConfig,
         # wandb_api_key: str | None = None,
@@ -42,7 +43,9 @@ class FineTuning:
         )
 
         parameter_payload = FinetuneRequest(
-            training_config=training_config, infrastructure_config=infrastructure_config
+            project_id=project_id,
+            training_config=training_config,
+            infrastructure_config=infrastructure_config,
         ).model_dump()
 
         response, _, _ = requestor.request(
@@ -250,6 +253,7 @@ class AsyncFineTuning:
     async def create(
         self,
         *,
+        project_id: int,
         training_config: TrainingConfig,
         infrastructure_config: InfrastructureConfig,
     ) -> FinetuneResponse:
@@ -266,7 +270,9 @@ class AsyncFineTuning:
         )
 
         parameter_payload = FinetuneRequest(
-            training_config=training_config, infrastructure_config=infrastructure_config
+            project_id=project_id,
+            training_config=training_config,
+            infrastructure_config=infrastructure_config,
         ).model_dump()
 
         response, _, _ = await requestor.arequest(

seekrai-0.3.0/src/seekrai/resources/projects.py

@@ -0,0 +1,129 @@
+from seekrai.abstract import api_requestor
+from seekrai.seekrflow_response import SeekrFlowResponse
+from seekrai.types import SeekrFlowClient, SeekrFlowRequest
+from seekrai.types.projects import (
+    GetProjectsResponse,
+    PostProjectRequest,
+)
+from seekrai.types.projects import (
+    Project as ProjectSchema,
+)
+
+
+class Projects:
+    def __init__(self, client: SeekrFlowClient) -> None:
+        self._client = client
+
+    def list(self, skip: int = 0, limit: int = 100) -> GetProjectsResponse:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url="flow/projects",
+                params={"skip": skip, "limit": limit},
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return GetProjectsResponse(**response.data)
+
+    def retrieve(self, project_id: int) -> ProjectSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/projects/{project_id}",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return ProjectSchema(**response.data)
+
+    def create(self, name: str, description: str) -> ProjectSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        parameter_payload = PostProjectRequest(
+            name=name,
+            description=description,
+        ).model_dump()
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="POST",
+                url="flow/projects",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return ProjectSchema(**response.data)
+
+
+class AsyncProjects:
+    def __init__(self, client: SeekrFlowClient) -> None:
+        self._client = client
+
+    async def list(self, skip: int = 0, limit: int = 100) -> GetProjectsResponse:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url="flow/projects",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return GetProjectsResponse(**response.data)
+
+    async def retrieve(self, project_id: int) -> ProjectSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/projects/{project_id}",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return ProjectSchema(**response.data)
+
+    async def create(self, name: str, description: str) -> ProjectSchema:
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        parameter_payload = PostProjectRequest(
+            name=name,
+            description=description,
+        ).model_dump()
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="POST",
+                url="flow/projects",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return ProjectSchema(**response.data)

{seekrai-0.2.2 → seekrai-0.3.0}/src/seekrai/types/__init__.py

@@ -16,6 +16,15 @@ from seekrai.types.completions import (
     CompletionRequest,
     CompletionResponse,
 )
+from seekrai.types.deployments import (
+    Deployment,
+    DeploymentProcessor,
+    DeploymentStatus,
+    DeploymentType,
+    GetDeploymentsResponse,
+    HardwareType,
+    NewDeploymentRequest,
+)
 from seekrai.types.embeddings import EmbeddingRequest, EmbeddingResponse
 from seekrai.types.files import (
     FileDeleteResponse,
@@ -40,6 +49,12 @@ from seekrai.types.images import (
     ImageResponse,
 )
 from seekrai.types.models import ModelList, ModelResponse
+from seekrai.types.projects import (
+    GetProjectsResponse,
+    PostProjectRequest,
+    Project,
+    ProjectWithRuns,
+)
 
 
 __all__ = [
@@ -75,4 +90,15 @@ __all__ = [
     "AlignmentResponse",
     "AlignmentJobStatus",
     "AlignmentList",
+    "Project",
+    "ProjectWithRuns",
+    "GetProjectsResponse",
+    "PostProjectRequest",
+    "Deployment",
+    "DeploymentProcessor",
+    "DeploymentStatus",
+    "DeploymentType",
+    "GetDeploymentsResponse",
+    "HardwareType",
+    "NewDeploymentRequest",
 ]

{seekrai-0.2.2 → seekrai-0.3.0}/src/seekrai/types/alignment.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from datetime import datetime
 from enum import Enum
 from typing import List, Literal, Optional

{seekrai-0.2.2 → seekrai-0.3.0}/src/seekrai/types/chat_completions.py

@@ -8,8 +8,9 @@ from pydantic import Field
 from seekrai.types.abstract import BaseModel
 from seekrai.types.common import (
     DeltaContent,
+    DeltaLogProbs,
     FinishReason,
-    LogprobsPart,
+    LogProbs,
     ObjectType,
     PromptPart,
     UsageData,
@@ -89,9 +90,10 @@ class ChatCompletionRequest(BaseModel):
     # stream SSE token chunks
     stream: bool = False
     # return logprobs
-    logprobs: int = 0
+    logprobs: bool | None = False
+    top_logprobs: int | None = 0
     # echo prompt.
-    # can be used with logprobs to return prompt logprobs
+    # can be used with logprobs to return prompt logprobs (is this supported in Seekr API/worker implementation?)
     echo: bool = False
     # number of output generations
     n: int = 1
@@ -105,9 +107,9 @@ class ChatCompletionRequest(BaseModel):
 
 class ChatCompletionChoicesData(BaseModel):
     index: int | None = None
-    logprobs: LogprobsPart | None = None
     finish_reason: FinishReason | None = None
     message: ChatCompletionMessage | None = None
+    logprobs: LogProbs | None = None
 
 
 class ChatCompletionResponse(BaseModel):
@@ -129,9 +131,9 @@ class ChatCompletionResponse(BaseModel):
 
 class ChatCompletionChoicesChunk(BaseModel):
     index: int | None = None
-    logprobs: float | None = None
     finish_reason: FinishReason | None = None
     delta: DeltaContent | None = None
+    logprobs: DeltaLogProbs | None = None
 
 
 class ChatCompletionChunk(BaseModel):

{seekrai-0.2.2 → seekrai-0.3.0}/src/seekrai/types/common.py

@@ -1,9 +1,9 @@
 from __future__ import annotations
 
 from enum import Enum
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Optional
 
-from pydantic import ConfigDict
+from pydantic import ConfigDict, Field
 
 from seekrai.types.abstract import BaseModel
 
@@ -33,6 +33,13 @@ class ObjectType(str, Enum):
     Model = "model"
 
 
+class LogProbs(BaseModel):  # OpenAI style
+    text_offset: List[int] = Field(default_factory=list)
+    token_logprobs: List[Optional[float]] = Field(default_factory=list)
+    tokens: List[str] = Field(default_factory=list)
+    top_logprobs: List[Optional[Dict[str, float]]] = Field(default_factory=list)
+
+
 class LogprobsPart(BaseModel):
     # token list
     tokens: List[str | None] | None = None
@@ -51,6 +58,13 @@ class DeltaContent(BaseModel):
     content: str | None = None
 
 
+class DeltaLogProbs(BaseModel):
+    text_offset: int | None = None
+    token_logprob: float | None = None
+    token: str | None = None
+    top_logprobs: Dict[str, float] | None = None
+
+
 class SeekrFlowRequest(BaseModel):
     model_config = ConfigDict(arbitrary_types_allowed=True)
 

seekrai-0.3.0/src/seekrai/types/deployments.py

@@ -0,0 +1,69 @@
+import enum
+from datetime import datetime
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class DeploymentType(str, enum.Enum):
+    # Matches UI
+    FINE_TUNED_RUN = "Fine-tuned Run"
+    BASE_MODEL = "Base Model"  # TODO - clean up spacing, capital, etc.
+
+
+class DeploymentStatus(str, enum.Enum):
+    # dedicated
+    INACTIVE = "Inactive"  # Shared with serverless.
+    PENDING = "Pending"
+    ACTIVE = "Active"  # Shared with serverless.
+    FAILED = "Failed"
+    STARTED = "Started"
+    SUCCESS = "Success"
+
+
+class HardwareType(str, enum.Enum):
+    # Matches UI
+    SERVERLESS = "Serverless"
+    DEDICATED = "Dedicated"
+
+
+class DeploymentProcessor(str, enum.Enum):
+    GAUDI2 = "GAUDI2"
+    GAUDI3 = "GAUDI3"
+    A100 = "A100"
+    H100 = "H100"
+    XEON = "XEON"
+    NVIDIA = "NVIDIA"  # TODO - this doesnt make sense with A100, etc.
+
+
+class NewDeploymentRequest(BaseModel):
+    model_type: DeploymentType
+    model_id: str
+    name: str = Field(min_length=5, max_length=100)
+    description: str = Field(min_length=5, max_length=1000)
+    n_instances: int = Field(..., ge=1, le=50)
+
+
+class Deployment(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    id: str
+    model_type: DeploymentType
+    model_id: str
+    name: str
+    description: str
+    status: DeploymentStatus
+    memory: Optional[str] = None
+    hardware_type: HardwareType = HardwareType.DEDICATED
+    total_input_tokens: int
+    total_output_tokens: int
+    created_at: datetime
+    last_deployed_at: Optional[datetime] = None
+    updated_at: datetime
+    processor: DeploymentProcessor = DeploymentProcessor.GAUDI2
+    n_instances: int
+    user_id: int
+
+
+class GetDeploymentsResponse(BaseModel):
+    data: list[Deployment]

{seekrai-0.2.2 → seekrai-0.3.0}/src/seekrai/types/finetune.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from datetime import datetime
 from enum import Enum
-from typing import List, Literal
+from typing import List, Literal, Optional
 
 from seekrai.types.abstract import BaseModel
 from seekrai.types.common import (
@@ -118,11 +118,13 @@ class InfrastructureConfig(BaseModel):
     accel_type: AcceleratorType
     n_accel: int
 
+
 class FinetuneRequest(BaseModel):
     """
     Fine-tune request type
     """
 
+    project_id: int
     training_config: TrainingConfig
     infrastructure_config: InfrastructureConfig
 
@@ -168,6 +170,8 @@ class FinetuneResponse(BaseModel):
     # list of fine-tune events
     events: List[FinetuneEvent] | None = None
     inference_available: bool = False
+    project_id: Optional[int] = None  # TODO - fix this
+
     # dataset token count
     # TODO
     # token_count: int | None = None

seekrai-0.3.0/src/seekrai/types/projects.py

@@ -0,0 +1,31 @@
+from datetime import datetime
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class Project(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    id: int
+    name: str
+    description: str
+    user_id: int
+    created_at: datetime
+    updated_at: datetime
+
+
+class ProjectWithRuns(Project):
+    runs: int
+    runs_deployed: int
+    last_modified: datetime
+
+
+class GetProjectsResponse(BaseModel):
+    data: list[ProjectWithRuns]
+
+
+class PostProjectRequest(BaseModel):
+    id: Optional[int] = Field(default=None)
+    name: str = Field(min_length=5, max_length=100)
+    description: str = Field(min_length=5, max_length=1000)