seekrai 0.5.15__tar.gz → 0.5.17__tar.gz

{seekrai-0.5.15 → seekrai-0.5.17}/PKG-INFO

@@ -1,7 +1,8 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.1
 Name: seekrai
-Version: 0.5.15
+Version: 0.5.17
 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
@@ -12,8 +13,6 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: click (>=8.1.7,<9.0.0)
 Requires-Dist: eval-type-backport (>=0.1.3,<0.3.0)
 Requires-Dist: filelock (>=3.13.1,<4.0.0)

{seekrai-0.5.15 → seekrai-0.5.17}/pyproject.toml

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

{seekrai-0.5.15 → seekrai-0.5.17}/src/seekrai/client.py

@@ -25,6 +25,7 @@ class SeekrFlow:
     vector_database: resources.VectorDatabase
     agents: resources.Agents
     observability: resources.AgentObservability
+    explainability: resources.Explainability
 
     # client options
     client: SeekrFlowClient
@@ -91,6 +92,7 @@ class SeekrFlow:
         self.vector_database = resources.VectorDatabase(self.client)
         self.agents = resources.Agents(self.client)
         self.observability = resources.AgentObservability(self.client)
+        self.explainability = resources.Explainability(self.client)
 
 
 class AsyncSeekrFlow:
@@ -108,6 +110,7 @@ class AsyncSeekrFlow:
     vector_database: resources.AsyncVectorDatabase
     agents: resources.AsyncAgents
     observability: resources.AsyncAgentObservability
+    explainability: resources.AsyncExplainability
 
     # client options
     client: SeekrFlowClient
@@ -174,6 +177,7 @@ class AsyncSeekrFlow:
         self.vector_database = resources.AsyncVectorDatabase(self.client)
         self.agents = resources.AsyncAgents(self.client)
         self.observability = resources.AsyncAgentObservability(self.client)
+        self.explainability = resources.AsyncExplainability(self.client)
 
 
 Client = SeekrFlow

{seekrai-0.5.15 → seekrai-0.5.17}/src/seekrai/resources/agents/agent_inference.py

@@ -19,7 +19,8 @@ class AgentInference:
         stream: bool = False,
         model_settings: ModelSettings = ModelSettings(),
         response_format: Optional[Any] = None,
-        group: Optional[str] = None,
+        group: Optional[str] = "default_group",
+        metadata: Optional[dict[str, str]] = None,
     ) -> Union[RunResponse, Iterator[Any]]:
         """
         Run an inference call on a deployed agent.
@@ -30,7 +31,8 @@ class AgentInference:
             stream (bool, optional): Whether to stream the response. Defaults to False.
             model_settings (optional): Additional parameters (such as temperature, max_tokens, etc).
             response_format: Optional structured output specification. If provided, the LLM will be constrained to return JSON matching this schema.
-            group (str, optional): Label used to associate a group of runs. Defaults to None.
+            group (str, optional): Label used to associate a group of runs. Defaults to 'default_group'.
+            metadata (dict[str, str], optional): Additional metadata used to label runs. Defaults to None.
 
         Returns:
             A dictionary with the response (if non-streaming) or an iterator over response chunks.
@@ -42,6 +44,7 @@ class AgentInference:
             if response_format
             else None,
             group=group,
+            metadata=metadata,
         ).model_dump()
         endpoint = f"threads/{thread_id}/runs"
         if stream:
@@ -159,7 +162,8 @@ class AsyncAgentInference:
         stream: bool = False,
         model_settings: ModelSettings = ModelSettings(),
         response_format: Optional[Any] = None,
-        group: Optional[str] = None,
+        group: Optional[str] = "default_group",
+        metadata: Optional[dict[str, str]] = None,
     ) -> Union[RunResponse, AsyncGenerator[Any, None]]:
         """
         Run an inference call on a deployed agent.
@@ -170,7 +174,8 @@ class AsyncAgentInference:
             stream (bool, optional): Whether to stream the response. Defaults to False.
             model_settings (optional): Additional parameters (such as temperature, max_tokens, etc).
             response_format: Optional structured output specification. If provided, the LLM will be constrained to return JSON matching this schema.
-            group (str, optional): Label used to associate a group of runs. Defaults to None.
+            group (str, optional): Label used to associate a group of runs. Defaults to 'default_group'.
+            metadata (dict[str, str], optional): Additional metadata used to label runs. Defaults to None.
 
         Returns:
             A dictionary with the response (if non-streaming) or an iterator over response chunks.
@@ -182,6 +187,7 @@ class AsyncAgentInference:
             if response_format
             else None,
             group=group,
+            metadata=metadata,
         ).model_dump()
         endpoint = f"threads/{thread_id}/runs"
         if stream:

{seekrai-0.5.15 → seekrai-0.5.17}/src/seekrai/resources/agents/agent_observability.py

@@ -27,6 +27,7 @@ class AgentObservability:
         trace_id: Optional[str] = None,
         thread_id: Optional[str] = None,
         group: Optional[str] = None,
+        metadata: Optional[dict[str, str]] = None,
         limit: int = 100,
         order: str = "desc",
         offset: int = 0,
@@ -42,6 +43,7 @@ class AgentObservability:
             trace_id=trace_id,
             thread_id=thread_id,
             group=group,
+            metadata=metadata,
             limit=limit,
             order=order,
             offset=offset,
@@ -86,6 +88,7 @@ class AsyncAgentObservability:
         trace_id: Optional[str] = None,
         thread_id: Optional[str] = None,
         group: Optional[str] = None,
+        metadata: Optional[dict[str, str]] = None,
         limit: int = 100,
         order: str = "desc",
         offset: int = 0,
@@ -101,6 +104,7 @@ class AsyncAgentObservability:
             trace_id=trace_id,
             thread_id=thread_id,
             group=group,
+            metadata=metadata,
             limit=limit,
             order=order,
             offset=offset,

{seekrai-0.5.15 → seekrai-0.5.17}/src/seekrai/resources/agents/python_functions.py

@@ -17,15 +17,12 @@ class CustomFunctions:
             client=self._client,
         )
 
-    def create(
-        self, file_path: Union[str, Path], description: Union[str, None] = None
-    ) -> PythonFunctionResponse:
+    def create(self, file_path: Union[str, Path]) -> PythonFunctionResponse:
         """
         Upload a new Python function for the user.
 
         Args:
             file_path: Path to the Python function file to upload (can be relative or absolute).
-            description: Optional description for the function.
 
         Returns:
             The newly created Python function.
@@ -40,16 +37,12 @@ class CustomFunctions:
 
         # Prepare multipart form data
         files = {"file": (file_path.name, file_content, "text/plain")}
-        params = {}
-        if description:
-            params["description"] = description
 
         response, _, _ = self._requestor.request(
             options=SeekrFlowRequest(
                 method="POST",
                 url="functions/",
                 files=files,
-                params=params,
             ),
         )
 
@@ -103,10 +96,7 @@ class CustomFunctions:
         return functions
 
     def update(
-        self,
-        function_id: str,
-        file_path: Union[str, Path, None] = None,
-        description: Union[str, None] = None,
+        self, function_id: str, file_path: Union[str, Path]
     ) -> PythonFunctionResponse:
         """
         Update an existing Python function.
@@ -119,30 +109,22 @@ class CustomFunctions:
         Returns:
             The updated Python function.
         """
-        files = None
-        params = {}
-
-        if file_path:
-            # Convert string to Path if needed
-            if isinstance(file_path, str):
-                file_path = Path(file_path)
-
-            # Read the file contents
-            with file_path.open("rb") as f:
-                file_content = f.read()
+        # Convert string to Path if needed
+        if isinstance(file_path, str):
+            file_path = Path(file_path)
 
-            # Prepare multipart form data
-            files = {"file": (file_path.name, file_content, "text/plain")}
+        # Read the file contents
+        with file_path.open("rb") as f:
+            file_content = f.read()
 
-        if description:
-            params["description"] = description
+        # Prepare multipart form data
+        files = {"file": (file_path.name, file_content, "text/plain")}
 
         response, _, _ = self._requestor.request(
             options=SeekrFlowRequest(
                 method="PATCH",
                 url=f"functions/{function_id}",
                 files=files,
-                params=params,
             ),
         )
 
@@ -177,15 +159,12 @@ class AsyncCustomFunctions:
             client=self._client,
         )
 
-    async def create(
-        self, file_path: Union[str, Path], description: Union[str, None] = None
-    ) -> PythonFunctionResponse:
+    async def create(self, file_path: Union[str, Path]) -> PythonFunctionResponse:
         """
         Upload a new Python function for the user.
 
         Args:
             file_path: Path to the Python function file to upload (can be relative or absolute).
-            description: Optional description for the function.
 
         Returns:
             The newly created Python function.
@@ -200,16 +179,12 @@ class AsyncCustomFunctions:
 
         # Prepare multipart form data
         files = {"file": (file_path.name, file_content, "text/plain")}
-        params = {}
-        if description:
-            params["description"] = description
 
         response, _, _ = await self._requestor.arequest(
             options=SeekrFlowRequest(
                 method="POST",
                 url="functions/",
                 files=files,
-                params=params,
             ),
         )
 
@@ -265,44 +240,34 @@ class AsyncCustomFunctions:
     async def update(
         self,
         function_id: str,
-        file_path: Union[str, Path, None] = None,
-        description: Union[str, None] = None,
+        file_path: Union[str, Path],
     ) -> PythonFunctionResponse:
         """
         Update an existing Python function.
 
         Args:
             function_id: The ID of the Python function to update.
-            file_path: Optional path to a new Python function file (can be relative or absolute).
-            description: Optional new description for the function.
+            file_path: Path to a new Python function file (can be relative or absolute).
 
         Returns:
             The updated Python function.
         """
-        files = None
-        params = {}
-
-        if file_path:
-            # Convert string to Path if needed
-            if isinstance(file_path, str):
-                file_path = Path(file_path)
-
-            # Read the file contents
-            with file_path.open("rb") as f:
-                file_content = f.read()
+        # Convert string to Path if needed
+        if isinstance(file_path, str):
+            file_path = Path(file_path)
 
-            # Prepare multipart form data
-            files = {"file": (file_path.name, file_content, "text/plain")}
+        # Read the file contents
+        with file_path.open("rb") as f:
+            file_content = f.read()
 
-        if description:
-            params["description"] = description
+        # Prepare multipart form data
+        files = {"file": (file_path.name, file_content, "text/plain")}
 
         response, _, _ = await self._requestor.arequest(
             options=SeekrFlowRequest(
                 method="PATCH",
                 url=f"functions/{function_id}",
                 files=files,
-                params=params,
             ),
         )
 

{seekrai-0.5.15 → seekrai-0.5.17}/src/seekrai/resources/explainability.py

@@ -2,7 +2,6 @@ from typing import Optional
 
 from seekrai.abstract import api_requestor
 from seekrai.resources.resource_base import ResourceBase
-from seekrai.seekrflow_response import SeekrFlowResponse
 from seekrai.types import (
     SeekrFlowRequest,
 )
@@ -14,15 +13,19 @@ from seekrai.types.explainability import (
 
 class Explainability(ResourceBase):
     def get_influential_finetuning_data(
-        self, model_id: str, question: str, answer: Optional[str], k: int = 5
+        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,
-            - k (int): the number of results to be retrieved (5 by default)
         Returns:
             InfluentialFinetuningDataResponse: Object containing the influential fine tuning data.
         """
@@ -31,7 +34,7 @@ class Explainability(ResourceBase):
         )
         # Create query parameters dictionary
         parameter_payload = InfluentialFinetuningDataRequest(
-            question=question, answer=answer, k=k
+            question=question, system_prompt=system_prompt, answer=answer
         ).model_dump()
 
         # if limit is not None:
@@ -41,26 +44,29 @@ class Explainability(ResourceBase):
         response, _, _ = requestor.request(
             options=SeekrFlowRequest(
                 method="GET",
-                url=f"v1/flow/explain/models/{model_id}/influential-finetuning-data",
+                url=f"flow/explain/models/{model_id}/influential-finetuning-data",
                 params=parameter_payload,
             ),
             stream=False,
         )
-        assert isinstance(response, SeekrFlowResponse)
         return InfluentialFinetuningDataResponse(**response.data)
 
 
 class AsyncExplainability(ResourceBase):
     async def get_influential_finetuning_data(
-        self, model_id: str, question: str, answer: Optional[str], k: int = 5
+        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,
-            - k (int): the number of results to be retrieved (5 by default),
         Returns:
             InfluentialFinetuningDataResponse: Object containing the influential finetuning data.
         """
@@ -69,16 +75,18 @@ class AsyncExplainability(ResourceBase):
         )
         # Create query parameters dictionary
         parameter_payload = InfluentialFinetuningDataRequest(
-            model_id=model_id, question=question, answer=answer, k=k
+            model_id=model_id,
+            question=question,
+            system_prompt=system_prompt,
+            answer=answer,
         ).model_dump()
 
         response, _, _ = await requestor.arequest(
             options=SeekrFlowRequest(
                 method="GET",
-                url=f"v1/flow/explain/models/{model_id}/influential-finetuning-data",
+                url=f"flow/explain/models/{model_id}/influential-finetuning-data",
                 params=parameter_payload,
             ),
             stream=False,
         )
-        assert isinstance(response, SeekrFlowResponse)
         return InfluentialFinetuningDataResponse(**response.data)

{seekrai-0.5.15 → seekrai-0.5.17}/src/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-0.5.15 → seekrai-0.5.17}/src/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()
@@ -338,7 +340,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 +363,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()

{seekrai-0.5.15 → seekrai-0.5.17}/src/seekrai/types/agents/observability.py

@@ -16,6 +16,7 @@ class ObservabilitySpansRequest(BaseModel):
     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

{seekrai-0.5.15 → seekrai-0.5.17}/src/seekrai/types/agents/runs.py

@@ -59,6 +59,7 @@ class RunRequest(BaseModel):
     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):

seekrai-0.5.17/src/seekrai/types/explainability.py

@@ -0,0 +1,26 @@
+from __future__ import annotations
+
+from typing import Any, Dict, List, Optional
+
+from pydantic import Field
+
+from seekrai.types.abstract import BaseModel
+
+
+class InfluentialFinetuningDataResponse(BaseModel):
+    results: List[Dict[str, Any]] = Field(
+        ..., description="List of influential training data results"
+    )
+    version: str = Field(..., description="Version of the explainability service")
+
+
+class InfluentialFinetuningDataRequest(BaseModel):
+    question: str = Field(..., description="Question from user")
+    system_prompt: Optional[str] = Field(
+        None,
+        description="System prompt for the user's question.",
+    )
+    answer: Optional[str] = Field(
+        None,
+        description="Answer of the finetuned model to the question; if None, the answer is retrieved from the finetuned model",
+    )

{seekrai-0.5.15 → seekrai-0.5.17}/src/seekrai/types/vectordb.py

@@ -37,7 +37,12 @@ class VectorDatabaseIngestionRequest(BaseModel):
     """Request model for creating a new vector database ingestion job."""
 
     file_ids: List[str] = Field(..., description="List of file IDs to ingest")
-    method: str = Field(..., description="Method to use for ingestion")
+    method: Optional[str] = Field(
+        default="accuracy-optimized", description="Method to use for ingestion"
+    )
+    chunking_method: Optional[str] = Field(
+        default="markdown", description="Configure how your content will be segmented"
+    )
     token_count: int = Field(default=800, description="Token count for ingestion")
     overlap_tokens: int = Field(default=100, description="Overlap tokens for ingestion")
 

seekrai-0.5.15/src/seekrai/types/explainability.py

@@ -1,57 +0,0 @@
-from __future__ import annotations
-
-from datetime import datetime
-from enum import Enum
-from typing import Any, Dict, List, Literal
-
-from pydantic import Field
-
-from seekrai.types.abstract import BaseModel
-
-
-class InfluentialFinetuningDataResponse(BaseModel):
-    results: List[Dict[str, Any]]
-    version: str
-
-
-class InfluentialFinetuningDataRequest(BaseModel):
-    question: str
-    answer: str = Field(
-        default="",
-        description="Response could be generated or given",
-    )
-    k: int
-
-
-class ExplainabilityJobStatus(Enum):
-    QUEUED = "queued"
-    RUNNING = "running"
-    COMPLETED = "completed"
-    FAILED = "failed"
-
-    # TODO should titles along the following get added:
-    # create_index
-    # populate_index
-    # delete_index
-    # influential-finetuning-data
-
-
-class ExplainabilityRequest(BaseModel):
-    files: List[str] = Field(
-        default=..., description="List of file ids to use for fine tuning"
-    )
-    method: str = Field(default="best", description="Method to use for explainability")
-
-
-class ExplainabilityResponse(BaseModel):
-    id: str = Field(default=..., description="Explainability job ID")
-    created_at: datetime
-    status: ExplainabilityJobStatus
-    output_files: List[str]
-
-
-class ExplainabilityList(BaseModel):
-    # object type
-    object: Literal["list"] | None = None
-    # list of fine-tune job objects
-    data: List[ExplainabilityResponse] | None = None