diff --git a/src/truefoundry_sdk/_wrapped_clients.py b/src/truefoundry_sdk/_wrapped_clients.py index 5b1b4f56..16a39598 100644 --- a/src/truefoundry_sdk/_wrapped_clients.py +++ b/src/truefoundry_sdk/_wrapped_clients.py @@ -1,7 +1,5 @@ from typing import Any, Optional, Protocol, TypeVar -from .agent_versions.client import AgentVersionsClient, AsyncAgentVersionsClient -from .agents.client import AgentsClient, AsyncAgentsClient from .applications.client import ApplicationsClient, AsyncApplicationsClient from .artifact_versions.client import ArtifactVersionsClient, AsyncArtifactVersionsClient from .artifacts.client import ArtifactsClient, AsyncArtifactsClient @@ -15,11 +13,6 @@ from .prompt_versions.client import AsyncPromptVersionsClient, PromptVersionsClient from .prompts.client import AsyncPromptsClient, PromptsClient from .secret_groups.client import AsyncSecretGroupsClient, SecretGroupsClient -from .tool_versions.client import AsyncToolVersionsClient, ToolVersionsClient -from .tools.client import AsyncToolsClient, ToolsClient -from .tracing_projects.client import AsyncTracingProjectsClient, TracingProjectsClient -from .types.get_agent_response import GetAgentResponse -from .types.get_agent_version_response import GetAgentVersionResponse from .types.get_application_response import GetApplicationResponse from .types.get_artifact_response import GetArtifactResponse from .types.get_artifact_version_response import GetArtifactVersionResponse @@ -29,9 +22,6 @@ from .types.get_prompt_response import GetPromptResponse from .types.get_prompt_version_response import GetPromptVersionResponse from .types.get_secret_group_response import GetSecretGroupResponse -from .types.get_tool_response import GetToolResponse -from .types.get_tool_version_response import GetToolVersionResponse -from .types.get_tracing_project_response import GetTracingProjectResponse from .types.get_workspace_response import GetWorkspaceResponse from .types.http_error import HttpError from .workspaces.client import AsyncWorkspacesClient, WorkspacesClient @@ -82,118 +72,422 @@ async def _aget_by_fqn( return result -class WrappedAgentsClient(AgentsClient): - def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetAgentResponse: - item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetAgentResponse, {"data": item}) # type: ignore[arg-type,var-annotated] +class WrappedApplicationsClient(ApplicationsClient): + def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetApplicationResponse: + """ + Get Application by FQN. + Parameters + ---------- + fqn : str + FQN of the application -class WrappedAgentVersionsClient(AgentVersionsClient): - def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetAgentVersionResponse: - item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetAgentVersionResponse, {"data": item}) + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + Returns + ------- + GetApplicationResponse + Application details -class WrappedApplicationsClient(ApplicationsClient): - def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetApplicationResponse: + Examples + -------- + from truefoundry_sdk import TrueFoundry + + client = TrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + client.applications.get_by_fqn( + fqn="fqn", + ) + """ item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetApplicationResponse, {"data": item}) class WrappedArtifactsClient(ArtifactsClient): def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetArtifactResponse: + """ + Get Artifact by FQN. + + Parameters + ---------- + fqn : str + FQN of the artifact + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetArtifactResponse + Artifact details + + Examples + -------- + from truefoundry_sdk import TrueFoundry + + client = TrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + client.artifacts.get_by_fqn( + fqn="fqn", + ) + """ item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetArtifactResponse, {"data": item}) class WrappedArtifactVersionsClient(ArtifactVersionsClient): def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetArtifactVersionResponse: + """ + Get Artifact Version by FQN. + + Parameters + ---------- + fqn : str + FQN of the artifact version + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetArtifactVersionResponse + Artifact version details + + Examples + -------- + from truefoundry_sdk import TrueFoundry + + client = TrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + client.artifact_versions.get_by_fqn( + fqn="fqn", + ) + """ item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetArtifactVersionResponse, {"data": item}) class WrappedDataDirectoriesClient(DataDirectoriesClient): def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetDataDirectoryResponse: + """ + Get Data Directory by FQN. + + Parameters + ---------- + fqn : str + FQN of the data directory + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetDataDirectoryResponse + Data directory details + + Examples + -------- + from truefoundry_sdk import TrueFoundry + + client = TrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + client.data_directories.get_by_fqn( + fqn="fqn", + ) + """ item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetDataDirectoryResponse, {"data": item}) class WrappedModelsClient(ModelsClient): def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetModelResponse: + """ + Get Model by FQN. + + Parameters + ---------- + fqn : str + FQN of the model + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetModelResponse + Model details + + Examples + -------- + from truefoundry_sdk import TrueFoundry + + client = TrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + client.models.get_by_fqn( + fqn="fqn", + ) + """ item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetModelResponse, {"data": item}) class WrappedModelVersionsClient(ModelVersionsClient): def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetModelVersionResponse: + """ + Get Model Version by FQN. + + Parameters + ---------- + fqn : str + FQN of the model version + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetModelVersionResponse + Model version details + + Examples + -------- + from truefoundry_sdk import TrueFoundry + + client = TrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + client.model_versions.get_by_fqn( + fqn="fqn", + ) + """ item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetModelVersionResponse, {"data": item}) class WrappedPromptsClient(PromptsClient): def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetPromptResponse: + """ + Get Prompt by FQN. + + Parameters + ---------- + fqn : str + FQN of the prompt + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetPromptResponse + Prompt details + + Examples + -------- + from truefoundry_sdk import TrueFoundry + + client = TrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + client.prompts.get_by_fqn( + fqn="fqn", + ) + """ item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetPromptResponse, {"data": item}) class WrappedPromptVersionsClient(PromptVersionsClient): def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetPromptVersionResponse: + """ + Get Prompt Version by FQN. + + Parameters + ---------- + fqn : str + FQN of the prompt version + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetPromptVersionResponse + Prompt version details + + Examples + -------- + from truefoundry_sdk import TrueFoundry + + client = TrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + client.prompt_versions.get_by_fqn( + fqn="fqn", + ) + """ item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetPromptVersionResponse, {"data": item}) class WrappedSecretGroupsClient(SecretGroupsClient): def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetSecretGroupResponse: - item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetSecretGroupResponse, {"data": item}) + """ + Get Secret Group by FQN. + Parameters + ---------- + fqn : str + FQN of the secret group -class WrappedToolsClient(ToolsClient): - def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetToolResponse: - item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetToolResponse, {"data": item}) + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + Returns + ------- + GetSecretGroupResponse + Secret group details -class WrappedToolVersionsClient(ToolVersionsClient): - def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetToolVersionResponse: - item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetToolVersionResponse, {"data": item}) + Examples + -------- + from truefoundry_sdk import TrueFoundry - -class WrappedTracingProjectsClient(TracingProjectsClient): - def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetTracingProjectResponse: + client = TrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + client.secret_groups.get_by_fqn( + fqn="fqn", + ) + """ item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetTracingProjectResponse, {"data": item}) + return parse_obj_as(GetSecretGroupResponse, {"data": item}) class WrappedWorkspacesClient(WorkspacesClient): def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetWorkspaceResponse: - item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetWorkspaceResponse, {"data": item}) + """ + Get Workspace by FQN. + Parameters + ---------- + fqn : str + FQN of the workspace -class WrappedAsyncAgentsClient(AsyncAgentsClient): - async def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetAgentResponse: - item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetAgentResponse, {"data": item}) + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + Returns + ------- + GetWorkspaceResponse + Workspace details -class WrappedAsyncAgentVersionsClient(AsyncAgentVersionsClient): - async def get_by_fqn( - self, fqn: str, *, request_options: Optional[RequestOptions] = None - ) -> GetAgentVersionResponse: - item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetAgentVersionResponse, {"data": item}) + Examples + -------- + from truefoundry_sdk import TrueFoundry + + client = TrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + client.workspaces.get_by_fqn( + fqn="fqn", + ) + """ + item = _get_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] + return parse_obj_as(GetWorkspaceResponse, {"data": item}) class WrappedAsyncApplicationsClient(AsyncApplicationsClient): async def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetApplicationResponse: + """ + Get Application by FQN. + + Parameters + ---------- + fqn : str + FQN of the application + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetApplicationResponse + Application details + + Examples + -------- + import asyncio + + from truefoundry_sdk import AsyncTrueFoundry + + client = AsyncTrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + + async def main() -> None: + await client.applications.get_by_fqn( + fqn="fqn", + ) + + asyncio.run(main()) + """ item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetApplicationResponse, {"data": item}) class WrappedAsyncArtifactsClient(AsyncArtifactsClient): async def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetArtifactResponse: + """ + Get Artifact by FQN. + + Parameters + ---------- + fqn : str + FQN of the artifact + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetArtifactResponse + Artifact details + + Examples + -------- + import asyncio + + from truefoundry_sdk import AsyncTrueFoundry + + client = AsyncTrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + + async def main() -> None: + await client.artifacts.get_by_fqn( + fqn="fqn", + ) + + asyncio.run(main()) + """ item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetArtifactResponse, {"data": item}) @@ -202,6 +496,40 @@ class WrappedAsyncArtifactVersionsClient(AsyncArtifactVersionsClient): async def get_by_fqn( self, fqn: str, *, request_options: Optional[RequestOptions] = None ) -> GetArtifactVersionResponse: + """ + Get Artifact Version by FQN. + + Parameters + ---------- + fqn : str + FQN of the artifact version + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetArtifactVersionResponse + Artifact version details + + Examples + -------- + import asyncio + + from truefoundry_sdk import AsyncTrueFoundry + + client = AsyncTrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + + async def main() -> None: + await client.artifact_versions.get_by_fqn( + fqn="fqn", + ) + + asyncio.run(main()) + """ item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetArtifactVersionResponse, {"data": item}) @@ -210,12 +538,80 @@ class WrappedAsyncDataDirectoriesClient(AsyncDataDirectoriesClient): async def get_by_fqn( self, fqn: str, *, request_options: Optional[RequestOptions] = None ) -> GetDataDirectoryResponse: + """ + Get Data Directory by FQN. + + Parameters + ---------- + fqn : str + FQN of the data directory + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetDataDirectoryResponse + Data directory details + + Examples + -------- + import asyncio + + from truefoundry_sdk import AsyncTrueFoundry + + client = AsyncTrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + + async def main() -> None: + await client.data_directories.get_by_fqn( + fqn="fqn", + ) + + asyncio.run(main()) + """ item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetDataDirectoryResponse, {"data": item}) class WrappedAsyncModelsClient(AsyncModelsClient): async def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetModelResponse: + """ + Get Model by FQN. + + Parameters + ---------- + fqn : str + FQN of the model + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetModelResponse + Model details + + Examples + -------- + import asyncio + + from truefoundry_sdk import AsyncTrueFoundry + + client = AsyncTrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + + async def main() -> None: + await client.models.get_by_fqn( + fqn="fqn", + ) + + asyncio.run(main()) + """ item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetModelResponse, {"data": item}) @@ -224,12 +620,80 @@ class WrappedAsyncModelVersionsClient(AsyncModelVersionsClient): async def get_by_fqn( self, fqn: str, *, request_options: Optional[RequestOptions] = None ) -> GetModelVersionResponse: + """ + Get Model Version by FQN. + + Parameters + ---------- + fqn : str + FQN of the model version + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetModelVersionResponse + Model version details + + Examples + -------- + import asyncio + + from truefoundry_sdk import AsyncTrueFoundry + + client = AsyncTrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + + async def main() -> None: + await client.model_versions.get_by_fqn( + fqn="fqn", + ) + + asyncio.run(main()) + """ item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetModelVersionResponse, {"data": item}) class WrappedAsyncPromptsClient(AsyncPromptsClient): async def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetPromptResponse: + """ + Get Prompt by FQN. + + Parameters + ---------- + fqn : str + FQN of the prompt + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetPromptResponse + Prompt details + + Examples + -------- + import asyncio + + from truefoundry_sdk import AsyncTrueFoundry + + client = AsyncTrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + + async def main() -> None: + await client.prompts.get_by_fqn( + fqn="fqn", + ) + + asyncio.run(main()) + """ item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetPromptResponse, {"data": item}) @@ -238,37 +702,119 @@ class WrappedAsyncPromptVersionsClient(AsyncPromptVersionsClient): async def get_by_fqn( self, fqn: str, *, request_options: Optional[RequestOptions] = None ) -> GetPromptVersionResponse: + """ + Get Prompt Version by FQN. + + Parameters + ---------- + fqn : str + FQN of the prompt version + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetPromptVersionResponse + Prompt version details + + Examples + -------- + import asyncio + + from truefoundry_sdk import AsyncTrueFoundry + + client = AsyncTrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + + async def main() -> None: + await client.prompt_versions.get_by_fqn( + fqn="fqn", + ) + + asyncio.run(main()) + """ item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetPromptVersionResponse, {"data": item}) class WrappedAsyncSecretGroupsClient(AsyncSecretGroupsClient): async def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetSecretGroupResponse: - item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetSecretGroupResponse, {"data": item}) + """ + Get Secret Group by FQN. + Parameters + ---------- + fqn : str + FQN of the secret group -class WrappedAsyncToolsClient(AsyncToolsClient): - async def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetToolResponse: - item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetToolResponse, {"data": item}) + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + Returns + ------- + GetSecretGroupResponse + Secret group details -class WrappedAsyncToolVersionsClient(AsyncToolVersionsClient): - async def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetToolVersionResponse: - item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetToolVersionResponse, {"data": item}) + Examples + -------- + import asyncio + from truefoundry_sdk import AsyncTrueFoundry -class WrappedAsyncTracingProjectsClient(AsyncTracingProjectsClient): - async def get_by_fqn( - self, fqn: str, *, request_options: Optional[RequestOptions] = None - ) -> GetTracingProjectResponse: + client = AsyncTrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + + async def main() -> None: + await client.secret_groups.get_by_fqn( + fqn="fqn", + ) + + asyncio.run(main()) + """ item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] - return parse_obj_as(GetTracingProjectResponse, {"data": item}) + return parse_obj_as(GetSecretGroupResponse, {"data": item}) class WrappedAsyncWorkspacesClient(AsyncWorkspacesClient): async def get_by_fqn(self, fqn: str, *, request_options: Optional[RequestOptions] = None) -> GetWorkspaceResponse: + """ + Get Workspace by FQN. + + Parameters + ---------- + fqn : str + FQN of the workspace + + request_options : typing.Optional[RequestOptions] + Request-specific configuration. + + Returns + ------- + GetWorkspaceResponse + Workspace details + + Examples + -------- + import asyncio + + from truefoundry_sdk import AsyncTrueFoundry + + client = AsyncTrueFoundry( + api_key="YOUR_API_KEY", + base_url="https://yourhost.com/path/to/api", + ) + + async def main() -> None: + await client.workspaces.get_by_fqn( + fqn="fqn", + ) + + asyncio.run(main()) + """ item = await _aget_by_fqn(self, fqn=fqn, request_options=request_options) # type: ignore[arg-type,var-annotated] return parse_obj_as(GetWorkspaceResponse, {"data": item}) diff --git a/src/truefoundry_sdk/client.py b/src/truefoundry_sdk/client.py index aed7fe37..30498230 100644 --- a/src/truefoundry_sdk/client.py +++ b/src/truefoundry_sdk/client.py @@ -4,13 +4,9 @@ import httpx from truefoundry_sdk._wrapped_clients import ( - WrappedAgentsClient, - WrappedAgentVersionsClient, WrappedApplicationsClient, WrappedArtifactsClient, WrappedArtifactVersionsClient, - WrappedAsyncAgentsClient, - WrappedAsyncAgentVersionsClient, WrappedAsyncApplicationsClient, WrappedAsyncArtifactsClient, WrappedAsyncArtifactVersionsClient, @@ -20,9 +16,6 @@ WrappedAsyncPromptsClient, WrappedAsyncPromptVersionsClient, WrappedAsyncSecretGroupsClient, - WrappedAsyncToolsClient, - WrappedAsyncToolVersionsClient, - WrappedAsyncTracingProjectsClient, WrappedAsyncWorkspacesClient, WrappedDataDirectoriesClient, WrappedModelsClient, @@ -30,9 +23,6 @@ WrappedPromptsClient, WrappedPromptVersionsClient, WrappedSecretGroupsClient, - WrappedToolsClient, - WrappedToolVersionsClient, - WrappedTracingProjectsClient, WrappedWorkspacesClient, ) from truefoundry_sdk.base_client import AsyncBaseTrueFoundry, BaseTrueFoundry @@ -55,8 +45,6 @@ def __init__( follow_redirects=follow_redirects, httpx_client=httpx_client, ) - self.agents = WrappedAgentsClient(client_wrapper=self._client_wrapper) - self.agent_versions = WrappedAgentVersionsClient(client_wrapper=self._client_wrapper) self.applications = WrappedApplicationsClient(client_wrapper=self._client_wrapper) self.artifacts = WrappedArtifactsClient(client_wrapper=self._client_wrapper) self.artifact_versions = WrappedArtifactVersionsClient(client_wrapper=self._client_wrapper) @@ -66,9 +54,6 @@ def __init__( self.prompts = WrappedPromptsClient(client_wrapper=self._client_wrapper) self.prompt_versions = WrappedPromptVersionsClient(client_wrapper=self._client_wrapper) self.secret_groups = WrappedSecretGroupsClient(client_wrapper=self._client_wrapper) - self.tools = WrappedToolsClient(client_wrapper=self._client_wrapper) - self.tool_versions = WrappedToolVersionsClient(client_wrapper=self._client_wrapper) - self.tracing_projects = WrappedTracingProjectsClient(client_wrapper=self._client_wrapper) self.workspaces = WrappedWorkspacesClient(client_wrapper=self._client_wrapper) @@ -89,8 +74,6 @@ def __init__( follow_redirects=follow_redirects, httpx_client=httpx_client, ) - self.agents = WrappedAsyncAgentsClient(client_wrapper=self._client_wrapper) - self.agent_versions = WrappedAsyncAgentVersionsClient(client_wrapper=self._client_wrapper) self.applications = WrappedAsyncApplicationsClient(client_wrapper=self._client_wrapper) self.artifacts = WrappedAsyncArtifactsClient(client_wrapper=self._client_wrapper) self.artifact_versions = WrappedAsyncArtifactVersionsClient(client_wrapper=self._client_wrapper) @@ -100,9 +83,6 @@ def __init__( self.prompts = WrappedAsyncPromptsClient(client_wrapper=self._client_wrapper) self.prompt_versions = WrappedAsyncPromptVersionsClient(client_wrapper=self._client_wrapper) self.secret_groups = WrappedAsyncSecretGroupsClient(client_wrapper=self._client_wrapper) - self.tools = WrappedAsyncToolsClient(client_wrapper=self._client_wrapper) - self.tool_versions = WrappedAsyncToolVersionsClient(client_wrapper=self._client_wrapper) - self.tracing_projects = WrappedAsyncTracingProjectsClient(client_wrapper=self._client_wrapper) self.workspaces = WrappedAsyncWorkspacesClient(client_wrapper=self._client_wrapper)