Skip to content

Commit

Permalink
SDK - Client - Added API models to the generated API (#2418)
Browse files Browse the repository at this point in the history
* SDK - Client - Added API models to the generated API

The API objects should now be available in the generated client API: `kfp.Client().jobs.models.ApiJob`.

* Simplified the change
  • Loading branch information
Ark-kun authored and k8s-ci-robot committed Nov 2, 2019
1 parent 8112af1 commit c063a05
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions sdk/python/kfp/_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,14 +53,19 @@ def camel_case_to_snake_case(name):
short_api_name = camel_case_to_snake_case(api_name[0:-len('ServiceApi')]) + 's'
api_struct = Struct()
setattr(target_struct, short_api_name, api_struct)
service_api = getattr(api_module, api_name)
service_api = getattr(api_module.api, api_name)
initialized_service_api = service_api(api_client)
for member_name in dir(initialized_service_api):
if member_name.startswith('_') or member_name.endswith('_with_http_info'):
continue

bound_member = getattr(initialized_service_api, member_name)
setattr(api_struct, member_name, bound_member)
models_struct = Struct()
for member_name in dir(api_module.models):
if not member_name[0].islower():
setattr(models_struct, member_name, getattr(api_module.models, member_name))
target_struct.api_models = models_struct


KF_PIPELINES_ENDPOINT_ENV = 'KF_PIPELINES_ENDPOINT'
Expand Down Expand Up @@ -93,7 +98,7 @@ def __init__(self, host=None, client_id=None, namespace='kubeflow'):
self._uihost = os.environ.get(KF_PIPELINES_UI_ENDPOINT_ENV, host)
config = self._load_config(host, client_id, namespace)
api_client = kfp_server_api.api_client.ApiClient(config)
_add_generated_apis(self, kfp_server_api.api, api_client)
_add_generated_apis(self, kfp_server_api, api_client)
self._run_api = kfp_server_api.api.run_service_api.RunServiceApi(api_client)
self._experiment_api = kfp_server_api.api.experiment_service_api.ExperimentServiceApi(api_client)
self._pipelines_api = kfp_server_api.api.pipeline_service_api.PipelineServiceApi(api_client)
Expand Down

0 comments on commit c063a05

Please sign in to comment.