adaptive-sdk 0.1.6__tar.gz → 0.1.7__tar.gz

{adaptive_sdk-0.1.6 → adaptive_sdk-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: adaptive-sdk
-Version: 0.1.6
+Version: 0.1.7
 Summary: Python SDK for Adaptive Engine
 Author-email: Vincent Debergue <vincent@adaptive-ml.com>, Joao Moura <joao@adaptive-ml.com>, Yacine Bouraoui <yacine@adaptive-ml.com>
 Requires-Python: >=3.10

{adaptive_sdk-0.1.6 → adaptive_sdk-0.1.7}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "adaptive-sdk"
-version = "0.1.6"
+version = "0.1.7"
 description = "Python SDK for Adaptive Engine"
 authors = [
   { name = "Vincent Debergue", email = "vincent@adaptive-ml.com" },
@@ -76,3 +76,5 @@ allow_redefinition = true
 [tool.black]
 required-version = '25.1.0'
 
+[tool.uv]
+constraint-dependencies = ["graphql-core==3.2.6"]

{adaptive_sdk-0.1.6 → adaptive_sdk-0.1.7}/src/adaptive_sdk/base_client.py

@@ -42,10 +42,16 @@ class BaseSyncClient:
         base_url: str,
         api_key: Optional[str] = None,
         default_headers: Dict[str, str] | None = None,
-        timeout_secs: float = 90.0,
+        timeout_secs: float | None = 90.0,
     ):
         """
         Construct a new synchronous client instance.
+
+        Args:
+            base_url: The base URL for the API
+            api_key: API key for authentication (or set ADAPTIVE_API_KEY env var)
+            default_headers: Additional headers to include in requests
+            timeout_secs: Timeout in seconds for requests (None for no timeout)
         """
         if not base_url:
             raise ValueError("base_url must be provided")
@@ -68,7 +74,7 @@ class BaseSyncClient:
         self._gql_client = GQLClient(base_url + Routes.GQL.value, headers=headers)
         self._gql_client.get_data = graphql_multi_error_handler(self._gql_client.get_data)  # type: ignore[method-assign]
 
-        timeout = httpx.Timeout(timeout=timeout_secs)
+        timeout = httpx.Timeout(timeout=timeout_secs) if timeout_secs is not None else None
 
         self._gql_client.http_client.timeout = timeout
         self._rest_client = httpx.Client(headers=headers, base_url=base_url + Routes.REST.value, timeout=timeout)
@@ -84,10 +90,16 @@ class BaseAsyncClient:
         base_url: str,
         api_key: Optional[str] = None,
         default_headers: Dict[str, str] | None = None,
-        timeout_secs: float = 90.0,
+        timeout_secs: float | None = 90.0,
     ):
         """
         Construct a new asynchronous client instance.
+
+        Args:
+            base_url: The base URL for the API
+            api_key: API key for authentication (or set ADAPTIVE_API_KEY env var)
+            default_headers: Additional headers to include in requests
+            timeout_secs: Timeout in seconds for requests (None for no timeout)
         """
         if not base_url:
             raise ValueError("base_url must be provided")
@@ -111,7 +123,7 @@ class BaseAsyncClient:
         self._gql_client = AsyncGQLClient(base_url + Routes.GQL.value, headers=headers)
         self._gql_client.get_data = graphql_multi_error_handler(self._gql_client.get_data)  # type: ignore[method-assign]
 
-        timeout = httpx.Timeout(timeout=timeout_secs)
+        timeout = httpx.Timeout(timeout=timeout_secs) if timeout_secs is not None else None
         self._gql_client.http_client.timeout = timeout
         self._rest_client = httpx.AsyncClient(headers=headers, base_url=base_url + Routes.REST.value, timeout=timeout)
 

{adaptive_sdk-0.1.6 → adaptive_sdk-0.1.7}/src/adaptive_sdk/client.py

@@ -9,6 +9,7 @@ class Adaptive(BaseSyncClient, UseCaseClient):
         base_url: str,
         api_key: str | None = None,
         default_headers: Dict[str, str] | None = None,
+        timeout_secs: float | None = 90.0,
     ) -> None:
         """
         Instantiates a new synchronous Adaptive client bounded to a use case.
@@ -18,9 +19,11 @@ class Adaptive(BaseSyncClient, UseCaseClient):
             base_url (str): The base URL for the Adaptive API.
             api_key (str, optional): API key for authentication.
                 Defaults to None, in which case environment variable `ADAPTIVE_API_KEY` needs to be set.
+            timeout_secs (float | None, optional): Timeout in seconds for HTTP requests.
+                Defaults to 90.0 seconds. Set to None for no timeout.
 
         """
-        super().__init__(base_url, api_key, default_headers)
+        super().__init__(base_url, api_key, default_headers, timeout_secs)
         self.__use_case_key = None
 
         self.ab_tests: resources.ABTests = resources.ABTests(self)
@@ -62,6 +65,7 @@ class AsyncAdaptive(BaseAsyncClient, UseCaseClient):
         base_url: str,
         api_key: str | None = None,
         default_headers: Dict[str, str] | None = None,
+        timeout_secs: float | None = 90.0,
     ) -> None:
         """
         Instantiates a new asynchronous Adaptive client bounded to a use case.
@@ -71,9 +75,11 @@ class AsyncAdaptive(BaseAsyncClient, UseCaseClient):
             base_url (str): The base URL for the Adaptive API.
             api_key (str, optional): API key for authentication.
                 Defaults to None, in which case environment variable `ADAPTIVE_API_KEY` needs to be set.
+            timeout_secs (float | None, optional): Timeout in seconds for HTTP requests.
+                Defaults to 90.0 seconds. Set to None for no timeout.
 
         """
-        super().__init__(base_url, api_key, default_headers)
+        super().__init__(base_url, api_key, default_headers, timeout_secs)
         self.__use_case_key = None
 
         self.ab_tests: resources.AsyncABTests = resources.AsyncABTests(self)

{adaptive_sdk-0.1.6 → adaptive_sdk-0.1.7}/src/adaptive_sdk/external/reward_client.py

@@ -60,10 +60,18 @@ class RewardClient:
         self, base_url: str, max_connections: int = 32, timeout: float | None = None
     ):
         headers = dict()  # type: ignore[var-annotated]
+        if base_url.startswith("https://"):
+            self.use_secure_protocol = True
+            self.base_url = base_url.removeprefix("https://")
+        else:
+            assert base_url.startswith("http://"), f"Unknown url format {base_url}"
+            self.use_secure_protocol = False
+            self.base_url = base_url.removeprefix("http://")
+
         self.base_url = base_url.removeprefix("http://").removeprefix("https://")
         self._client = httpx.AsyncClient(
             headers=dict(),
-            base_url=f"http://{self.base_url}",
+            base_url=self._get_http_url(),
             timeout=timeout,
             limits=Limits(max_connections=max_connections),
         )
@@ -71,13 +79,25 @@ class RewardClient:
         self.use_websocket = False
         self.max_connections = max_connections
 
+    def _get_http_url(self):
+        if self.use_secure_protocol:
+            return f"https://{self.base_url}"
+        else:
+            return f"http://{self.base_url}"
+
+    def _get_ws_url(self):
+        if self.use_secure_protocol:
+            return f"wss://{self.base_url}/ws"
+        else:
+            return f"ws://{self.base_url}/ws"
+
     async def setup(self):
         await self.connect_websocket()
 
     async def connect_websocket(self) -> Self:
         self.use_websocket = True
         self.ws_client: ClientConnection = await connect(
-            f"ws://{self.base_url}/ws", ping_timeout=None
+            self._get_ws_url(), ping_timeout=None
         )
         self.ws_responses: dict[int, Response | asyncio.Event] = dict()
         logger.info("Spawning_read_task")

adaptive_sdk-0.1.7/src/adaptive_sdk/graphql_client/__init__.py

@@ -0,0 +1,81 @@
+from .async_base_client_open_telemetry import AsyncBaseClientOpenTelemetry
+from .async_client import AsyncGQLClient
+from .add_external_model import AddExternalModel, AddExternalModelAddExternalModel
+from .add_hf_model import AddHFModel, AddHFModelImportHfModel
+from .add_model import AddModel, AddModelAddModel, AddModelAddModelBackbone
+from .add_remote_env import AddRemoteEnv, AddRemoteEnvAddRemoteEnv
+from .attach_model_to_use_case import AttachModelToUseCase, AttachModelToUseCaseAttachModel
+from .base_client_open_telemetry import BaseClientOpenTelemetry
+from .base_model import BaseModel, Upload
+from .cancel_ab_campaign import CancelABCampaign
+from .cancel_job import CancelJob, CancelJobCancelJob
+from .client import GQLClient
+from .create_ab_campaign import CreateAbCampaign, CreateAbCampaignCreateAbCampaign
+from .create_custom_recipe import CreateCustomRecipe, CreateCustomRecipeCreateCustomRecipe
+from .create_grader import CreateGrader, CreateGraderCreateGrader
+from .create_job import CreateJob, CreateJobCreateJob
+from .create_judge import CreateJudge, CreateJudgeCreateJudge
+from .create_metric import CreateMetric, CreateMetricCreateMetric
+from .create_prebuilt_judge import CreatePrebuiltJudge, CreatePrebuiltJudgeCreatePrebuiltJudge
+from .create_role import CreateRole, CreateRoleCreateRole
+from .create_team import CreateTeam, CreateTeamCreateTeam
+from .create_use_case import CreateUseCase, CreateUseCaseCreateUseCase
+from .create_user import CreateUser, CreateUserCreateUser
+from .delete_custom_recipe import DeleteCustomRecipe
+from .delete_dataset import DeleteDataset
+from .delete_grader import DeleteGrader, DeleteGraderDeleteGrader
+from .delete_judge import DeleteJudge, DeleteJudgeDeleteJudge
+from .delete_user import DeleteUser, DeleteUserDeleteUser
+from .deploy_model import DeployModel
+from .describe_ab_campaign import DescribeAbCampaign, DescribeAbCampaignAbCampaign, DescribeAbCampaignAbCampaignReport
+from .describe_dataset import DescribeDataset, DescribeDatasetDataset
+from .describe_interaction import DescribeInteraction, DescribeInteractionCompletion
+from .describe_job import DescribeJob, DescribeJobJob
+from .describe_metric import DescribeMetric, DescribeMetricMetric
+from .describe_metric_admin import DescribeMetricAdmin, DescribeMetricAdminMetric
+from .describe_model import DescribeModel, DescribeModelModel, DescribeModelModelBackbone
+from .describe_model_admin import DescribeModelAdmin, DescribeModelAdminModel, DescribeModelAdminModelBackbone
+from .describe_use_case import DescribeUseCase, DescribeUseCaseUseCase
+from .enums import AbcampaignStatus, AuthProviderKind, BillingCycle, CompletionGroupBy, CompletionSource, ComputePoolCapability, DatasetKind, DatasetSource, DatasetStatus, DateBucketUnit, ExternalModelProviderName, FeedbackType, GraderTypeEnum, HarmonyStatus, JobArtifactKind, JobKind, JobStatus, JobStatusOutput, JudgeCapability, MetricAggregation, MetricKind, MetricScoringType, ModelCapabilityFilter, ModelOnline, ModelserviceStatus, OpenAIModel, PrebuiltCriteriaKey, Protocol, ProviderName, RemoteEnvStatus, SelectionTypeInput, SessionStatus, SortDirection, TimeseriesInterval, UnitPosition
+from .exceptions import GraphQLClientError, GraphQLClientGraphQLError, GraphQLClientGraphQLMultiError, GraphQLClientHttpError, GraphQLClientInvalidResponseError
+from .fragments import AbCampaignCreateData, AbCampaignDetailData, AbCampaignDetailDataMetric, AbCampaignDetailDataModels, AbCampaignDetailDataUseCase, AbCampaignReportData, AbCampaignReportDataVariants, AbCampaignReportDataVariantsComparisons, AbCampaignReportDataVariantsComparisonsVariant, AbCampaignReportDataVariantsInterval, AbCampaignReportDataVariantsVariant, CompletionComparisonFeedbackData, CompletionComparisonFeedbackDataModel, CompletionData, CompletionDataChatMessages, CompletionDataComparisonFeedbacks, CompletionDataComparisonFeedbacksMetric, CompletionDataComparisonFeedbacksOtherCompletion, CompletionDataComparisonFeedbacksPreferedCompletion, CompletionDataComparisonFeedbacksUsecase, CompletionDataDirectFeedbacks, CompletionDataDirectFeedbacksMetric, CompletionDataLabels, CompletionDataMetadata, CompletionDataMetadataUsage, CompletionDataModel, CustomRecipeData, CustomRecipeDataCreatedBy, CustomRecipeDataLabels, DatasetData, DatasetDataMetricsUsage, DatasetDataMetricsUsageMetric, GraderData, GraderDataGraderConfigCustomConfigOutput, GraderDataGraderConfigJudgeConfigOutput, GraderDataGraderConfigJudgeConfigOutputExamples, GraderDataGraderConfigJudgeConfigOutputExamplesInput, GraderDataGraderConfigJudgeConfigOutputModel, GraderDataGraderConfigPrebuiltConfigOutput, GraderDataGraderConfigPrebuiltConfigOutputModel, GraderDataGraderConfigPrebuiltConfigOutputPrebuiltCriteria, GraderDataGraderConfigRemoteConfigOutput, GraderDataMetric, GraderDataUseCase, HarmonyGroupData, HarmonyGroupDataComputePool, HarmonyGroupDataOnlineModels, JobData, JobDataCreatedBy, JobDataDetails, JobDataDetailsArtifacts, JobDataRecipe, JobDataStages, JobDataStagesInfoBatchInferenceJobStageOutput, JobDataStagesInfoEvalJobStageOutput, JobDataStagesInfoTrainingJobStageOutput, JobDataUseCase, JobStageOutputData, JobStageOutputDataInfoBatchInferenceJobStageOutput, JobStageOutputDataInfoEvalJobStageOutput, JobStageOutputDataInfoTrainingJobStageOutput, JudgeData, JudgeDataExamples, JudgeDataExamplesInput, JudgeDataMetric, JudgeDataModel, MetricData, MetricDataAdmin, MetricDataAdminUseCases, MetricWithContextData, ModelData, ModelDataAdmin, ModelDataAdminUseCases, ModelDataComputeConfig, ModelServiceData, ModelServiceDataModel, ModelServiceDataModelBackbone, RemoteEnvData, UseCaseData, UseCaseDataMetrics, UseCaseDataModelServices, UseCaseDataShares, UseCaseDataSharesRole, UseCaseDataSharesTeam, UserData, UserDataTeams, UserDataTeamsRole, UserDataTeamsTeam
+from .get_custom_recipe import GetCustomRecipe, GetCustomRecipeCustomRecipe
+from .get_grader import GetGrader, GetGraderGrader
+from .get_judge import GetJudge, GetJudgeJudge
+from .input_types import AbcampaignCreate, AbCampaignFilter, AddExternalModelInput, AddHFModelInput, AddModelInput, AnthropicProviderDataInput, ApiKeyCreate, ArtifactFilter, AttachModel, AzureProviderDataInput, CancelAllocationInput, CapabilityFilter, CompletionComparisonFilterInput, CompletionFeedbackFilterInput, CompletionLabelValue, CompletionsByFilters, CompletionsById, CreateRecipeInput, CreateToolProviderInput, CursorPageInput, CustomConfigInput, CustomRecipeFilterInput, DatasetCompletionQuery, DatasetCreate, DatasetCreateFromFilters, DatasetCreateFromMultipartUpload, DatasetUploadProcessingStatusInput, EmojiInput, FeedbackAddInput, FeedbackFilterInput, FeedbackUpdateInput, FromGroupsQuery, GlobalUsageFilterInput, GoogleProviderDataInput, GraderConfigInput, GraderCreateInput, GraderUpdateInput, GroupSelection, GroupSelectionQuery, JobArtifactFilter, JobInput, JudgeConfigInput, JudgeCreate, JudgeExampleInput, JudgeExampleInputTurnEntry, JudgeUpdate, LabelFilter, LabelInput, ListCompletionsFilterInput, ListJobsFilterInput, MetricCreate, MetricGetOrCreate, MetricLink, MetricTrendInput, MetricUnlink, ModelComputeConfigInput, ModelFilter, ModelPlacementInput, ModelProviderDataInput, ModelServiceDisconnect, ModelServiceFilter, NvidiaProviderDataInput, OpenAIProviderDataInput, OrderPair, PrebuiltConfigInput, PrebuiltJudgeCreate, RemoteConfigInput, RemoteEnvCreate, ResizePartitionInput, RoleCreate, SampleConfig, SystemPromptTemplateCreate, SystemPromptTemplateUpdate, TeamCreate, TeamMemberRemove, TeamMemberSet, TimeRange, TimeseriesInput, UnitConfigInput, UpdateCompletion, UpdateModelService, UpdateRecipeInput, UpdateToolProviderInput, UsageFilterInput, UsagePerUseCaseFilterInput, UseCaseCreate, UseCaseFilter, UseCaseMetadataInput, UseCaseSettingsInput, UseCaseShareInput, UseCaseShares, UseCaseUpdate, UserCreate, UserCreateTeamWithRole, WidgetInput
+from .link_metric import LinkMetric, LinkMetricLinkMetric
+from .list_ab_campaigns import ListAbCampaigns, ListAbCampaignsAbCampaigns
+from .list_compute_pools import ListComputePools, ListComputePoolsComputePools, ListComputePoolsComputePoolsHarmonyGroups
+from .list_custom_recipes import ListCustomRecipes, ListCustomRecipesCustomRecipes
+from .list_datasets import ListDatasets, ListDatasetsDatasets
+from .list_graders import ListGraders, ListGradersGraders
+from .list_grouped_interactions import ListGroupedInteractions, ListGroupedInteractionsCompletionsGrouped, ListGroupedInteractionsCompletionsGroupedNodes, ListGroupedInteractionsCompletionsGroupedNodesCompletions, ListGroupedInteractionsCompletionsGroupedNodesCompletionsNodes, ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStats, ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStatsMetric, ListGroupedInteractionsCompletionsGroupedPageInfo
+from .list_harmony_groups import ListHarmonyGroups, ListHarmonyGroupsHarmonyGroups
+from .list_interactions import ListInteractions, ListInteractionsCompletions, ListInteractionsCompletionsNodes, ListInteractionsCompletionsPageInfo
+from .list_jobs import ListJobs, ListJobsJobs, ListJobsJobsNodes, ListJobsJobsPageInfo
+from .list_judge_versions import ListJudgeVersions, ListJudgeVersionsJudgeVersions
+from .list_judges import ListJudges, ListJudgesJudges
+from .list_metrics import ListMetrics, ListMetricsMetrics
+from .list_models import ListModels, ListModelsModels, ListModelsModelsBackbone
+from .list_permissions import ListPermissions
+from .list_remote_envs import ListRemoteEnvs, ListRemoteEnvsRemoteEnvs
+from .list_roles import ListRoles, ListRolesRoles
+from .list_teams import ListTeams, ListTeamsTeams
+from .list_use_cases import ListUseCases, ListUseCasesUseCases
+from .list_users import ListUsers, ListUsersUsers
+from .load_dataset import LoadDataset, LoadDatasetCreateDataset
+from .lock_grader import LockGrader, LockGraderLockGrader
+from .me import Me, MeMe, MeMeApiKeys
+from .remove_remote_env import RemoveRemoteEnv
+from .remove_team_member import RemoveTeamMember, RemoveTeamMemberRemoveTeamMember
+from .share_use_case import ShareUseCase, ShareUseCaseShareUseCase
+from .terminate_model import TerminateModel
+from .test_remote_env import TestRemoteEnv, TestRemoteEnvTestRemoteEnvRemoteEnvTestOffline, TestRemoteEnvTestRemoteEnvRemoteEnvTestOnline
+from .unlink_metric import UnlinkMetric
+from .update_custom_recipe import UpdateCustomRecipe, UpdateCustomRecipeUpdateCustomRecipe
+from .update_grader import UpdateGrader, UpdateGraderUpdateGrader
+from .update_judge import UpdateJudge, UpdateJudgeUpdateJudge
+from .update_model import UpdateModel, UpdateModelUpdateModelService
+from .update_model_compute_config import UpdateModelComputeConfig, UpdateModelComputeConfigUpdateModelComputeConfig
+from .update_user import UpdateUser, UpdateUserSetTeamMember, UpdateUserSetTeamMemberRole, UpdateUserSetTeamMemberTeam, UpdateUserSetTeamMemberUser
+__all__ = ['AbCampaignCreateData', 'AbCampaignDetailData', 'AbCampaignDetailDataMetric', 'AbCampaignDetailDataModels', 'AbCampaignDetailDataUseCase', 'AbCampaignFilter', 'AbCampaignReportData', 'AbCampaignReportDataVariants', 'AbCampaignReportDataVariantsComparisons', 'AbCampaignReportDataVariantsComparisonsVariant', 'AbCampaignReportDataVariantsInterval', 'AbCampaignReportDataVariantsVariant', 'AbcampaignCreate', 'AbcampaignStatus', 'AddExternalModel', 'AddExternalModelAddExternalModel', 'AddExternalModelInput', 'AddHFModel', 'AddHFModelImportHfModel', 'AddHFModelInput', 'AddModel', 'AddModelAddModel', 'AddModelAddModelBackbone', 'AddModelInput', 'AddRemoteEnv', 'AddRemoteEnvAddRemoteEnv', 'AnthropicProviderDataInput', 'ApiKeyCreate', 'ArtifactFilter', 'AsyncBaseClientOpenTelemetry', 'AsyncGQLClient', 'AttachModel', 'AttachModelToUseCase', 'AttachModelToUseCaseAttachModel', 'AuthProviderKind', 'AzureProviderDataInput', 'BaseClientOpenTelemetry', 'BaseModel', 'BillingCycle', 'CancelABCampaign', 'CancelAllocationInput', 'CancelJob', 'CancelJobCancelJob', 'CapabilityFilter', 'CompletionComparisonFeedbackData', 'CompletionComparisonFeedbackDataModel', 'CompletionComparisonFilterInput', 'CompletionData', 'CompletionDataChatMessages', 'CompletionDataComparisonFeedbacks', 'CompletionDataComparisonFeedbacksMetric', 'CompletionDataComparisonFeedbacksOtherCompletion', 'CompletionDataComparisonFeedbacksPreferedCompletion', 'CompletionDataComparisonFeedbacksUsecase', 'CompletionDataDirectFeedbacks', 'CompletionDataDirectFeedbacksMetric', 'CompletionDataLabels', 'CompletionDataMetadata', 'CompletionDataMetadataUsage', 'CompletionDataModel', 'CompletionFeedbackFilterInput', 'CompletionGroupBy', 'CompletionLabelValue', 'CompletionSource', 'CompletionsByFilters', 'CompletionsById', 'ComputePoolCapability', 'CreateAbCampaign', 'CreateAbCampaignCreateAbCampaign', 'CreateCustomRecipe', 'CreateCustomRecipeCreateCustomRecipe', 'CreateGrader', 'CreateGraderCreateGrader', 'CreateJob', 'CreateJobCreateJob', 'CreateJudge', 'CreateJudgeCreateJudge', 'CreateMetric', 'CreateMetricCreateMetric', 'CreatePrebuiltJudge', 'CreatePrebuiltJudgeCreatePrebuiltJudge', 'CreateRecipeInput', 'CreateRole', 'CreateRoleCreateRole', 'CreateTeam', 'CreateTeamCreateTeam', 'CreateToolProviderInput', 'CreateUseCase', 'CreateUseCaseCreateUseCase', 'CreateUser', 'CreateUserCreateUser', 'CursorPageInput', 'CustomConfigInput', 'CustomRecipeData', 'CustomRecipeDataCreatedBy', 'CustomRecipeDataLabels', 'CustomRecipeFilterInput', 'DatasetCompletionQuery', 'DatasetCreate', 'DatasetCreateFromFilters', 'DatasetCreateFromMultipartUpload', 'DatasetData', 'DatasetDataMetricsUsage', 'DatasetDataMetricsUsageMetric', 'DatasetKind', 'DatasetSource', 'DatasetStatus', 'DatasetUploadProcessingStatusInput', 'DateBucketUnit', 'DeleteCustomRecipe', 'DeleteDataset', 'DeleteGrader', 'DeleteGraderDeleteGrader', 'DeleteJudge', 'DeleteJudgeDeleteJudge', 'DeleteUser', 'DeleteUserDeleteUser', 'DeployModel', 'DescribeAbCampaign', 'DescribeAbCampaignAbCampaign', 'DescribeAbCampaignAbCampaignReport', 'DescribeDataset', 'DescribeDatasetDataset', 'DescribeInteraction', 'DescribeInteractionCompletion', 'DescribeJob', 'DescribeJobJob', 'DescribeMetric', 'DescribeMetricAdmin', 'DescribeMetricAdminMetric', 'DescribeMetricMetric', 'DescribeModel', 'DescribeModelAdmin', 'DescribeModelAdminModel', 'DescribeModelAdminModelBackbone', 'DescribeModelModel', 'DescribeModelModelBackbone', 'DescribeUseCase', 'DescribeUseCaseUseCase', 'EmojiInput', 'ExternalModelProviderName', 'FeedbackAddInput', 'FeedbackFilterInput', 'FeedbackType', 'FeedbackUpdateInput', 'FromGroupsQuery', 'GQLClient', 'GetCustomRecipe', 'GetCustomRecipeCustomRecipe', 'GetGrader', 'GetGraderGrader', 'GetJudge', 'GetJudgeJudge', 'GlobalUsageFilterInput', 'GoogleProviderDataInput', 'GraderConfigInput', 'GraderCreateInput', 'GraderData', 'GraderDataGraderConfigCustomConfigOutput', 'GraderDataGraderConfigJudgeConfigOutput', 'GraderDataGraderConfigJudgeConfigOutputExamples', 'GraderDataGraderConfigJudgeConfigOutputExamplesInput', 'GraderDataGraderConfigJudgeConfigOutputModel', 'GraderDataGraderConfigPrebuiltConfigOutput', 'GraderDataGraderConfigPrebuiltConfigOutputModel', 'GraderDataGraderConfigPrebuiltConfigOutputPrebuiltCriteria', 'GraderDataGraderConfigRemoteConfigOutput', 'GraderDataMetric', 'GraderDataUseCase', 'GraderTypeEnum', 'GraderUpdateInput', 'GraphQLClientError', 'GraphQLClientGraphQLError', 'GraphQLClientGraphQLMultiError', 'GraphQLClientHttpError', 'GraphQLClientInvalidResponseError', 'GroupSelection', 'GroupSelectionQuery', 'HarmonyGroupData', 'HarmonyGroupDataComputePool', 'HarmonyGroupDataOnlineModels', 'HarmonyStatus', 'JobArtifactFilter', 'JobArtifactKind', 'JobData', 'JobDataCreatedBy', 'JobDataDetails', 'JobDataDetailsArtifacts', 'JobDataRecipe', 'JobDataStages', 'JobDataStagesInfoBatchInferenceJobStageOutput', 'JobDataStagesInfoEvalJobStageOutput', 'JobDataStagesInfoTrainingJobStageOutput', 'JobDataUseCase', 'JobInput', 'JobKind', 'JobStageOutputData', 'JobStageOutputDataInfoBatchInferenceJobStageOutput', 'JobStageOutputDataInfoEvalJobStageOutput', 'JobStageOutputDataInfoTrainingJobStageOutput', 'JobStatus', 'JobStatusOutput', 'JudgeCapability', 'JudgeConfigInput', 'JudgeCreate', 'JudgeData', 'JudgeDataExamples', 'JudgeDataExamplesInput', 'JudgeDataMetric', 'JudgeDataModel', 'JudgeExampleInput', 'JudgeExampleInputTurnEntry', 'JudgeUpdate', 'LabelFilter', 'LabelInput', 'LinkMetric', 'LinkMetricLinkMetric', 'ListAbCampaigns', 'ListAbCampaignsAbCampaigns', 'ListCompletionsFilterInput', 'ListComputePools', 'ListComputePoolsComputePools', 'ListComputePoolsComputePoolsHarmonyGroups', 'ListCustomRecipes', 'ListCustomRecipesCustomRecipes', 'ListDatasets', 'ListDatasetsDatasets', 'ListGraders', 'ListGradersGraders', 'ListGroupedInteractions', 'ListGroupedInteractionsCompletionsGrouped', 'ListGroupedInteractionsCompletionsGroupedNodes', 'ListGroupedInteractionsCompletionsGroupedNodesCompletions', 'ListGroupedInteractionsCompletionsGroupedNodesCompletionsNodes', 'ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStats', 'ListGroupedInteractionsCompletionsGroupedNodesDirectFeedbacksStatsMetric', 'ListGroupedInteractionsCompletionsGroupedPageInfo', 'ListHarmonyGroups', 'ListHarmonyGroupsHarmonyGroups', 'ListInteractions', 'ListInteractionsCompletions', 'ListInteractionsCompletionsNodes', 'ListInteractionsCompletionsPageInfo', 'ListJobs', 'ListJobsFilterInput', 'ListJobsJobs', 'ListJobsJobsNodes', 'ListJobsJobsPageInfo', 'ListJudgeVersions', 'ListJudgeVersionsJudgeVersions', 'ListJudges', 'ListJudgesJudges', 'ListMetrics', 'ListMetricsMetrics', 'ListModels', 'ListModelsModels', 'ListModelsModelsBackbone', 'ListPermissions', 'ListRemoteEnvs', 'ListRemoteEnvsRemoteEnvs', 'ListRoles', 'ListRolesRoles', 'ListTeams', 'ListTeamsTeams', 'ListUseCases', 'ListUseCasesUseCases', 'ListUsers', 'ListUsersUsers', 'LoadDataset', 'LoadDatasetCreateDataset', 'LockGrader', 'LockGraderLockGrader', 'Me', 'MeMe', 'MeMeApiKeys', 'MetricAggregation', 'MetricCreate', 'MetricData', 'MetricDataAdmin', 'MetricDataAdminUseCases', 'MetricGetOrCreate', 'MetricKind', 'MetricLink', 'MetricScoringType', 'MetricTrendInput', 'MetricUnlink', 'MetricWithContextData', 'ModelCapabilityFilter', 'ModelComputeConfigInput', 'ModelData', 'ModelDataAdmin', 'ModelDataAdminUseCases', 'ModelDataComputeConfig', 'ModelFilter', 'ModelOnline', 'ModelPlacementInput', 'ModelProviderDataInput', 'ModelServiceData', 'ModelServiceDataModel', 'ModelServiceDataModelBackbone', 'ModelServiceDisconnect', 'ModelServiceFilter', 'ModelserviceStatus', 'NvidiaProviderDataInput', 'OpenAIModel', 'OpenAIProviderDataInput', 'OrderPair', 'PrebuiltConfigInput', 'PrebuiltCriteriaKey', 'PrebuiltJudgeCreate', 'Protocol', 'ProviderName', 'RemoteConfigInput', 'RemoteEnvCreate', 'RemoteEnvData', 'RemoteEnvStatus', 'RemoveRemoteEnv', 'RemoveTeamMember', 'RemoveTeamMemberRemoveTeamMember', 'ResizePartitionInput', 'RoleCreate', 'SampleConfig', 'SelectionTypeInput', 'SessionStatus', 'ShareUseCase', 'ShareUseCaseShareUseCase', 'SortDirection', 'SystemPromptTemplateCreate', 'SystemPromptTemplateUpdate', 'TeamCreate', 'TeamMemberRemove', 'TeamMemberSet', 'TerminateModel', 'TestRemoteEnv', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOffline', 'TestRemoteEnvTestRemoteEnvRemoteEnvTestOnline', 'TimeRange', 'TimeseriesInput', 'TimeseriesInterval', 'UnitConfigInput', 'UnitPosition', 'UnlinkMetric', 'UpdateCompletion', 'UpdateCustomRecipe', 'UpdateCustomRecipeUpdateCustomRecipe', 'UpdateGrader', 'UpdateGraderUpdateGrader', 'UpdateJudge', 'UpdateJudgeUpdateJudge', 'UpdateModel', 'UpdateModelComputeConfig', 'UpdateModelComputeConfigUpdateModelComputeConfig', 'UpdateModelService', 'UpdateModelUpdateModelService', 'UpdateRecipeInput', 'UpdateToolProviderInput', 'UpdateUser', 'UpdateUserSetTeamMember', 'UpdateUserSetTeamMemberRole', 'UpdateUserSetTeamMemberTeam', 'UpdateUserSetTeamMemberUser', 'Upload', 'UsageFilterInput', 'UsagePerUseCaseFilterInput', 'UseCaseCreate', 'UseCaseData', 'UseCaseDataMetrics', 'UseCaseDataModelServices', 'UseCaseDataShares', 'UseCaseDataSharesRole', 'UseCaseDataSharesTeam', 'UseCaseFilter', 'UseCaseMetadataInput', 'UseCaseSettingsInput', 'UseCaseShareInput', 'UseCaseShares', 'UseCaseUpdate', 'UserCreate', 'UserCreateTeamWithRole', 'UserData', 'UserDataTeams', 'UserDataTeamsRole', 'UserDataTeamsTeam', 'WidgetInput']

{adaptive_sdk-0.1.6 → adaptive_sdk-0.1.7}/src/adaptive_sdk/graphql_client/async_base_client_open_telemetry.py

@@ -103,7 +103,7 @@ class AsyncBaseClientOpenTelemetry:
             yield message
 
     async def _execute(self, query: str, operation_name: Optional[str]=None, variables: Optional[Dict[str, Any]]=None, **kwargs: Any) -> httpx.Response:
-        (processed_variables, files, files_map) = self._process_variables(variables)
+        processed_variables, files, files_map = self._process_variables(variables)
         if files and files_map:
             return await self._execute_multipart(query=query, operation_name=operation_name, variables=processed_variables, files=files, files_map=files_map, **kwargs)
         return await self._execute_json(query=query, operation_name=operation_name, variables=processed_variables, **kwargs)
@@ -115,7 +115,7 @@ class AsyncBaseClientOpenTelemetry:
         return self._get_files_from_variables(serializable_variables)
 
     def _convert_dict_to_json_serializable(self, dict_: Dict[str, Any]) -> Dict[str, Any]:
-        return {key: self._convert_value(value) for (key, value) in dict_.items() if value is not UNSET}
+        return {key: self._convert_value(value) for key, value in dict_.items() if value is not UNSET}
 
     def _convert_value(self, value: Any) -> Any:
         if isinstance(value, BaseModel):
@@ -131,13 +131,13 @@ class AsyncBaseClientOpenTelemetry:
         def separate_files(path: str, obj: Any) -> Any:
             if isinstance(obj, list):
                 nulled_list = []
-                for (index, value) in enumerate(obj):
+                for index, value in enumerate(obj):
                     value = separate_files(f'{path}.{index}', value)
                     nulled_list.append(value)
                 return nulled_list
             if isinstance(obj, dict):
                 nulled_dict = {}
-                for (key, value) in obj.items():
+                for key, value in obj.items():
                     value = separate_files(f'{path}.{key}', value)
                     nulled_dict[key] = value
                 return nulled_dict
@@ -152,7 +152,7 @@ class AsyncBaseClientOpenTelemetry:
                 return None
             return obj
         nulled_variables = separate_files('variables', variables)
-        files: Dict[str, Tuple[str, IO[bytes], str]] = {str(i): (file_.filename, cast(IO[bytes], file_.content), file_.content_type) for (i, file_) in enumerate(files_list)}
+        files: Dict[str, Tuple[str, IO[bytes], str]] = {str(i): (file_.filename, cast(IO[bytes], file_.content), file_.content_type) for i, file_ in enumerate(files_list)}
         return (nulled_variables, files, files_map)
 
     async def _execute_multipart(self, query: str, operation_name: Optional[str], variables: Dict[str, Any], files: Dict[str, Tuple[str, IO[bytes], str]], files_map: Dict[str, List[str]], **kwargs: Any) -> httpx.Response:
@@ -220,7 +220,7 @@ class AsyncBaseClientOpenTelemetry:
     async def _execute_with_telemetry(self, query: str, operation_name: Optional[str]=None, variables: Optional[Dict[str, Any]]=None, **kwargs: Any) -> httpx.Response:
         with self.tracer.start_as_current_span(self.root_span_name, context=self.root_context) as root_span:
             root_span.set_attribute('component', 'GraphQL Client')
-            (processed_variables, files, files_map) = self._process_variables(variables)
+            processed_variables, files, files_map = self._process_variables(variables)
             if files and files_map:
                 return await self._execute_multipart_with_telemetry(root_span=root_span, query=query, operation_name=operation_name, variables=processed_variables, files=files, files_map=files_map, **kwargs)
             return await self._execute_json_with_telemetry(root_span=root_span, query=query, operation_name=operation_name, variables=processed_variables, **kwargs)