seekrai 0.5.17__py3-none-any.whl → 0.5.25__py3-none-any.whl

seekrai/client.py

@@ -91,6 +91,7 @@ class SeekrFlow:
         self.deployments = resources.Deployments(self.client)
         self.vector_database = resources.VectorDatabase(self.client)
         self.agents = resources.Agents(self.client)
+        self.tools = resources.Tools(self.client)
         self.observability = resources.AgentObservability(self.client)
         self.explainability = resources.Explainability(self.client)
 
@@ -176,6 +177,7 @@ class AsyncSeekrFlow:
         self.deployments = resources.AsyncDeployments(self.client)
         self.vector_database = resources.AsyncVectorDatabase(self.client)
         self.agents = resources.AsyncAgents(self.client)
+        self.tools = resources.AsyncTools(self.client)
         self.observability = resources.AsyncAgentObservability(self.client)
         self.explainability = resources.AsyncExplainability(self.client)
 

seekrai/resources/__init__.py

@@ -2,10 +2,16 @@ from seekrai.resources.agents import (
     AgentInference,
     AgentObservability,
     Agents,
+    AsyncAgentInference,
     AsyncAgentObservability,
     AsyncAgents,
 )
-from seekrai.resources.alignment import Alignment, AsyncAlignment
+from seekrai.resources.alignment import (
+    Alignment,
+    AsyncAlignment,
+    AsyncSystemPromptResource,
+    SystemPromptResource,
+)
 from seekrai.resources.chat import AsyncChat, Chat
 from seekrai.resources.completions import AsyncCompletions, Completions
 from seekrai.resources.deployments import AsyncDeployments, Deployments
@@ -17,12 +23,15 @@ from seekrai.resources.images import AsyncImages, Images
 from seekrai.resources.ingestion import AsyncIngestion, Ingestion
 from seekrai.resources.models import AsyncModels, Models
 from seekrai.resources.projects import AsyncProjects, Projects
+from seekrai.resources.tools import AsyncTools, Tools
 from seekrai.resources.vectordb import AsyncVectorDatabase, VectorDatabase
 
 
 __all__ = [
     "AsyncAlignment",
     "Alignment",
+    "AsyncSystemPromptResource",
+    "SystemPromptResource",
     "AsyncCompletions",
     "Completions",
     "AsyncChat",
@@ -35,8 +44,8 @@ __all__ = [
     "Files",
     "AsyncImages",
     "Images",
-    "Ingestion",
     "AsyncIngestion",
+    "Ingestion",
     "AsyncModels",
     "Models",
     "AsyncProjects",
@@ -45,11 +54,14 @@ __all__ = [
     "Deployments",
     "AsyncAgents",
     "Agents",
-    "AgentObservability",
     "AsyncAgentObservability",
-    "VectorDatabase",
+    "AgentObservability",
     "AsyncVectorDatabase",
+    "VectorDatabase",
+    "AsyncAgentInference",
     "AgentInference",
     "AsyncExplainability",
     "Explainability",
+    "AsyncTools",
+    "Tools",
 ]

seekrai/resources/alignment.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import Any, Dict, List
 
 from seekrai.abstract import api_requestor
 from seekrai.resources.resource_base import ResourceBase
@@ -7,14 +7,23 @@ from seekrai.types import (
     AlignmentEstimationRequest,
     AlignmentEstimationResponse,
     AlignmentList,
+    AlignmentOutput,
     AlignmentRequest,
     AlignmentResponse,
     AlignmentType,
     SeekrFlowRequest,
+    SystemPrompt,
+    SystemPromptCreateRequest,
+    SystemPromptUpdateRequest,
 )
+from seekrai.types.abstract import SeekrFlowClient
 
 
 class Alignment(ResourceBase):
+    def __init__(self, client: SeekrFlowClient) -> None:
+        super().__init__(client)
+        self.system_prompt = SystemPromptResource(client)
+
     def generate(
         self,
         instructions: str,
@@ -92,6 +101,86 @@ class Alignment(ResourceBase):
 
         return AlignmentResponse(**response.data)
 
+    def outputs(self, id: str) -> List[AlignmentOutput]:
+        """
+        Retrieves output files for an alignment job.
+
+        Args:
+            id (str): Alignment job ID whose outputs to fetch.
+
+        Returns:
+            list[AlignmentOutput]: Collection of alignment output metadata.
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/alignment/{id}/outputs",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+
+        return [AlignmentOutput(**output) for output in response.data]  # type: ignore[arg-type]
+
+    def delete(self, id: str) -> None:
+        """
+        Deletes an alignment job.
+
+        Args:
+            id (str): Alignment job ID to delete.
+
+        Returns:
+            None
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"flow/alignment/{id}",
+            ),
+            stream=False,
+        )
+
+        # Endpoint returns 204 No Content
+        return None
+
+    def cancel(self, id: str) -> AlignmentResponse:
+        """
+        Method to cancel a running alignment job
+
+        Args:
+            id (str): Alignment job ID to cancel. A string that starts with `al-`.
+
+        Returns:
+            AlignmentResponse: Object containing information about cancelled alignment job.
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="POST",
+                url=f"flow/alignment/{id}/cancel",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+
+        return AlignmentResponse(**response.data)
+
     def estimate(self, files: List[str]) -> AlignmentEstimationResponse:
         requestor = api_requestor.APIRequestor(
             client=self._client,
@@ -115,6 +204,10 @@ class Alignment(ResourceBase):
 
 
 class AsyncAlignment(ResourceBase):
+    def __init__(self, client: SeekrFlowClient) -> None:
+        super().__init__(client)
+        self.system_prompt = AsyncSystemPromptResource(client)
+
     async def generate(
         self,
         instructions: str,
@@ -192,6 +285,86 @@ class AsyncAlignment(ResourceBase):
 
         return AlignmentResponse(**response.data)
 
+    async def outputs(self, id: str) -> List[AlignmentOutput]:
+        """
+        Retrieves output files for an alignment job asynchronously.
+
+        Args:
+            id (str): Alignment job ID whose outputs to fetch.
+
+        Returns:
+            list[AlignmentOutput]: Collection of alignment output metadata.
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/alignment/{id}/outputs",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+
+        return [AlignmentOutput(**output) for output in response.data]  # type: ignore[arg-type]
+
+    async def delete(self, id: str) -> None:
+        """
+        Deletes an alignment job asynchronously.
+
+        Args:
+            id (str): Alignment job ID to delete.
+
+        Returns:
+            None
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"flow/alignment/{id}",
+            ),
+            stream=False,
+        )
+
+        # Endpoint returns 204 No Content
+        return None
+
+    async def cancel(self, id: str) -> AlignmentResponse:
+        """
+        Async method to cancel a running alignment job
+
+        Args:
+            id (str): Alignment job ID to cancel. A string that starts with `al-`.
+
+        Returns:
+            AlignmentResponse: Object containing information about cancelled alignment job.
+        """
+
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="POST",
+                url=f"flow/alignment/{id}/cancel",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+
+        return AlignmentResponse(**response.data)
+
     async def estimate(self, files: List[str]) -> AlignmentEstimationResponse:
         requestor = api_requestor.APIRequestor(
             client=self._client,
@@ -212,3 +385,289 @@ class AsyncAlignment(ResourceBase):
 
         assert isinstance(response, SeekrFlowResponse)
         return AlignmentEstimationResponse(**response.data)
+
+
+class SystemPromptResource(ResourceBase):
+    def create(self, source_id: str, instructions: str) -> SystemPrompt:
+        """
+        Creates a new AI-generated system prompt for the given source_id.
+
+        Args:
+            source_id (str): The ID of the source to create the system prompt for
+            instructions (str): Instructions for generating the system prompt
+
+        Returns:
+            SystemPrompt: The created system prompt
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        parameter_payload = SystemPromptCreateRequest(
+            instructions=instructions
+        ).model_dump()
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="POST",
+                url=f"flow/alignment/system_prompt/{source_id}",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return SystemPrompt(**response.data)
+
+    def get(self, source_id: str) -> SystemPrompt:
+        """
+        Retrieves the system prompt for the given source_id.
+
+        Args:
+            source_id (str): The ID of the source to retrieve the system prompt for
+
+        Returns:
+            SystemPrompt: The retrieved system prompt
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/alignment/system_prompt/{source_id}",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return SystemPrompt(**response.data)
+
+    def regenerate(self, source_id: str, instructions: str) -> SystemPrompt:
+        """
+        Regenerates the AI-generated system prompt for the given source_id.
+
+        Args:
+            source_id (str): The ID of the source to regenerate the system prompt for
+            instructions (str): Instructions for regenerating the system prompt
+
+        Returns:
+            SystemPrompt: The regenerated system prompt
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        parameter_payload = SystemPromptCreateRequest(
+            instructions=instructions
+        ).model_dump()
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="POST",
+                url=f"flow/alignment/system_prompt/{source_id}/regenerate",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return SystemPrompt(**response.data)
+
+    def update(self, source_id: str, content: str) -> SystemPrompt:
+        """
+        Updates the system prompt for the given source_id with custom content.
+
+        Args:
+            source_id (str): The ID of the source to update the system prompt for
+            content (str): The custom content for the system prompt
+
+        Returns:
+            SystemPrompt: The updated system prompt
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        parameter_payload = SystemPromptUpdateRequest(content=content).model_dump()
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="PUT",
+                url=f"flow/alignment/system_prompt/{source_id}",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return SystemPrompt(**response.data)
+
+    def delete(self, source_id: str) -> Dict[str, Any]:
+        """
+        Deletes the system prompt for the given source_id.
+
+        Args:
+            source_id (str): The ID of the source to delete the system prompt for
+
+        Returns:
+            dict: A dictionary with the deletion result
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"flow/alignment/system_prompt/{source_id}",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return response.data
+
+
+class AsyncSystemPromptResource(ResourceBase):
+    async def create(self, source_id: str, instructions: str) -> SystemPrompt:
+        """
+        Asynchronously creates a new AI-generated system prompt for the given source_id.
+
+        Args:
+            source_id (str): The ID of the source to create the system prompt for
+            instructions (str): Instructions for generating the system prompt
+
+        Returns:
+            SystemPrompt: The created system prompt
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        parameter_payload = SystemPromptCreateRequest(
+            instructions=instructions
+        ).model_dump()
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="POST",
+                url=f"flow/alignment/system_prompt/{source_id}",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return SystemPrompt(**response.data)
+
+    async def get(self, source_id: str) -> SystemPrompt:
+        """
+        Asynchronously retrieves the system prompt for the given source_id.
+
+        Args:
+            source_id (str): The ID of the source to retrieve the system prompt for
+
+        Returns:
+            SystemPrompt: The retrieved system prompt
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/alignment/system_prompt/{source_id}",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return SystemPrompt(**response.data)
+
+    async def regenerate(self, source_id: str, instructions: str) -> SystemPrompt:
+        """
+        Asynchronously regenerates the AI-generated system prompt for the given source_id.
+
+        Args:
+            source_id (str): The ID of the source to regenerate the system prompt for
+            instructions (str): Instructions for regenerating the system prompt
+
+        Returns:
+            SystemPrompt: The regenerated system prompt
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        parameter_payload = SystemPromptCreateRequest(
+            instructions=instructions
+        ).model_dump()
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="POST",
+                url=f"flow/alignment/system_prompt/{source_id}/regenerate",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return SystemPrompt(**response.data)
+
+    async def update(self, source_id: str, content: str) -> SystemPrompt:
+        """
+        Asynchronously updates the system prompt for the given source_id with custom content.
+
+        Args:
+            source_id (str): The ID of the source to update the system prompt for
+            content (str): The custom content for the system prompt
+
+        Returns:
+            SystemPrompt: The updated system prompt
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        parameter_payload = SystemPromptUpdateRequest(content=content).model_dump()
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="PUT",
+                url=f"flow/alignment/system_prompt/{source_id}",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return SystemPrompt(**response.data)
+
+    async def delete(self, source_id: str) -> Dict[str, Any]:
+        """
+        Asynchronously deletes the system prompt for the given source_id.
+
+        Args:
+            source_id (str): The ID of the source to delete the system prompt for
+
+        Returns:
+            dict: A dictionary with the deletion result
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"flow/alignment/system_prompt/{source_id}",
+            ),
+            stream=False,
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+        return response.data

seekrai/resources/finetune.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 from pathlib import Path
 
 from seekrai.abstract import api_requestor
+from seekrai.error import InvalidRequestError
 from seekrai.resources.resource_base import ResourceBase
 from seekrai.seekrflow_response import SeekrFlowResponse
 from seekrai.types import (
@@ -17,6 +18,29 @@ from seekrai.types import (
 )
 
 
+def validate_lora_support(
+    models_response: SeekrFlowResponse, training_config: TrainingConfig
+) -> None:
+    assert isinstance(models_response, SeekrFlowResponse)
+    model_entry = None
+    for model in models_response.data.get("data", []):
+        model_id = str(model.get("id")) if model.get("id") is not None else None
+        if (
+            model_id == training_config.model
+            or model.get("name") == training_config.model
+        ):
+            model_entry = model
+            break
+    if not model_entry:
+        raise InvalidRequestError(
+            f"Model '{training_config.model}' not found; cannot enable LoRA."
+        )
+    if not model_entry.get("supports_lora", False):
+        raise InvalidRequestError(
+            f"Model '{training_config.model}' does not support LoRA fine-tuning."
+        )
+
+
 class FineTuning(ResourceBase):
     def create(
         self,
@@ -39,6 +63,16 @@ class FineTuning(ResourceBase):
             client=self._client,
         )
 
+        if training_config.lora_config is not None:
+            models_response, _, _ = requestor.request(
+                options=SeekrFlowRequest(
+                    method="GET",
+                    url="flow/models",
+                ),
+                stream=False,
+            )
+            validate_lora_support(models_response, training_config)
+
         parameter_payload = FinetuneRequest(
             project_id=project_id,
             training_config=training_config,
@@ -263,6 +297,16 @@ class AsyncFineTuning(ResourceBase):
             client=self._client,
         )
 
+        if training_config.lora_config is not None:
+            models_response, _, _ = await requestor.arequest(
+                options=SeekrFlowRequest(
+                    method="GET",
+                    url="flow/models",
+                ),
+                stream=False,
+            )
+            validate_lora_support(models_response, training_config)
+
         parameter_payload = FinetuneRequest(
             project_id=project_id,
             training_config=training_config,