20
20
maybe_transform ,
21
21
async_maybe_transform ,
22
22
)
23
- from .metadata import (
24
- MetadataResource ,
25
- AsyncMetadataResource ,
26
- MetadataResourceWithRawResponse ,
27
- AsyncMetadataResourceWithRawResponse ,
28
- MetadataResourceWithStreamingResponse ,
29
- AsyncMetadataResourceWithStreamingResponse ,
30
- )
31
23
from ..._compat import cached_property
32
24
from .tune .tune import (
33
25
TuneResource ,
63
55
EvaluateResourceWithStreamingResponse ,
64
56
AsyncEvaluateResourceWithStreamingResponse ,
65
57
)
58
+ from ...types .agent_metadata import AgentMetadata
66
59
from ...types .create_agent_output import CreateAgentOutput
67
60
68
61
__all__ = ["AgentsResource" , "AsyncAgentsResource" ]
69
62
70
63
71
64
class AgentsResource (SyncAPIResource ):
72
- @cached_property
73
- def metadata (self ) -> MetadataResource :
74
- return MetadataResource (self ._client )
75
-
76
65
@cached_property
77
66
def query (self ) -> QueryResource :
78
67
return QueryResource (self ._client )
@@ -331,12 +320,43 @@ def delete(
331
320
cast_to = object ,
332
321
)
333
322
323
+ def metadata (
324
+ self ,
325
+ agent_id : str ,
326
+ * ,
327
+ # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
328
+ # The extra values given here take precedence over values defined on the client or passed to this method.
329
+ extra_headers : Headers | None = None ,
330
+ extra_query : Query | None = None ,
331
+ extra_body : Body | None = None ,
332
+ timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
333
+ ) -> AgentMetadata :
334
+ """
335
+ Get metadata and configuration of a given `Agent`.
334
336
335
- class AsyncAgentsResource ( AsyncAPIResource ) :
336
- @ cached_property
337
- def metadata ( self ) -> AsyncMetadataResource :
338
- return AsyncMetadataResource ( self . _client )
337
+ Args :
338
+ agent_id: Agent ID of the agent to retrieve details for
339
+
340
+ extra_headers: Send extra headers
339
341
342
+ extra_query: Add additional query parameters to the request
343
+
344
+ extra_body: Add additional JSON properties to the request
345
+
346
+ timeout: Override the client-level default timeout for this request, in seconds
347
+ """
348
+ if not agent_id :
349
+ raise ValueError (f"Expected a non-empty value for `agent_id` but received { agent_id !r} " )
350
+ return self ._get (
351
+ f"/agents/{ agent_id } /metadata" ,
352
+ options = make_request_options (
353
+ extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
354
+ ),
355
+ cast_to = AgentMetadata ,
356
+ )
357
+
358
+
359
+ class AsyncAgentsResource (AsyncAPIResource ):
340
360
@cached_property
341
361
def query (self ) -> AsyncQueryResource :
342
362
return AsyncQueryResource (self ._client )
@@ -595,6 +615,41 @@ async def delete(
595
615
cast_to = object ,
596
616
)
597
617
618
+ async def metadata (
619
+ self ,
620
+ agent_id : str ,
621
+ * ,
622
+ # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
623
+ # The extra values given here take precedence over values defined on the client or passed to this method.
624
+ extra_headers : Headers | None = None ,
625
+ extra_query : Query | None = None ,
626
+ extra_body : Body | None = None ,
627
+ timeout : float | httpx .Timeout | None | NotGiven = NOT_GIVEN ,
628
+ ) -> AgentMetadata :
629
+ """
630
+ Get metadata and configuration of a given `Agent`.
631
+
632
+ Args:
633
+ agent_id: Agent ID of the agent to retrieve details for
634
+
635
+ extra_headers: Send extra headers
636
+
637
+ extra_query: Add additional query parameters to the request
638
+
639
+ extra_body: Add additional JSON properties to the request
640
+
641
+ timeout: Override the client-level default timeout for this request, in seconds
642
+ """
643
+ if not agent_id :
644
+ raise ValueError (f"Expected a non-empty value for `agent_id` but received { agent_id !r} " )
645
+ return await self ._get (
646
+ f"/agents/{ agent_id } /metadata" ,
647
+ options = make_request_options (
648
+ extra_headers = extra_headers , extra_query = extra_query , extra_body = extra_body , timeout = timeout
649
+ ),
650
+ cast_to = AgentMetadata ,
651
+ )
652
+
598
653
599
654
class AgentsResourceWithRawResponse :
600
655
def __init__ (self , agents : AgentsResource ) -> None :
@@ -612,10 +667,9 @@ def __init__(self, agents: AgentsResource) -> None:
612
667
self .delete = to_raw_response_wrapper (
613
668
agents .delete ,
614
669
)
615
-
616
- @cached_property
617
- def metadata (self ) -> MetadataResourceWithRawResponse :
618
- return MetadataResourceWithRawResponse (self ._agents .metadata )
670
+ self .metadata = to_raw_response_wrapper (
671
+ agents .metadata ,
672
+ )
619
673
620
674
@cached_property
621
675
def query (self ) -> QueryResourceWithRawResponse :
@@ -650,10 +704,9 @@ def __init__(self, agents: AsyncAgentsResource) -> None:
650
704
self .delete = async_to_raw_response_wrapper (
651
705
agents .delete ,
652
706
)
653
-
654
- @cached_property
655
- def metadata (self ) -> AsyncMetadataResourceWithRawResponse :
656
- return AsyncMetadataResourceWithRawResponse (self ._agents .metadata )
707
+ self .metadata = async_to_raw_response_wrapper (
708
+ agents .metadata ,
709
+ )
657
710
658
711
@cached_property
659
712
def query (self ) -> AsyncQueryResourceWithRawResponse :
@@ -688,10 +741,9 @@ def __init__(self, agents: AgentsResource) -> None:
688
741
self .delete = to_streamed_response_wrapper (
689
742
agents .delete ,
690
743
)
691
-
692
- @cached_property
693
- def metadata (self ) -> MetadataResourceWithStreamingResponse :
694
- return MetadataResourceWithStreamingResponse (self ._agents .metadata )
744
+ self .metadata = to_streamed_response_wrapper (
745
+ agents .metadata ,
746
+ )
695
747
696
748
@cached_property
697
749
def query (self ) -> QueryResourceWithStreamingResponse :
@@ -726,10 +778,9 @@ def __init__(self, agents: AsyncAgentsResource) -> None:
726
778
self .delete = async_to_streamed_response_wrapper (
727
779
agents .delete ,
728
780
)
729
-
730
- @cached_property
731
- def metadata (self ) -> AsyncMetadataResourceWithStreamingResponse :
732
- return AsyncMetadataResourceWithStreamingResponse (self ._agents .metadata )
781
+ self .metadata = async_to_streamed_response_wrapper (
782
+ agents .metadata ,
783
+ )
733
784
734
785
@cached_property
735
786
def query (self ) -> AsyncQueryResourceWithStreamingResponse :
0 commit comments