seekrai 0.5.14__tar.gz → 0.5.16__tar.gz

{seekrai-0.5.14 → seekrai-0.5.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: seekrai
-Version: 0.5.14
+Version: 0.5.16
 Summary: Python client for SeekrAI
 License: Apache-2.0
 Author: SeekrFlow

{seekrai-0.5.14 → seekrai-0.5.16}/pyproject.toml

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

{seekrai-0.5.14 → seekrai-0.5.16}/src/seekrai/client.py

@@ -24,6 +24,7 @@ class SeekrFlow:
     deployments: resources.Deployments
     vector_database: resources.VectorDatabase
     agents: resources.Agents
+    observability: resources.AgentObservability
 
     # client options
     client: SeekrFlowClient
@@ -89,6 +90,7 @@ class SeekrFlow:
         self.deployments = resources.Deployments(self.client)
         self.vector_database = resources.VectorDatabase(self.client)
         self.agents = resources.Agents(self.client)
+        self.observability = resources.AgentObservability(self.client)
 
 
 class AsyncSeekrFlow:
@@ -105,6 +107,7 @@ class AsyncSeekrFlow:
     deployments: resources.AsyncDeployments
     vector_database: resources.AsyncVectorDatabase
     agents: resources.AsyncAgents
+    observability: resources.AsyncAgentObservability
 
     # client options
     client: SeekrFlowClient
@@ -170,6 +173,7 @@ class AsyncSeekrFlow:
         self.deployments = resources.AsyncDeployments(self.client)
         self.vector_database = resources.AsyncVectorDatabase(self.client)
         self.agents = resources.AsyncAgents(self.client)
+        self.observability = resources.AsyncAgentObservability(self.client)
 
 
 Client = SeekrFlow

{seekrai-0.5.14 → seekrai-0.5.16}/src/seekrai/resources/__init__.py

@@ -1,4 +1,10 @@
-from seekrai.resources.agents import AgentInference, Agents, AsyncAgents
+from seekrai.resources.agents import (
+    AgentInference,
+    AgentObservability,
+    Agents,
+    AsyncAgentObservability,
+    AsyncAgents,
+)
 from seekrai.resources.alignment import Alignment, AsyncAlignment
 from seekrai.resources.chat import AsyncChat, Chat
 from seekrai.resources.completions import AsyncCompletions, Completions
@@ -39,6 +45,8 @@ __all__ = [
     "Deployments",
     "AsyncAgents",
     "Agents",
+    "AgentObservability",
+    "AsyncAgentObservability",
     "VectorDatabase",
     "AsyncVectorDatabase",
     "AgentInference",

{seekrai-0.5.14 → seekrai-0.5.16}/src/seekrai/resources/agents/__init__.py

@@ -1,4 +1,8 @@
 from seekrai.resources.agents.agent_inference import AgentInference, AsyncAgentInference
+from seekrai.resources.agents.agent_observability import (
+    AgentObservability,
+    AsyncAgentObservability,
+)
 from seekrai.resources.agents.agents import Agents, AsyncAgents
 from seekrai.resources.agents.python_functions import (
     AsyncCustomFunctions,
@@ -16,4 +20,6 @@ __all__ = [
     "AsyncAgentThreads",
     "CustomFunctions",
     "AsyncCustomFunctions",
+    "AgentObservability",
+    "AsyncAgentObservability",
 ]

{seekrai-0.5.14 → seekrai-0.5.16}/src/seekrai/resources/agents/agent_inference.py

@@ -19,6 +19,8 @@ class AgentInference:
         stream: bool = False,
         model_settings: ModelSettings = ModelSettings(),
         response_format: Optional[Any] = None,
+        group: Optional[str] = "default_group",
+        metadata: Optional[dict[str, str]] = None,
     ) -> Union[RunResponse, Iterator[Any]]:
         """
         Run an inference call on a deployed agent.
@@ -29,6 +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 '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.
@@ -39,6 +43,8 @@ class AgentInference:
             response_format=ResponseFormat.from_value(response_format)
             if response_format
             else None,
+            group=group,
+            metadata=metadata,
         ).model_dump()
         endpoint = f"threads/{thread_id}/runs"
         if stream:
@@ -156,6 +162,8 @@ class AsyncAgentInference:
         stream: bool = False,
         model_settings: ModelSettings = ModelSettings(),
         response_format: Optional[Any] = 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.
@@ -166,6 +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 '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.
@@ -176,6 +186,8 @@ class AsyncAgentInference:
             response_format=ResponseFormat.from_value(response_format)
             if response_format
             else None,
+            group=group,
+            metadata=metadata,
         ).model_dump()
         endpoint = f"threads/{thread_id}/runs"
         if stream:

seekrai-0.5.16/src/seekrai/resources/agents/agent_observability.py

@@ -0,0 +1,135 @@
+from datetime import datetime
+from typing import Any, Optional
+
+from seekrai.abstract import api_requestor
+from seekrai.seekrflow_response import SeekrFlowResponse
+from seekrai.types import SeekrFlowRequest
+from seekrai.types.agents.observability import (
+    ObservabilitySpansRequest,
+    ObservabilitySpansResponse,
+)
+
+
+BASE_OBSERVABILITY_ENDPOINT = "observability/spans"
+
+
+class AgentObservability:
+    def __init__(self, client: Any):
+        self._client = client
+        self._requestor = api_requestor.APIRequestor(client=self._client)
+
+    def query_spans(
+        self,
+        min_start_time: Optional[datetime] = None,
+        max_start_time: Optional[datetime] = None,
+        agent_id: Optional[str] = None,
+        run_id: Optional[str] = None,
+        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,
+    ) -> ObservabilitySpansResponse:
+        """
+        Retrieve spans for a given run or group of runs given a set of facets.
+        """
+        payload = ObservabilitySpansRequest(
+            min_start_datetime=min_start_time,
+            max_start_datetime=max_start_time,
+            agent_id=agent_id,
+            run_id=run_id,
+            trace_id=trace_id,
+            thread_id=thread_id,
+            group=group,
+            metadata=metadata,
+            limit=limit,
+            order=order,
+            offset=offset,
+        ).model_dump()
+
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(
+                method="POST", url=BASE_OBSERVABILITY_ENDPOINT, params=payload
+            )
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+
+        return ObservabilitySpansResponse(spans=response.data)
+
+    def retrieve_span(self, span_id: str) -> Optional[dict[str, Any]]:
+        """
+        Retrieve a specific span given a span_id.
+        """
+        endpoint = f"{BASE_OBSERVABILITY_ENDPOINT}/{span_id}"
+
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(method="GET", url=endpoint)
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+
+        return response.data
+
+
+class AsyncAgentObservability:
+    def __init__(self, client: Any) -> None:
+        self._client = client
+        self._requestor = api_requestor.APIRequestor(client=self._client)
+
+    async def query_spans(
+        self,
+        min_start_time: Optional[datetime] = None,
+        max_start_time: Optional[datetime] = None,
+        agent_id: Optional[str] = None,
+        run_id: Optional[str] = None,
+        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,
+    ) -> ObservabilitySpansResponse:
+        """
+        Retrieve spans for a given run or group of runs given a set of facets.
+        """
+        payload = ObservabilitySpansRequest(
+            min_start_datetime=min_start_time,
+            max_start_datetime=max_start_time,
+            agent_id=agent_id,
+            run_id=run_id,
+            trace_id=trace_id,
+            thread_id=thread_id,
+            group=group,
+            metadata=metadata,
+            limit=limit,
+            order=order,
+            offset=offset,
+        ).model_dump()
+
+        response, _, _ = await self._requestor.arequest(
+            options=SeekrFlowRequest(
+                method="POST", url=BASE_OBSERVABILITY_ENDPOINT, params=payload
+            )
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+
+        return ObservabilitySpansResponse(spans=response.data)
+
+    async def retrieve_span(self, span_id: str) -> Optional[dict[str, Any]]:
+        """
+        Retrieve a specific span given a span_id.
+        """
+        endpoint = f"{BASE_OBSERVABILITY_ENDPOINT}/{span_id}"
+
+        response, _, _ = self._requestor.request(
+            options=SeekrFlowRequest(method="GET", url=endpoint)
+        )
+
+        assert isinstance(response, SeekrFlowResponse)
+
+        return response.data

{seekrai-0.5.14 → seekrai-0.5.16}/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.14 → seekrai-0.5.16}/src/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):

seekrai-0.5.16/src/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-0.5.14 → seekrai-0.5.16}/src/seekrai/types/agents/runs.py

@@ -58,6 +58,8 @@ class RunRequest(BaseModel):
     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):
@@ -66,6 +68,7 @@ class RunResponse(BaseModel):
     run_id: str
     thread_id: str
     status: str
+    group: Optional[str] = None
 
 
 class RunStatus(str, Enum):

{seekrai-0.5.14 → seekrai-0.5.16}/src/seekrai/types/finetune.py

@@ -26,6 +26,7 @@ class FinetuneJobStatus(str, Enum):
     STATUS_CANCELLED = "cancelled"
     STATUS_FAILED = "failed"
     STATUS_COMPLETED = "completed"
+    STATUS_DELETED = "deleted"
 
 
 class FinetuneEventLevels(str, Enum):
@@ -182,6 +183,7 @@ class FinetuneResponse(BaseModel):
     experiment_name: str | None = None
     # job status
     status: FinetuneJobStatus | None = None
+    deleted_at: datetime | None = None
 
     # list of fine-tune events
     events: List[FinetuneEvent] | None = None