seekrai 0.5.2__py3-none-any.whl → 0.5.24__py3-none-any.whl

seekrai/resources/explainability.py

@@ -0,0 +1,92 @@
+from typing import Optional
+
+from seekrai.abstract import api_requestor
+from seekrai.resources.resource_base import ResourceBase
+from seekrai.types import (
+    SeekrFlowRequest,
+)
+from seekrai.types.explainability import (
+    InfluentialFinetuningDataRequest,
+    InfluentialFinetuningDataResponse,
+)
+
+
+class Explainability(ResourceBase):
+    def get_influential_finetuning_data(
+        self,
+        model_id: str,
+        question: str,
+        system_prompt: Optional[str] = None,
+        answer: Optional[str] = None,
+    ) -> InfluentialFinetuningDataResponse:
+        """
+        Retrieve influential QA pair fine tuning data for a specific model.
+        Args:
+            - model_id (str): ID of the model to explain.
+            - question (str): question from user,
+            - system_prompt (str | None): System prompt for the user's question.
+            - answer (str | None): answer of the finetuned model to the question; if None, the answer is retrieved from the finetuned model specified by model_id,
+        Returns:
+            InfluentialFinetuningDataResponse: Object containing the influential fine tuning data.
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+        # Create query parameters dictionary
+        parameter_payload = InfluentialFinetuningDataRequest(
+            question=question, system_prompt=system_prompt, answer=answer
+        ).model_dump()
+
+        # if limit is not None:
+        #     params["limit"] = limit
+        # TODO  limits =? timeout: float | None = None,  max_retries: int | None = None,
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/explain/models/{model_id}/influential-finetuning-data",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+        return InfluentialFinetuningDataResponse(**response.data)
+
+
+class AsyncExplainability(ResourceBase):
+    async def get_influential_finetuning_data(
+        self,
+        model_id: str,
+        question: str,
+        system_prompt: Optional[str] = None,
+        answer: Optional[str] = None,
+    ) -> InfluentialFinetuningDataResponse:
+        """
+        Retrieve influential QA pair finetuning data for a specific model asynchronously.
+        Args:
+            - model_id (str): ID of the model to explain.
+            - question (str): question from user,
+            - system_prompt (str | None): System prompt for the user's question.
+            - answer (str | None): answer of the finetuned model to the question; if None, the answer is retrieved from the finetuned model specified by model_id,
+        Returns:
+            InfluentialFinetuningDataResponse: Object containing the influential finetuning data.
+        """
+        requestor = api_requestor.APIRequestor(
+            client=self._client,
+        )
+        # Create query parameters dictionary
+        parameter_payload = InfluentialFinetuningDataRequest(
+            model_id=model_id,
+            question=question,
+            system_prompt=system_prompt,
+            answer=answer,
+        ).model_dump()
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="GET",
+                url=f"flow/explain/models/{model_id}/influential-finetuning-data",
+                params=parameter_payload,
+            ),
+            stream=False,
+        )
+        return InfluentialFinetuningDataResponse(**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,

seekrai/resources/ingestion.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 
 from seekrai.abstract import api_requestor
 from seekrai.resources.resource_base import ResourceBase
@@ -13,6 +13,7 @@ class Ingestion(ResourceBase):
     def ingest(
         self,
         files: List[str],
+        method: Optional[str] = "accuracy-optimized",
     ) -> IngestionResponse:
         """
         Start an ingestion job for the specified files.
@@ -27,9 +28,7 @@ class Ingestion(ResourceBase):
             client=self._client,
         )
 
-        parameter_payload = IngestionRequest(
-            files=files,
-        ).model_dump()
+        parameter_payload = IngestionRequest(files=files, method=method).model_dump()
 
         response, _, _ = requestor.request(
             options=SeekrFlowRequest(
@@ -95,6 +94,7 @@ class AsyncIngestion(ResourceBase):
     async def ingest(
         self,
         files: List[str],
+        method: Optional[str] = "accuracy-optimized",
     ) -> IngestionResponse:
         """
         Start an ingestion job for the specified files asynchronously.
@@ -109,9 +109,7 @@ class AsyncIngestion(ResourceBase):
             client=self._client,
         )
 
-        parameter_payload = IngestionRequest(
-            files=files,
-        ).model_dump()
+        parameter_payload = IngestionRequest(files=files, method=method).model_dump()
 
         response, _, _ = await requestor.arequest(
             options=SeekrFlowRequest(

seekrai/resources/models.py

@@ -5,7 +5,6 @@ from seekrai.resources.resource_base import ResourceBase
 from seekrai.seekrflow_response import SeekrFlowResponse
 from seekrai.types import ModelList, ModelResponse, SeekrFlowRequest
 from seekrai.types.common import ObjectType
-from seekrai.utils import parse_timestamp
 
 
 class Models(ResourceBase):
@@ -38,7 +37,6 @@ class Models(ResourceBase):
                 object=ObjectType.Model,
                 name=model["name"],
                 bytes=model["size"],
-                created_at=parse_timestamp(model["created_at"]),
                 model_type=model["model_type"],
             )
             for model in response.data["data"]
@@ -79,7 +77,6 @@ class AsyncModels(ResourceBase):
                 object=ObjectType.Model,
                 name=model["name"],
                 bytes=model["size"],
-                created_at=parse_timestamp(model["created_at"]),
                 model_type=model["model_type"],
             )
             for model in response.data["data"]

seekrai/resources/vectordb.py

@@ -107,7 +107,8 @@ class VectorDatabase(ResourceBase):
         self,
         database_id: str,
         files: List[str],
-        method: str,
+        method: Optional[str] = "accuracy-optimized",
+        chunking_method: Optional[str] = "markdown",
         token_count: int = 800,
         overlap_tokens: int = 100,
     ) -> VectorDatabaseIngestionResponse:
@@ -129,6 +130,7 @@ class VectorDatabase(ResourceBase):
         parameter_payload = VectorDatabaseIngestionRequest(
             file_ids=files,
             method=method,
+            chunking_method=chunking_method,
             token_count=token_count,
             overlap_tokens=overlap_tokens,
         ).model_dump()
@@ -248,6 +250,21 @@ class VectorDatabase(ResourceBase):
         # The endpoint returns 204 No Content
         return None
 
+    def delete_file(self, database_id: str, file_id: str) -> None:
+        """Delete a file from a vector database."""
+        requestor = api_requestor.APIRequestor(client=self._client)
+
+        response, _, _ = requestor.request(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"flow/vectordb/{database_id}/files/{file_id}",
+            ),
+            stream=False,
+        )
+
+        # The endpoint returns 204 No Content
+        return None
+
 
 class AsyncVectorDatabase(ResourceBase):
     async def create(
@@ -338,7 +355,8 @@ class AsyncVectorDatabase(ResourceBase):
         self,
         database_id: str,
         files: List[str],
-        method: str,
+        method: Optional[str] = "accuracy-optimized",
+        chunking_method: Optional[str] = "markdown",
         token_count: int = 800,
         overlap_tokens: int = 100,
     ) -> VectorDatabaseIngestionResponse:
@@ -360,6 +378,7 @@ class AsyncVectorDatabase(ResourceBase):
         parameter_payload = VectorDatabaseIngestionRequest(
             file_ids=files,
             method=method,
+            chunking_method=chunking_method,
             token_count=token_count,
             overlap_tokens=overlap_tokens,
         ).model_dump()
@@ -480,3 +499,18 @@ class AsyncVectorDatabase(ResourceBase):
 
         # The endpoint returns 204 No Content
         return None
+
+    async def delete_file(self, database_id: str, file_id: str) -> None:
+        """Delete a file from a vector database asynchronously."""
+        requestor = api_requestor.APIRequestor(client=self._client)
+
+        response, _, _ = await requestor.arequest(
+            options=SeekrFlowRequest(
+                method="DELETE",
+                url=f"flow/vectordb/{database_id}/files/{file_id}",
+            ),
+            stream=False,
+        )
+
+        # The endpoint returns 204 No Content
+        return None

seekrai/types/__init__.py

@@ -4,7 +4,6 @@ from seekrai.types.agents import (
     AgentDeleteResponse,
     AgentStatus,
     CreateAgentRequest,
-    Env,
     EnvConfig,
     FileSearch,
     FileSearchEnv,
@@ -13,10 +12,14 @@ from seekrai.types.agents import (
     InputMessage,
     InputText,
     MessageUpdateRequest,
+    ModelSettings,
     OutputGuardrail,
     OutputMessage,
     OutputText,
+    ReasoningEffort,
     Run,
+    RunPython,
+    RunPythonEnv,
     RunRequest,
     RunResponse,
     RunStatus,
@@ -42,16 +45,29 @@ from seekrai.types.agents import (
     Tool,
     ToolBase,
     ToolType,
+    WebSearch,
+    WebSearchEnv,
+)
+from seekrai.types.agents.tools.schemas import (
+    FileSearch,
+    FileSearchEnv,
+    RunPython,
+    RunPythonEnv,
+    WebSearch,
+    WebSearchEnv,
 )
-from seekrai.types.agents.tools.schemas import FileSearch, FileSearchEnv
 from seekrai.types.alignment import (
     AlignmentEstimationRequest,
     AlignmentEstimationResponse,
     AlignmentJobStatus,
     AlignmentList,
+    AlignmentOutput,
     AlignmentRequest,
     AlignmentResponse,
     AlignmentType,
+    SystemPrompt,
+    SystemPromptCreateRequest,
+    SystemPromptUpdateRequest,
 )
 from seekrai.types.chat_completions import (
     ChatCompletionChunk,
@@ -90,6 +106,7 @@ from seekrai.types.finetune import (
     FinetuneRequest,
     FinetuneResponse,
     InfrastructureConfig,
+    LoRAConfig,
     TrainingConfig,
 )
 from seekrai.types.images import (
@@ -123,6 +140,7 @@ __all__ = [
     "FinetuneDownloadResult",
     "InfrastructureConfig",
     "TrainingConfig",
+    "LoRAConfig",
     "FileRequest",
     "FileResponse",
     "FileList",
@@ -139,8 +157,12 @@ __all__ = [
     "AlignmentEstimationResponse",
     "AlignmentResponse",
     "AlignmentJobStatus",
+    "AlignmentOutput",
     "AlignmentList",
     "AlignmentType",
+    "SystemPrompt",
+    "SystemPromptCreateRequest",
+    "SystemPromptUpdateRequest",
     "Project",
     "ProjectWithRuns",
     "GetProjectsResponse",
@@ -183,15 +205,20 @@ __all__ = [
     "RunUsage",
     "RunStatus",
     "RunStepUsage",
+    "ModelSettings",
     "Agent",
     "AgentStatus",
     "CreateAgentRequest",
+    "ReasoningEffort",
     "AgentDeleteResponse",
     "ToolBase",
     "ToolType",
     "EnvConfig",
-    "Env",
     "Tool",
     "FileSearch",
     "FileSearchEnv",
+    "RunPython",
+    "RunPythonEnv",
+    "WebSearch",
+    "WebSearchEnv",
 ]

seekrai/types/agents/__init__.py

@@ -3,8 +3,15 @@ from seekrai.types.agents.agent import (
     AgentDeleteResponse,
     AgentStatus,
     CreateAgentRequest,
+    ReasoningEffort,
+)
+from seekrai.types.agents.python_functions import (
+    DeletePythonFunctionResponse,
+    PythonFunctionBase,
+    PythonFunctionResponse,
 )
 from seekrai.types.agents.runs import (
+    ModelSettings,
     Run,
     RunRequest,
     RunResponse,
@@ -39,8 +46,15 @@ from seekrai.types.agents.threads import (
     ThreadMessageContentType,
     ThreadStatus,
 )
-from seekrai.types.agents.tools import Env, EnvConfig, Tool, ToolBase, ToolType
-from seekrai.types.agents.tools.schemas import FileSearch, FileSearchEnv
+from seekrai.types.agents.tools import EnvConfig, Tool, ToolBase, ToolType
+from seekrai.types.agents.tools.schemas import (
+    FileSearch,
+    FileSearchEnv,
+    RunPython,
+    RunPythonEnv,
+    WebSearch,
+    WebSearchEnv,
+)
 
 
 __all__ = [
@@ -51,6 +65,7 @@ __all__ = [
     "RunUsage",
     "RunStatus",
     "RunStepUsage",
+    "ModelSettings",
     "MessageUpdateRequest",
     "ThreadCreateRequest",
     "ThreadStatus",
@@ -78,12 +93,19 @@ __all__ = [
     "Agent",
     "AgentStatus",
     "CreateAgentRequest",
+    "ReasoningEffort",
     "AgentDeleteResponse",
     "ToolBase",
     "ToolType",
     "EnvConfig",
-    "Env",
     "Tool",
     "FileSearch",
     "FileSearchEnv",
+    "RunPython",
+    "RunPythonEnv",
+    "WebSearch",
+    "WebSearchEnv",
+    "PythonFunctionBase",
+    "PythonFunctionResponse",
+    "DeletePythonFunctionResponse",
 ]

seekrai/types/agents/agent.py

@@ -14,11 +14,17 @@ class AgentStatus(str, enum.Enum):
     FAILED = "Failed"
 
 
+class ReasoningEffort(str, enum.Enum):
+    PERFORMANCE_OPTIMIZED = "performance_optimized"
+    SPEED_OPTIMIZED = "speed_optimized"
+
+
 class CreateAgentRequest(BaseModel):
     name: str
     instructions: str
     tools: list[Tool]
     model_id: str
+    reasoning_effort: Optional[ReasoningEffort] = None
 
 
 class Agent(BaseModel):
@@ -35,8 +41,13 @@ class Agent(BaseModel):
     updated_at: datetime
     last_deployed_at: Optional[datetime] = None
     active_duration: int = Field(default=0, ge=0)
+    reasoning_effort: ReasoningEffort
 
 
 class AgentDeleteResponse(BaseModel):
     id: str
     deleted: bool
+
+
+class UpdateAgentRequest(CreateAgentRequest):
+    pass

seekrai/types/agents/observability.py

@@ -0,0 +1,34 @@
+from datetime import datetime
+from typing import Any, Dict, List, Optional
+
+from pydantic import field_serializer
+
+from seekrai.types.abstract import BaseModel
+
+
+class ObservabilitySpansRequest(BaseModel):
+    """Request model for requesting observability spans."""
+
+    min_start_datetime: Optional[datetime]
+    max_start_datetime: Optional[datetime]
+    agent_id: Optional[str]
+    run_id: Optional[str]
+    trace_id: Optional[str]
+    thread_id: Optional[str]
+    group: Optional[str]
+    metadata: Optional[dict[str, str]]
+    limit: int = 100
+    order: str = "desc"
+    offset: int = 0
+
+    @field_serializer("min_start_datetime", "max_start_datetime")
+    def serialize_dt(self, dt: Optional[datetime], _info: Any) -> Optional[str]:
+        if dt is not None:
+            return dt.isoformat()
+        return None
+
+
+class ObservabilitySpansResponse(BaseModel):
+    """Response model for requesting observability spans."""
+
+    spans: List[Dict[str, Any]]

seekrai/types/agents/python_functions.py

@@ -0,0 +1,29 @@
+from datetime import datetime
+
+from pydantic import BaseModel, ConfigDict
+
+
+class PythonFunctionBase(BaseModel):
+    """Base model for a Python function, including metadata fields."""
+
+    model_config = ConfigDict(from_attributes=True)
+    id: str
+    version: int
+    name: str
+    description: str
+    active: bool
+
+
+class PythonFunctionResponse(PythonFunctionBase):
+    """Response model for a Python function, including code and user info."""
+
+    code: str
+    user_id: str
+    created_at: datetime
+    updated_at: datetime
+
+
+class DeletePythonFunctionResponse(BaseModel):
+    """Response model for Python function deletion."""
+
+    deleted: bool

seekrai/types/agents/runs.py

@@ -1,16 +1,65 @@
 import datetime
 from enum import Enum
-from typing import Any, Optional, Union
+from typing import Any, Dict, Optional, Union
 
+import pydantic
 from pydantic import Field
 
 from seekrai.types.abstract import BaseModel
 
 
+class ModelSettings(BaseModel):
+    """Settings to use when calling an LLM.
+
+    This class holds optional model configuration parameters (e.g. temperature,
+    top_p, penalties, truncation, etc.).
+
+    Not all models/providers support all of these parameters, so please check the API documentation
+    for the specific model and provider you are using.
+    """
+
+    temperature: float = Field(default=1.0, ge=0.0, le=2.0)
+    top_p: float = Field(default=1.0, ge=0.0, le=1.0)
+    frequency_penalty: float = Field(default=0.0, ge=-2.0, le=2.0)
+    presence_penalty: float = Field(default=0.0, ge=-2.0, le=2.0)
+    max_tokens: Optional[int] = None
+
+
+class ResponseFormat(BaseModel):
+    """Specifies a JSON schema for the response format.
+
+    When provided, the LLM will be constrained to return a JSON response
+    that matches the specified schema.
+
+    Can be instantiated with:
+    - A JSON schema dictionary
+    - A Pydantic model class
+    - An existing ResponseFormat instance
+    """
+
+    json_schema: Dict[str, Any]
+
+    @classmethod
+    def from_value(cls, value: Any) -> "ResponseFormat":
+        if isinstance(value, cls):
+            return value
+        if isinstance(value, dict):
+            return cls(json_schema=value)
+        if isinstance(value, type) and issubclass(value, pydantic.BaseModel):
+            return cls(json_schema=value.model_json_schema())
+        raise ValueError(
+            "ResponseFormat configuration is invalid. Expected ResponseFormat, a valid schema or a Pydantic BaseModel."
+        )
+
+
 class RunRequest(BaseModel):
     """Request model for creating a run."""
 
     agent_id: str = Field(default="default_agent")
+    model_settings: ModelSettings = ModelSettings()
+    response_format: Optional[Union[ResponseFormat, Dict[str, Any], type]] = None
+    group: Optional[str] = None
+    metadata: Optional[Dict[str, str]] = None
 
 
 class RunResponse(BaseModel):
@@ -19,6 +68,7 @@ class RunResponse(BaseModel):
     run_id: str
     thread_id: str
     status: str
+    group: Optional[str] = None
 
 
 class RunStatus(str, Enum):

seekrai/types/agents/tools/__init__.py

@@ -1,7 +1,13 @@
 from seekrai.types.agents.tools.env_model_config import EnvConfig
-from seekrai.types.agents.tools.schemas import FileSearch, FileSearchEnv
+from seekrai.types.agents.tools.schemas import (
+    FileSearch,
+    FileSearchEnv,
+    RunPython,
+    RunPythonEnv,
+    WebSearch,
+    WebSearchEnv,
+)
 from seekrai.types.agents.tools.tool import ToolBase, ToolType
-from seekrai.types.agents.tools.tool_env_types import Env
 from seekrai.types.agents.tools.tool_types import Tool
 
 
@@ -13,4 +19,8 @@ __all__ = [
     "Tool",
     "FileSearch",
     "FileSearchEnv",
+    "RunPython",
+    "RunPythonEnv",
+    "WebSearch",
+    "WebSearchEnv",
 ]

seekrai/types/agents/tools/schemas/__init__.py

@@ -1,8 +1,16 @@
 from seekrai.types.agents.tools.schemas.file_search import FileSearch
 from seekrai.types.agents.tools.schemas.file_search_env import FileSearchEnv
+from seekrai.types.agents.tools.schemas.run_python import RunPython
+from seekrai.types.agents.tools.schemas.run_python_env import RunPythonEnv
+from seekrai.types.agents.tools.schemas.web_search import WebSearch
+from seekrai.types.agents.tools.schemas.web_search_env import WebSearchEnv
 
 
 __all__ = [
     "FileSearch",
     "FileSearchEnv",
+    "RunPython",
+    "RunPythonEnv",
+    "WebSearch",
+    "WebSearchEnv",
 ]

seekrai/types/agents/tools/schemas/file_search.py

@@ -4,6 +4,6 @@ from seekrai.types.agents.tools.schemas.file_search_env import FileSearchEnv
 from seekrai.types.agents.tools.tool import ToolBase, ToolType
 
 
-class FileSearch(ToolBase):
+class FileSearch(ToolBase[FileSearchEnv]):
     name: Literal[ToolType.FILE_SEARCH] = ToolType.FILE_SEARCH
     tool_env: FileSearchEnv