feat(axon): accept user metadata at creation

Stainless-Generated-From: 8365e99960676cdb8b7040e31597aefd6c9e8f39

Author: stlc-bot

scripts/mock

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Optional
+from typing import Dict, Optional
 from typing_extensions import Literal
 
 import httpx
@@ -75,6 +75,7 @@ def with_streaming_response(self) -> AxonsResourceWithStreamingResponse:
     def create(
         self,
         *,
+        metadata: Optional[Dict[str, str]] | Omit = omit,
         name: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -88,6 +89,8 @@ def create(
         [Beta] Create a new axon.
 
         Args:
+          metadata: User defined metadata to attach to the axon for organization.
+
           name: (Optional) Name for the axon.
 
           extra_headers: Send extra headers
@@ -102,7 +105,13 @@ def create(
         """
         return self._post(
             "/v1/axons",
-            body=maybe_transform({"name": name}, axon_create_params.AxonCreateParams),
+            body=maybe_transform(
+                {
+                    "metadata": metadata,
+                    "name": name,
+                },
+                axon_create_params.AxonCreateParams,
+            ),
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
@@ -390,6 +399,7 @@ def with_streaming_response(self) -> AsyncAxonsResourceWithStreamingResponse:
     async def create(
         self,
         *,
+        metadata: Optional[Dict[str, str]] | Omit = omit,
         name: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
@@ -403,6 +413,8 @@ async def create(
         [Beta] Create a new axon.
 
         Args:
+          metadata: User defined metadata to attach to the axon for organization.
+
           name: (Optional) Name for the axon.
 
           extra_headers: Send extra headers
@@ -417,7 +429,13 @@ async def create(
         """
         return await self._post(
             "/v1/axons",
-            body=await async_maybe_transform({"name": name}, axon_create_params.AxonCreateParams),
+            body=await async_maybe_transform(
+                {
+                    "metadata": metadata,
+                    "name": name,
+                },
+                axon_create_params.AxonCreateParams,
+            ),
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,

src/runloop_api_client/resources/axons/axons.py

@@ -2,12 +2,15 @@
 
 from __future__ import annotations
 
-from typing import Optional
+from typing import Dict, Optional
 from typing_extensions import TypedDict
 
 __all__ = ["AxonCreateParams"]
 
 
 class AxonCreateParams(TypedDict, total=False):
+    metadata: Optional[Dict[str, str]]
+    """User defined metadata to attach to the axon for organization."""
+
     name: Optional[str]
     """(Optional) Name for the axon."""

src/runloop_api_client/types/axon_create_params.py

@@ -1,6 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import Optional
+from typing import Dict, Optional
 
 from .._models import BaseModel
 
@@ -14,5 +14,8 @@ class AxonView(BaseModel):
     created_at_ms: int
     """Creation time in milliseconds since epoch."""
 
+    metadata: Dict[str, str]
+    """The user defined axon metadata."""
+
     name: Optional[str] = None
     """The name of the axon."""

src/runloop_api_client/types/axon_view.py

@@ -29,6 +29,7 @@ def test_method_create(self, client: Runloop) -> None:
     @parametrize
     def test_method_create_with_all_params(self, client: Runloop) -> None:
         axon = client.axons.create(
+            metadata={"foo": "string"},
             name="name",
         )
         assert_matches_type(AxonView, axon, path=["response"])
@@ -279,6 +280,7 @@ async def test_method_create(self, async_client: AsyncRunloop) -> None:
     @parametrize
     async def test_method_create_with_all_params(self, async_client: AsyncRunloop) -> None:
         axon = await async_client.axons.create(
+            metadata={"foo": "string"},
             name="name",
         )
         assert_matches_type(AxonView, axon, path=["response"])