Skip to content

Commit 993c748

Browse files
feat(api): update via SDK Studio
1 parent ed4eda6 commit 993c748

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

55 files changed

+1201
-1877
lines changed

api.md

+26-56
Large diffs are not rendered by default.

src/contextual/resources/agents/__init__.py

-14
Original file line numberDiff line numberDiff line change
@@ -40,22 +40,8 @@
4040
EvaluateResourceWithStreamingResponse,
4141
AsyncEvaluateResourceWithStreamingResponse,
4242
)
43-
from .metadata import (
44-
MetadataResource,
45-
AsyncMetadataResource,
46-
MetadataResourceWithRawResponse,
47-
AsyncMetadataResourceWithRawResponse,
48-
MetadataResourceWithStreamingResponse,
49-
AsyncMetadataResourceWithStreamingResponse,
50-
)
5143

5244
__all__ = [
53-
"MetadataResource",
54-
"AsyncMetadataResource",
55-
"MetadataResourceWithRawResponse",
56-
"AsyncMetadataResourceWithRawResponse",
57-
"MetadataResourceWithStreamingResponse",
58-
"AsyncMetadataResourceWithStreamingResponse",
5945
"QueryResource",
6046
"AsyncQueryResource",
6147
"QueryResourceWithRawResponse",

src/contextual/resources/agents/agents.py

+83-32
Original file line numberDiff line numberDiff line change
@@ -20,14 +20,6 @@
2020
maybe_transform,
2121
async_maybe_transform,
2222
)
23-
from .metadata import (
24-
MetadataResource,
25-
AsyncMetadataResource,
26-
MetadataResourceWithRawResponse,
27-
AsyncMetadataResourceWithRawResponse,
28-
MetadataResourceWithStreamingResponse,
29-
AsyncMetadataResourceWithStreamingResponse,
30-
)
3123
from ..._compat import cached_property
3224
from .tune.tune import (
3325
TuneResource,
@@ -63,16 +55,13 @@
6355
EvaluateResourceWithStreamingResponse,
6456
AsyncEvaluateResourceWithStreamingResponse,
6557
)
58+
from ...types.agent_metadata import AgentMetadata
6659
from ...types.create_agent_output import CreateAgentOutput
6760

6861
__all__ = ["AgentsResource", "AsyncAgentsResource"]
6962

7063

7164
class AgentsResource(SyncAPIResource):
72-
@cached_property
73-
def metadata(self) -> MetadataResource:
74-
return MetadataResource(self._client)
75-
7665
@cached_property
7766
def query(self) -> QueryResource:
7867
return QueryResource(self._client)
@@ -331,12 +320,43 @@ def delete(
331320
cast_to=object,
332321
)
333322

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`.
334336
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
339341
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):
340360
@cached_property
341361
def query(self) -> AsyncQueryResource:
342362
return AsyncQueryResource(self._client)
@@ -595,6 +615,41 @@ async def delete(
595615
cast_to=object,
596616
)
597617

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+
598653

599654
class AgentsResourceWithRawResponse:
600655
def __init__(self, agents: AgentsResource) -> None:
@@ -612,10 +667,9 @@ def __init__(self, agents: AgentsResource) -> None:
612667
self.delete = to_raw_response_wrapper(
613668
agents.delete,
614669
)
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+
)
619673

620674
@cached_property
621675
def query(self) -> QueryResourceWithRawResponse:
@@ -650,10 +704,9 @@ def __init__(self, agents: AsyncAgentsResource) -> None:
650704
self.delete = async_to_raw_response_wrapper(
651705
agents.delete,
652706
)
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+
)
657710

658711
@cached_property
659712
def query(self) -> AsyncQueryResourceWithRawResponse:
@@ -688,10 +741,9 @@ def __init__(self, agents: AgentsResource) -> None:
688741
self.delete = to_streamed_response_wrapper(
689742
agents.delete,
690743
)
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+
)
695747

696748
@cached_property
697749
def query(self) -> QueryResourceWithStreamingResponse:
@@ -726,10 +778,9 @@ def __init__(self, agents: AsyncAgentsResource) -> None:
726778
self.delete = async_to_streamed_response_wrapper(
727779
agents.delete,
728780
)
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+
)
733784

734785
@cached_property
735786
def query(self) -> AsyncQueryResourceWithStreamingResponse:

src/contextual/resources/agents/datasets/evaluation.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
evaluation_metadata_params,
3939
evaluation_retrieve_params,
4040
)
41-
from ....types.agents.dataset_response import DatasetResponse
41+
from ....types.agents.dataset_metadata import DatasetMetadata
4242
from ....types.agents.list_datasets_response import ListDatasetsResponse
4343
from ....types.agents.create_dataset_response import CreateDatasetResponse
4444

@@ -393,7 +393,7 @@ def metadata(
393393
extra_query: Query | None = None,
394394
extra_body: Body | None = None,
395395
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
396-
) -> DatasetResponse:
396+
) -> DatasetMetadata:
397397
"""
398398
Retrieve details of a specific evaluation `Dataset` version, or the latest
399399
version if no `version` is specified.
@@ -429,7 +429,7 @@ def metadata(
429429
timeout=timeout,
430430
query=maybe_transform({"version": version}, evaluation_metadata_params.EvaluationMetadataParams),
431431
),
432-
cast_to=DatasetResponse,
432+
cast_to=DatasetMetadata,
433433
)
434434

435435

@@ -783,7 +783,7 @@ async def metadata(
783783
extra_query: Query | None = None,
784784
extra_body: Body | None = None,
785785
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
786-
) -> DatasetResponse:
786+
) -> DatasetMetadata:
787787
"""
788788
Retrieve details of a specific evaluation `Dataset` version, or the latest
789789
version if no `version` is specified.
@@ -821,7 +821,7 @@ async def metadata(
821821
{"version": version}, evaluation_metadata_params.EvaluationMetadataParams
822822
),
823823
),
824-
cast_to=DatasetResponse,
824+
cast_to=DatasetMetadata,
825825
)
826826

827827

src/contextual/resources/agents/datasets/tune.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
tune_metadata_params,
3939
tune_retrieve_params,
4040
)
41-
from ....types.agents.dataset_response import DatasetResponse
41+
from ....types.agents.dataset_metadata import DatasetMetadata
4242
from ....types.agents.list_datasets_response import ListDatasetsResponse
4343
from ....types.agents.create_dataset_response import CreateDatasetResponse
4444

@@ -388,7 +388,7 @@ def metadata(
388388
extra_query: Query | None = None,
389389
extra_body: Body | None = None,
390390
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
391-
) -> DatasetResponse:
391+
) -> DatasetMetadata:
392392
"""
393393
Retrieve details of a specific tuning `Dataset` version, or the latest version
394394
if no `version` is specified.
@@ -424,7 +424,7 @@ def metadata(
424424
timeout=timeout,
425425
query=maybe_transform({"version": version}, tune_metadata_params.TuneMetadataParams),
426426
),
427-
cast_to=DatasetResponse,
427+
cast_to=DatasetMetadata,
428428
)
429429

430430

@@ -771,7 +771,7 @@ async def metadata(
771771
extra_query: Query | None = None,
772772
extra_body: Body | None = None,
773773
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
774-
) -> DatasetResponse:
774+
) -> DatasetMetadata:
775775
"""
776776
Retrieve details of a specific tuning `Dataset` version, or the latest version
777777
if no `version` is specified.
@@ -807,7 +807,7 @@ async def metadata(
807807
timeout=timeout,
808808
query=await async_maybe_transform({"version": version}, tune_metadata_params.TuneMetadataParams),
809809
),
810-
cast_to=DatasetResponse,
810+
cast_to=DatasetMetadata,
811811
)
812812

813813

0 commit comments

Comments
 (0)