seekrai 0.5.15__py3-none-any.whl → 0.5.16__py3-none-any.whl

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/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/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/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/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.15.dist-info → seekrai-0.5.16.dist-info}/METADATA

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

{seekrai-0.5.15.dist-info → seekrai-0.5.16.dist-info}/RECORD

@@ -8,10 +8,10 @@ seekrai/error.py,sha256=rAYL8qEd8INwYMMKvhS-HKeC3QkWL4Wq-zfazFU-zBg,4861
 seekrai/filemanager.py,sha256=bO2OvjZ9Cx5r2vr3Ocpd_0qVc3owRDT2LCU4Zmp2uDY,15489
 seekrai/resources/__init__.py,sha256=x39Gx6-ErQHNxhDB0CplXURk3kx-4OGpxm3r6HJr7Ds,1628
 seekrai/resources/agents/__init__.py,sha256=8zQIr2LEJyuaVwbU6WmItGu0F8VWTa7eYrquyHUY1QY,700
-seekrai/resources/agents/agent_inference.py,sha256=si8BzMV2p6_TCTIT7jd4j9ae4p4HwFlxCWUpo-J1d_g,10146
-seekrai/resources/agents/agent_observability.py,sha256=CnHK948NwHYpr_FBDzvmq8xxU6NFx0FowM6CbN32o1k,4089
+seekrai/resources/agents/agent_inference.py,sha256=KIxdEV9qMA7tscrVGp4-Fe1H_iBqdY7Oh7rkpPDNfjo,10568
+seekrai/resources/agents/agent_observability.py,sha256=VC_iyJnTwZ6PdQPp0T9XzipXOruh8P7h_UCwLjbY1Sk,4253
 seekrai/resources/agents/agents.py,sha256=frmLHvrqeUxnMkE9Xn4w1ugChM14S9ik4MwWyklPMWI,9607
-seekrai/resources/agents/python_functions.py,sha256=2GSiWhKTto8ixE5OG6WsN-GWEwyf1wpdhG3_EziACsg,10367
+seekrai/resources/agents/python_functions.py,sha256=VL1JSsPP5nN1m8I0Ihe3AYlb8TMRg2n9-FWsWeNZH2s,9277
 seekrai/resources/agents/threads.py,sha256=BwZ2_6wlezsb12PQjEw1fgdJh5S83SPgD6qZQoGvyIM,14544
 seekrai/resources/alignment.py,sha256=IOKlKK2I9_NhS9pwcrsd9-5OO7lVT8Uw0y_wuGHOnyA,5839
 seekrai/resources/chat/__init__.py,sha256=KmtPupgECtEN80NyvcnSmieTAFXhwmVxhMHP0qhspA4,618
@@ -33,9 +33,9 @@ seekrai/types/__init__.py,sha256=N72VsnmOdvLxnhBFLgfugIkT0tqr-M7WVFpAFtXEjLI,466
 seekrai/types/abstract.py,sha256=TqWFQV_6bPblywfCH-r8FCkXWvPkc9KlJ4QVgyrnaMc,642
 seekrai/types/agents/__init__.py,sha256=STRQlgnapxFT5egAP_i3yZSh9gauGmJBHTzdvH8tbdk,2395
 seekrai/types/agents/agent.py,sha256=85D4GeHF-bYYnPirJSi1MbFg_2uFE2fSEmAHV9LxZfQ,1132
-seekrai/types/agents/observability.py,sha256=E4FSHJEoA7t2rirZQw9UE267Z8vNtKgs4niOc0RVOqg,924
+seekrai/types/agents/observability.py,sha256=kOpBrNsrOzSbf-AsRBWAxRB9FqO0063CshUr-r2hHBE,963
 seekrai/types/agents/python_functions.py,sha256=31Jm46gryHsgNsC7nlivuyY0TSko58y2YVxsu_7bEAg,653
-seekrai/types/agents/runs.py,sha256=qjlPet4mr-ZPXF_jumDPNa7Pe-uwMCSUJSG0jQ1ZLxc,4887
+seekrai/types/agents/runs.py,sha256=JgvasNM_eesGbwnqeAbRzPNvbzLAG1_N2IU7M7OHD1w,4933
 seekrai/types/agents/threads.py,sha256=TinCMKv1bi5LzboDyCx1XI4Zzd8UzUZos4VOrTNhmEc,6835
 seekrai/types/agents/tools/__init__.py,sha256=4MmlL13JLhWgMUlL3TKfegiA-IXGG06YellZTSTVFC8,537
 seekrai/types/agents/tools/env_model_config.py,sha256=9POx2DPwfSXgoaziJv7QvKeMrhMsYD1exnanSRK48vw,177
@@ -69,8 +69,8 @@ seekrai/utils/api_helpers.py,sha256=0Y8BblNIr9h_R12zdmhkxgTlxgoRkbq84QNi4nNWGu8,
 seekrai/utils/files.py,sha256=7ixn_hgV-6pEhYqLyOp-EN0o8c1CzUwJzX9n3PQ5oqo,7164
 seekrai/utils/tools.py,sha256=jgJTL-dOIouDbEJLdQpQfpXhqaz_poQYS52adyUtBjo,1781
 seekrai/version.py,sha256=q6iGQVFor8zXiPP5F-3vy9TndOxKv5JXbaNJ2kdOQws,125
-seekrai-0.5.15.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-seekrai-0.5.15.dist-info/METADATA,sha256=q2xaCaeQH1Nywj_P_aoQrxfwz2AjgLk3zPmjCGciWow,4781
-seekrai-0.5.15.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-seekrai-0.5.15.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
-seekrai-0.5.15.dist-info/RECORD,,
+seekrai-0.5.16.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+seekrai-0.5.16.dist-info/METADATA,sha256=Qwkh2LCsf_1wzdjQ8k6RRk4Iu0qqowufPTlnIVvc-KY,4781
+seekrai-0.5.16.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+seekrai-0.5.16.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
+seekrai-0.5.16.dist-info/RECORD,,