truefoundry 0.5.5rc1__py3-none-any.whl → 0.5.6__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of truefoundry might be problematic. Click here for more details.

Files changed (35) hide show
  1. truefoundry/deploy/__init__.py +9 -2
  2. truefoundry/deploy/auto_gen/models.py +177 -57
  3. truefoundry/deploy/cli/commands/__init__.py +0 -1
  4. truefoundry/deploy/lib/clients/servicefoundry_client.py +20 -16
  5. truefoundry/deploy/v2/lib/deploy.py +23 -15
  6. truefoundry/deploy/v2/lib/deploy_workflow.py +9 -4
  7. truefoundry/deploy/v2/lib/deployable_patched_models.py +10 -0
  8. truefoundry/deploy/v2/lib/patched_models.py +9 -1
  9. truefoundry/ml/autogen/client/__init__.py +4 -4
  10. truefoundry/ml/autogen/client/api/experiments_api.py +0 -156
  11. truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py +49 -305
  12. truefoundry/ml/autogen/client/models/__init__.py +4 -4
  13. truefoundry/ml/autogen/client/models/agent.py +64 -13
  14. truefoundry/ml/autogen/client/models/agent_open_api_tool.py +63 -17
  15. truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py +63 -17
  16. truefoundry/ml/autogen/client/models/agent_with_fqn.py +64 -12
  17. truefoundry/ml/autogen/client/models/artifact_version_manifest.py +39 -4
  18. truefoundry/ml/autogen/client/models/chat_prompt.py +60 -22
  19. truefoundry/ml/autogen/client/models/{external_artifact_source.py → external_blob_storage_source.py} +10 -11
  20. truefoundry/ml/autogen/client/models/model_version_manifest.py +39 -4
  21. truefoundry/ml/autogen/client/models/source.py +22 -22
  22. truefoundry/ml/autogen/client/models/source1.py +22 -22
  23. truefoundry/ml/autogen/client/models/{true_foundry_artifact_source.py → true_foundry_managed_source.py} +12 -11
  24. truefoundry/ml/autogen/client_README.md +2 -3
  25. truefoundry/ml/autogen/entities/artifacts.py +87 -69
  26. truefoundry/ml/log_types/artifacts/artifact.py +26 -18
  27. truefoundry/ml/log_types/artifacts/general_artifact.py +4 -13
  28. truefoundry/ml/log_types/artifacts/model.py +28 -21
  29. truefoundry/ml/mlfoundry_api.py +2 -6
  30. truefoundry/workflow/task.py +1 -1
  31. {truefoundry-0.5.5rc1.dist-info → truefoundry-0.5.6.dist-info}/METADATA +2 -2
  32. {truefoundry-0.5.5rc1.dist-info → truefoundry-0.5.6.dist-info}/RECORD +34 -35
  33. truefoundry/deploy/cli/commands/build_logs_command.py +0 -89
  34. {truefoundry-0.5.5rc1.dist-info → truefoundry-0.5.6.dist-info}/WHEEL +0 -0
  35. {truefoundry-0.5.5rc1.dist-info → truefoundry-0.5.6.dist-info}/entry_points.txt +0 -0
@@ -12,7 +12,7 @@ Do not edit the class manually.
12
12
  """ # noqa: E501
13
13
 
14
14
  import re # noqa: F401
15
- from typing import Any, Optional
15
+ from typing import Optional
16
16
 
17
17
  from truefoundry.ml.autogen.client.api_client import ApiClient
18
18
  from truefoundry.ml.autogen.client.api_response import ApiResponse
@@ -513,26 +513,18 @@ class MlfoundryArtifactsApi:
513
513
 
514
514
  @validate_arguments
515
515
  def create_artifact_post(
516
- self,
517
- create_artifact_request_dto: CreateArtifactRequestDto,
518
- backend_store_uri: Optional[Any] = None,
519
- default_artifact_root: Optional[Any] = None,
520
- **kwargs,
516
+ self, create_artifact_request_dto: CreateArtifactRequestDto, **kwargs
521
517
  ) -> CreateArtifactResponseDto: # noqa: E501
522
518
  """Create Artifact # noqa: E501
523
519
 
524
520
  This method makes a synchronous HTTP request by default. To make an
525
521
  asynchronous HTTP request, please pass async_req=True
526
522
 
527
- >>> thread = api.create_artifact_post(create_artifact_request_dto, backend_store_uri, default_artifact_root, async_req=True)
523
+ >>> thread = api.create_artifact_post(create_artifact_request_dto, async_req=True)
528
524
  >>> result = thread.get()
529
525
 
530
526
  :param create_artifact_request_dto: (required)
531
527
  :type create_artifact_request_dto: CreateArtifactRequestDto
532
- :param backend_store_uri:
533
- :type backend_store_uri: object
534
- :param default_artifact_root:
535
- :type default_artifact_root: object
536
528
  :param async_req: Whether to execute the request asynchronously.
537
529
  :type async_req: bool, optional
538
530
  :param _request_timeout: timeout setting for this request.
@@ -549,34 +541,23 @@ class MlfoundryArtifactsApi:
549
541
  message = "Error! Please call the create_artifact_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
550
542
  raise ValueError(message)
551
543
  return self.create_artifact_post_with_http_info(
552
- create_artifact_request_dto,
553
- backend_store_uri,
554
- default_artifact_root,
555
- **kwargs,
544
+ create_artifact_request_dto, **kwargs
556
545
  ) # noqa: E501
557
546
 
558
547
  @validate_arguments
559
548
  def create_artifact_post_with_http_info(
560
- self,
561
- create_artifact_request_dto: CreateArtifactRequestDto,
562
- backend_store_uri: Optional[Any] = None,
563
- default_artifact_root: Optional[Any] = None,
564
- **kwargs,
549
+ self, create_artifact_request_dto: CreateArtifactRequestDto, **kwargs
565
550
  ) -> ApiResponse: # noqa: E501
566
551
  """Create Artifact # noqa: E501
567
552
 
568
553
  This method makes a synchronous HTTP request by default. To make an
569
554
  asynchronous HTTP request, please pass async_req=True
570
555
 
571
- >>> thread = api.create_artifact_post_with_http_info(create_artifact_request_dto, backend_store_uri, default_artifact_root, async_req=True)
556
+ >>> thread = api.create_artifact_post_with_http_info(create_artifact_request_dto, async_req=True)
572
557
  >>> result = thread.get()
573
558
 
574
559
  :param create_artifact_request_dto: (required)
575
560
  :type create_artifact_request_dto: CreateArtifactRequestDto
576
- :param backend_store_uri:
577
- :type backend_store_uri: object
578
- :param default_artifact_root:
579
- :type default_artifact_root: object
580
561
  :param async_req: Whether to execute the request asynchronously.
581
562
  :type async_req: bool, optional
582
563
  :param _preload_content: if False, the ApiResponse.data will
@@ -604,11 +585,7 @@ class MlfoundryArtifactsApi:
604
585
 
605
586
  _params = locals()
606
587
 
607
- _all_params = [
608
- "create_artifact_request_dto",
609
- "backend_store_uri",
610
- "default_artifact_root",
611
- ]
588
+ _all_params = ["create_artifact_request_dto"]
612
589
  _all_params.extend(
613
590
  [
614
591
  "async_req",
@@ -638,14 +615,6 @@ class MlfoundryArtifactsApi:
638
615
 
639
616
  # process the query parameters
640
617
  _query_params = []
641
- if _params.get("backend_store_uri") is not None: # noqa: E501
642
- _query_params.append(("backend_store_uri", _params["backend_store_uri"]))
643
-
644
- if _params.get("default_artifact_root") is not None: # noqa: E501
645
- _query_params.append(
646
- ("default_artifact_root", _params["default_artifact_root"])
647
- )
648
-
649
618
  # process the header parameters
650
619
  _header_params = dict(_params.get("_headers", {}))
651
620
  # process the form parameters
@@ -700,8 +669,6 @@ class MlfoundryArtifactsApi:
700
669
  def create_artifact_version_post(
701
670
  self,
702
671
  create_artifact_version_request_dto: CreateArtifactVersionRequestDto,
703
- backend_store_uri: Optional[Any] = None,
704
- default_artifact_root: Optional[Any] = None,
705
672
  **kwargs,
706
673
  ) -> CreateArtifactVersionResponseDto: # noqa: E501
707
674
  """Create Artifact Version # noqa: E501
@@ -710,15 +677,11 @@ class MlfoundryArtifactsApi:
710
677
  This method makes a synchronous HTTP request by default. To make an
711
678
  asynchronous HTTP request, please pass async_req=True
712
679
 
713
- >>> thread = api.create_artifact_version_post(create_artifact_version_request_dto, backend_store_uri, default_artifact_root, async_req=True)
680
+ >>> thread = api.create_artifact_version_post(create_artifact_version_request_dto, async_req=True)
714
681
  >>> result = thread.get()
715
682
 
716
683
  :param create_artifact_version_request_dto: (required)
717
684
  :type create_artifact_version_request_dto: CreateArtifactVersionRequestDto
718
- :param backend_store_uri:
719
- :type backend_store_uri: object
720
- :param default_artifact_root:
721
- :type default_artifact_root: object
722
685
  :param async_req: Whether to execute the request asynchronously.
723
686
  :type async_req: bool, optional
724
687
  :param _request_timeout: timeout setting for this request.
@@ -735,18 +698,13 @@ class MlfoundryArtifactsApi:
735
698
  message = "Error! Please call the create_artifact_version_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
736
699
  raise ValueError(message)
737
700
  return self.create_artifact_version_post_with_http_info(
738
- create_artifact_version_request_dto,
739
- backend_store_uri,
740
- default_artifact_root,
741
- **kwargs,
701
+ create_artifact_version_request_dto, **kwargs
742
702
  ) # noqa: E501
743
703
 
744
704
  @validate_arguments
745
705
  def create_artifact_version_post_with_http_info(
746
706
  self,
747
707
  create_artifact_version_request_dto: CreateArtifactVersionRequestDto,
748
- backend_store_uri: Optional[Any] = None,
749
- default_artifact_root: Optional[Any] = None,
750
708
  **kwargs,
751
709
  ) -> ApiResponse: # noqa: E501
752
710
  """Create Artifact Version # noqa: E501
@@ -755,15 +713,11 @@ class MlfoundryArtifactsApi:
755
713
  This method makes a synchronous HTTP request by default. To make an
756
714
  asynchronous HTTP request, please pass async_req=True
757
715
 
758
- >>> thread = api.create_artifact_version_post_with_http_info(create_artifact_version_request_dto, backend_store_uri, default_artifact_root, async_req=True)
716
+ >>> thread = api.create_artifact_version_post_with_http_info(create_artifact_version_request_dto, async_req=True)
759
717
  >>> result = thread.get()
760
718
 
761
719
  :param create_artifact_version_request_dto: (required)
762
720
  :type create_artifact_version_request_dto: CreateArtifactVersionRequestDto
763
- :param backend_store_uri:
764
- :type backend_store_uri: object
765
- :param default_artifact_root:
766
- :type default_artifact_root: object
767
721
  :param async_req: Whether to execute the request asynchronously.
768
722
  :type async_req: bool, optional
769
723
  :param _preload_content: if False, the ApiResponse.data will
@@ -791,11 +745,7 @@ class MlfoundryArtifactsApi:
791
745
 
792
746
  _params = locals()
793
747
 
794
- _all_params = [
795
- "create_artifact_version_request_dto",
796
- "backend_store_uri",
797
- "default_artifact_root",
798
- ]
748
+ _all_params = ["create_artifact_version_request_dto"]
799
749
  _all_params.extend(
800
750
  [
801
751
  "async_req",
@@ -825,14 +775,6 @@ class MlfoundryArtifactsApi:
825
775
 
826
776
  # process the query parameters
827
777
  _query_params = []
828
- if _params.get("backend_store_uri") is not None: # noqa: E501
829
- _query_params.append(("backend_store_uri", _params["backend_store_uri"]))
830
-
831
- if _params.get("default_artifact_root") is not None: # noqa: E501
832
- _query_params.append(
833
- ("default_artifact_root", _params["default_artifact_root"])
834
- )
835
-
836
778
  # process the header parameters
837
779
  _header_params = dict(_params.get("_headers", {}))
838
780
  # process the form parameters
@@ -2451,8 +2393,6 @@ class MlfoundryArtifactsApi:
2451
2393
  def finalize_artifact_version_post(
2452
2394
  self,
2453
2395
  finalize_artifact_version_request_dto: FinalizeArtifactVersionRequestDto,
2454
- backend_store_uri: Optional[Any] = None,
2455
- default_artifact_root: Optional[Any] = None,
2456
2396
  **kwargs,
2457
2397
  ) -> ArtifactVersionResponseDto: # noqa: E501
2458
2398
  """Finalize Artifact Version # noqa: E501
@@ -2460,15 +2400,11 @@ class MlfoundryArtifactsApi:
2460
2400
  This method makes a synchronous HTTP request by default. To make an
2461
2401
  asynchronous HTTP request, please pass async_req=True
2462
2402
 
2463
- >>> thread = api.finalize_artifact_version_post(finalize_artifact_version_request_dto, backend_store_uri, default_artifact_root, async_req=True)
2403
+ >>> thread = api.finalize_artifact_version_post(finalize_artifact_version_request_dto, async_req=True)
2464
2404
  >>> result = thread.get()
2465
2405
 
2466
2406
  :param finalize_artifact_version_request_dto: (required)
2467
2407
  :type finalize_artifact_version_request_dto: FinalizeArtifactVersionRequestDto
2468
- :param backend_store_uri:
2469
- :type backend_store_uri: object
2470
- :param default_artifact_root:
2471
- :type default_artifact_root: object
2472
2408
  :param async_req: Whether to execute the request asynchronously.
2473
2409
  :type async_req: bool, optional
2474
2410
  :param _request_timeout: timeout setting for this request.
@@ -2485,18 +2421,13 @@ class MlfoundryArtifactsApi:
2485
2421
  message = "Error! Please call the finalize_artifact_version_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
2486
2422
  raise ValueError(message)
2487
2423
  return self.finalize_artifact_version_post_with_http_info(
2488
- finalize_artifact_version_request_dto,
2489
- backend_store_uri,
2490
- default_artifact_root,
2491
- **kwargs,
2424
+ finalize_artifact_version_request_dto, **kwargs
2492
2425
  ) # noqa: E501
2493
2426
 
2494
2427
  @validate_arguments
2495
2428
  def finalize_artifact_version_post_with_http_info(
2496
2429
  self,
2497
2430
  finalize_artifact_version_request_dto: FinalizeArtifactVersionRequestDto,
2498
- backend_store_uri: Optional[Any] = None,
2499
- default_artifact_root: Optional[Any] = None,
2500
2431
  **kwargs,
2501
2432
  ) -> ApiResponse: # noqa: E501
2502
2433
  """Finalize Artifact Version # noqa: E501
@@ -2504,15 +2435,11 @@ class MlfoundryArtifactsApi:
2504
2435
  This method makes a synchronous HTTP request by default. To make an
2505
2436
  asynchronous HTTP request, please pass async_req=True
2506
2437
 
2507
- >>> thread = api.finalize_artifact_version_post_with_http_info(finalize_artifact_version_request_dto, backend_store_uri, default_artifact_root, async_req=True)
2438
+ >>> thread = api.finalize_artifact_version_post_with_http_info(finalize_artifact_version_request_dto, async_req=True)
2508
2439
  >>> result = thread.get()
2509
2440
 
2510
2441
  :param finalize_artifact_version_request_dto: (required)
2511
2442
  :type finalize_artifact_version_request_dto: FinalizeArtifactVersionRequestDto
2512
- :param backend_store_uri:
2513
- :type backend_store_uri: object
2514
- :param default_artifact_root:
2515
- :type default_artifact_root: object
2516
2443
  :param async_req: Whether to execute the request asynchronously.
2517
2444
  :type async_req: bool, optional
2518
2445
  :param _preload_content: if False, the ApiResponse.data will
@@ -2540,11 +2467,7 @@ class MlfoundryArtifactsApi:
2540
2467
 
2541
2468
  _params = locals()
2542
2469
 
2543
- _all_params = [
2544
- "finalize_artifact_version_request_dto",
2545
- "backend_store_uri",
2546
- "default_artifact_root",
2547
- ]
2470
+ _all_params = ["finalize_artifact_version_request_dto"]
2548
2471
  _all_params.extend(
2549
2472
  [
2550
2473
  "async_req",
@@ -2574,14 +2497,6 @@ class MlfoundryArtifactsApi:
2574
2497
 
2575
2498
  # process the query parameters
2576
2499
  _query_params = []
2577
- if _params.get("backend_store_uri") is not None: # noqa: E501
2578
- _query_params.append(("backend_store_uri", _params["backend_store_uri"]))
2579
-
2580
- if _params.get("default_artifact_root") is not None: # noqa: E501
2581
- _query_params.append(
2582
- ("default_artifact_root", _params["default_artifact_root"])
2583
- )
2584
-
2585
2500
  # process the header parameters
2586
2501
  _header_params = dict(_params.get("_headers", {}))
2587
2502
  # process the form parameters
@@ -2633,27 +2548,17 @@ class MlfoundryArtifactsApi:
2633
2548
  )
2634
2549
 
2635
2550
  @validate_arguments
2636
- def get_artifact_by_fqn_get(
2637
- self,
2638
- fqn: StrictStr,
2639
- backend_store_uri: Optional[Any] = None,
2640
- default_artifact_root: Optional[Any] = None,
2641
- **kwargs,
2642
- ) -> ArtifactResponseDto: # noqa: E501
2551
+ def get_artifact_by_fqn_get(self, fqn: StrictStr, **kwargs) -> ArtifactResponseDto: # noqa: E501
2643
2552
  """Get Artifact By Fqn # noqa: E501
2644
2553
 
2645
2554
  This method makes a synchronous HTTP request by default. To make an
2646
2555
  asynchronous HTTP request, please pass async_req=True
2647
2556
 
2648
- >>> thread = api.get_artifact_by_fqn_get(fqn, backend_store_uri, default_artifact_root, async_req=True)
2557
+ >>> thread = api.get_artifact_by_fqn_get(fqn, async_req=True)
2649
2558
  >>> result = thread.get()
2650
2559
 
2651
2560
  :param fqn: (required)
2652
2561
  :type fqn: str
2653
- :param backend_store_uri:
2654
- :type backend_store_uri: object
2655
- :param default_artifact_root:
2656
- :type default_artifact_root: object
2657
2562
  :param async_req: Whether to execute the request asynchronously.
2658
2563
  :type async_req: bool, optional
2659
2564
  :param _request_timeout: timeout setting for this request.
@@ -2669,32 +2574,22 @@ class MlfoundryArtifactsApi:
2669
2574
  if "_preload_content" in kwargs:
2670
2575
  message = "Error! Please call the get_artifact_by_fqn_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
2671
2576
  raise ValueError(message)
2672
- return self.get_artifact_by_fqn_get_with_http_info(
2673
- fqn, backend_store_uri, default_artifact_root, **kwargs
2674
- ) # noqa: E501
2577
+ return self.get_artifact_by_fqn_get_with_http_info(fqn, **kwargs) # noqa: E501
2675
2578
 
2676
2579
  @validate_arguments
2677
2580
  def get_artifact_by_fqn_get_with_http_info(
2678
- self,
2679
- fqn: StrictStr,
2680
- backend_store_uri: Optional[Any] = None,
2681
- default_artifact_root: Optional[Any] = None,
2682
- **kwargs,
2581
+ self, fqn: StrictStr, **kwargs
2683
2582
  ) -> ApiResponse: # noqa: E501
2684
2583
  """Get Artifact By Fqn # noqa: E501
2685
2584
 
2686
2585
  This method makes a synchronous HTTP request by default. To make an
2687
2586
  asynchronous HTTP request, please pass async_req=True
2688
2587
 
2689
- >>> thread = api.get_artifact_by_fqn_get_with_http_info(fqn, backend_store_uri, default_artifact_root, async_req=True)
2588
+ >>> thread = api.get_artifact_by_fqn_get_with_http_info(fqn, async_req=True)
2690
2589
  >>> result = thread.get()
2691
2590
 
2692
2591
  :param fqn: (required)
2693
2592
  :type fqn: str
2694
- :param backend_store_uri:
2695
- :type backend_store_uri: object
2696
- :param default_artifact_root:
2697
- :type default_artifact_root: object
2698
2593
  :param async_req: Whether to execute the request asynchronously.
2699
2594
  :type async_req: bool, optional
2700
2595
  :param _preload_content: if False, the ApiResponse.data will
@@ -2722,7 +2617,7 @@ class MlfoundryArtifactsApi:
2722
2617
 
2723
2618
  _params = locals()
2724
2619
 
2725
- _all_params = ["fqn", "backend_store_uri", "default_artifact_root"]
2620
+ _all_params = ["fqn"]
2726
2621
  _all_params.extend(
2727
2622
  [
2728
2623
  "async_req",
@@ -2755,14 +2650,6 @@ class MlfoundryArtifactsApi:
2755
2650
  if _params.get("fqn") is not None: # noqa: E501
2756
2651
  _query_params.append(("fqn", _params["fqn"]))
2757
2652
 
2758
- if _params.get("backend_store_uri") is not None: # noqa: E501
2759
- _query_params.append(("backend_store_uri", _params["backend_store_uri"]))
2760
-
2761
- if _params.get("default_artifact_root") is not None: # noqa: E501
2762
- _query_params.append(
2763
- ("default_artifact_root", _params["default_artifact_root"])
2764
- )
2765
-
2766
2653
  # process the header parameters
2767
2654
  _header_params = dict(_params.get("_headers", {}))
2768
2655
  # process the form parameters
@@ -2803,27 +2690,17 @@ class MlfoundryArtifactsApi:
2803
2690
  )
2804
2691
 
2805
2692
  @validate_arguments
2806
- def get_artifact_by_id_get(
2807
- self,
2808
- id: StrictStr,
2809
- backend_store_uri: Optional[Any] = None,
2810
- default_artifact_root: Optional[Any] = None,
2811
- **kwargs,
2812
- ) -> ArtifactResponseDto: # noqa: E501
2693
+ def get_artifact_by_id_get(self, id: StrictStr, **kwargs) -> ArtifactResponseDto: # noqa: E501
2813
2694
  """Get Artifact By Id # noqa: E501
2814
2695
 
2815
2696
  This method makes a synchronous HTTP request by default. To make an
2816
2697
  asynchronous HTTP request, please pass async_req=True
2817
2698
 
2818
- >>> thread = api.get_artifact_by_id_get(id, backend_store_uri, default_artifact_root, async_req=True)
2699
+ >>> thread = api.get_artifact_by_id_get(id, async_req=True)
2819
2700
  >>> result = thread.get()
2820
2701
 
2821
2702
  :param id: (required)
2822
2703
  :type id: str
2823
- :param backend_store_uri:
2824
- :type backend_store_uri: object
2825
- :param default_artifact_root:
2826
- :type default_artifact_root: object
2827
2704
  :param async_req: Whether to execute the request asynchronously.
2828
2705
  :type async_req: bool, optional
2829
2706
  :param _request_timeout: timeout setting for this request.
@@ -2839,32 +2716,22 @@ class MlfoundryArtifactsApi:
2839
2716
  if "_preload_content" in kwargs:
2840
2717
  message = "Error! Please call the get_artifact_by_id_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
2841
2718
  raise ValueError(message)
2842
- return self.get_artifact_by_id_get_with_http_info(
2843
- id, backend_store_uri, default_artifact_root, **kwargs
2844
- ) # noqa: E501
2719
+ return self.get_artifact_by_id_get_with_http_info(id, **kwargs) # noqa: E501
2845
2720
 
2846
2721
  @validate_arguments
2847
2722
  def get_artifact_by_id_get_with_http_info(
2848
- self,
2849
- id: StrictStr,
2850
- backend_store_uri: Optional[Any] = None,
2851
- default_artifact_root: Optional[Any] = None,
2852
- **kwargs,
2723
+ self, id: StrictStr, **kwargs
2853
2724
  ) -> ApiResponse: # noqa: E501
2854
2725
  """Get Artifact By Id # noqa: E501
2855
2726
 
2856
2727
  This method makes a synchronous HTTP request by default. To make an
2857
2728
  asynchronous HTTP request, please pass async_req=True
2858
2729
 
2859
- >>> thread = api.get_artifact_by_id_get_with_http_info(id, backend_store_uri, default_artifact_root, async_req=True)
2730
+ >>> thread = api.get_artifact_by_id_get_with_http_info(id, async_req=True)
2860
2731
  >>> result = thread.get()
2861
2732
 
2862
2733
  :param id: (required)
2863
2734
  :type id: str
2864
- :param backend_store_uri:
2865
- :type backend_store_uri: object
2866
- :param default_artifact_root:
2867
- :type default_artifact_root: object
2868
2735
  :param async_req: Whether to execute the request asynchronously.
2869
2736
  :type async_req: bool, optional
2870
2737
  :param _preload_content: if False, the ApiResponse.data will
@@ -2892,7 +2759,7 @@ class MlfoundryArtifactsApi:
2892
2759
 
2893
2760
  _params = locals()
2894
2761
 
2895
- _all_params = ["id", "backend_store_uri", "default_artifact_root"]
2762
+ _all_params = ["id"]
2896
2763
  _all_params.extend(
2897
2764
  [
2898
2765
  "async_req",
@@ -2925,14 +2792,6 @@ class MlfoundryArtifactsApi:
2925
2792
  if _params.get("id") is not None: # noqa: E501
2926
2793
  _query_params.append(("id", _params["id"]))
2927
2794
 
2928
- if _params.get("backend_store_uri") is not None: # noqa: E501
2929
- _query_params.append(("backend_store_uri", _params["backend_store_uri"]))
2930
-
2931
- if _params.get("default_artifact_root") is not None: # noqa: E501
2932
- _query_params.append(
2933
- ("default_artifact_root", _params["default_artifact_root"])
2934
- )
2935
-
2936
2795
  # process the header parameters
2937
2796
  _header_params = dict(_params.get("_headers", {}))
2938
2797
  # process the form parameters
@@ -2974,26 +2833,18 @@ class MlfoundryArtifactsApi:
2974
2833
 
2975
2834
  @validate_arguments
2976
2835
  def get_artifact_version_by_fqn_get(
2977
- self,
2978
- fqn: StrictStr,
2979
- backend_store_uri: Optional[Any] = None,
2980
- default_artifact_root: Optional[Any] = None,
2981
- **kwargs,
2836
+ self, fqn: StrictStr, **kwargs
2982
2837
  ) -> ArtifactVersionResponseDto: # noqa: E501
2983
2838
  """Get Artifact Version By Fqn # noqa: E501
2984
2839
 
2985
2840
  This method makes a synchronous HTTP request by default. To make an
2986
2841
  asynchronous HTTP request, please pass async_req=True
2987
2842
 
2988
- >>> thread = api.get_artifact_version_by_fqn_get(fqn, backend_store_uri, default_artifact_root, async_req=True)
2843
+ >>> thread = api.get_artifact_version_by_fqn_get(fqn, async_req=True)
2989
2844
  >>> result = thread.get()
2990
2845
 
2991
2846
  :param fqn: (required)
2992
2847
  :type fqn: str
2993
- :param backend_store_uri:
2994
- :type backend_store_uri: object
2995
- :param default_artifact_root:
2996
- :type default_artifact_root: object
2997
2848
  :param async_req: Whether to execute the request asynchronously.
2998
2849
  :type async_req: bool, optional
2999
2850
  :param _request_timeout: timeout setting for this request.
@@ -3009,32 +2860,22 @@ class MlfoundryArtifactsApi:
3009
2860
  if "_preload_content" in kwargs:
3010
2861
  message = "Error! Please call the get_artifact_version_by_fqn_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
3011
2862
  raise ValueError(message)
3012
- return self.get_artifact_version_by_fqn_get_with_http_info(
3013
- fqn, backend_store_uri, default_artifact_root, **kwargs
3014
- ) # noqa: E501
2863
+ return self.get_artifact_version_by_fqn_get_with_http_info(fqn, **kwargs) # noqa: E501
3015
2864
 
3016
2865
  @validate_arguments
3017
2866
  def get_artifact_version_by_fqn_get_with_http_info(
3018
- self,
3019
- fqn: StrictStr,
3020
- backend_store_uri: Optional[Any] = None,
3021
- default_artifact_root: Optional[Any] = None,
3022
- **kwargs,
2867
+ self, fqn: StrictStr, **kwargs
3023
2868
  ) -> ApiResponse: # noqa: E501
3024
2869
  """Get Artifact Version By Fqn # noqa: E501
3025
2870
 
3026
2871
  This method makes a synchronous HTTP request by default. To make an
3027
2872
  asynchronous HTTP request, please pass async_req=True
3028
2873
 
3029
- >>> thread = api.get_artifact_version_by_fqn_get_with_http_info(fqn, backend_store_uri, default_artifact_root, async_req=True)
2874
+ >>> thread = api.get_artifact_version_by_fqn_get_with_http_info(fqn, async_req=True)
3030
2875
  >>> result = thread.get()
3031
2876
 
3032
2877
  :param fqn: (required)
3033
2878
  :type fqn: str
3034
- :param backend_store_uri:
3035
- :type backend_store_uri: object
3036
- :param default_artifact_root:
3037
- :type default_artifact_root: object
3038
2879
  :param async_req: Whether to execute the request asynchronously.
3039
2880
  :type async_req: bool, optional
3040
2881
  :param _preload_content: if False, the ApiResponse.data will
@@ -3062,7 +2903,7 @@ class MlfoundryArtifactsApi:
3062
2903
 
3063
2904
  _params = locals()
3064
2905
 
3065
- _all_params = ["fqn", "backend_store_uri", "default_artifact_root"]
2906
+ _all_params = ["fqn"]
3066
2907
  _all_params.extend(
3067
2908
  [
3068
2909
  "async_req",
@@ -3095,14 +2936,6 @@ class MlfoundryArtifactsApi:
3095
2936
  if _params.get("fqn") is not None: # noqa: E501
3096
2937
  _query_params.append(("fqn", _params["fqn"]))
3097
2938
 
3098
- if _params.get("backend_store_uri") is not None: # noqa: E501
3099
- _query_params.append(("backend_store_uri", _params["backend_store_uri"]))
3100
-
3101
- if _params.get("default_artifact_root") is not None: # noqa: E501
3102
- _query_params.append(
3103
- ("default_artifact_root", _params["default_artifact_root"])
3104
- )
3105
-
3106
2939
  # process the header parameters
3107
2940
  _header_params = dict(_params.get("_headers", {}))
3108
2941
  # process the form parameters
@@ -3144,26 +2977,18 @@ class MlfoundryArtifactsApi:
3144
2977
 
3145
2978
  @validate_arguments
3146
2979
  def get_artifact_version_by_id_get(
3147
- self,
3148
- id: StrictStr,
3149
- backend_store_uri: Optional[Any] = None,
3150
- default_artifact_root: Optional[Any] = None,
3151
- **kwargs,
2980
+ self, id: StrictStr, **kwargs
3152
2981
  ) -> ArtifactVersionResponseDto: # noqa: E501
3153
2982
  """Get Artifact Version By Id # noqa: E501
3154
2983
 
3155
2984
  This method makes a synchronous HTTP request by default. To make an
3156
2985
  asynchronous HTTP request, please pass async_req=True
3157
2986
 
3158
- >>> thread = api.get_artifact_version_by_id_get(id, backend_store_uri, default_artifact_root, async_req=True)
2987
+ >>> thread = api.get_artifact_version_by_id_get(id, async_req=True)
3159
2988
  >>> result = thread.get()
3160
2989
 
3161
2990
  :param id: (required)
3162
2991
  :type id: str
3163
- :param backend_store_uri:
3164
- :type backend_store_uri: object
3165
- :param default_artifact_root:
3166
- :type default_artifact_root: object
3167
2992
  :param async_req: Whether to execute the request asynchronously.
3168
2993
  :type async_req: bool, optional
3169
2994
  :param _request_timeout: timeout setting for this request.
@@ -3179,32 +3004,22 @@ class MlfoundryArtifactsApi:
3179
3004
  if "_preload_content" in kwargs:
3180
3005
  message = "Error! Please call the get_artifact_version_by_id_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
3181
3006
  raise ValueError(message)
3182
- return self.get_artifact_version_by_id_get_with_http_info(
3183
- id, backend_store_uri, default_artifact_root, **kwargs
3184
- ) # noqa: E501
3007
+ return self.get_artifact_version_by_id_get_with_http_info(id, **kwargs) # noqa: E501
3185
3008
 
3186
3009
  @validate_arguments
3187
3010
  def get_artifact_version_by_id_get_with_http_info(
3188
- self,
3189
- id: StrictStr,
3190
- backend_store_uri: Optional[Any] = None,
3191
- default_artifact_root: Optional[Any] = None,
3192
- **kwargs,
3011
+ self, id: StrictStr, **kwargs
3193
3012
  ) -> ApiResponse: # noqa: E501
3194
3013
  """Get Artifact Version By Id # noqa: E501
3195
3014
 
3196
3015
  This method makes a synchronous HTTP request by default. To make an
3197
3016
  asynchronous HTTP request, please pass async_req=True
3198
3017
 
3199
- >>> thread = api.get_artifact_version_by_id_get_with_http_info(id, backend_store_uri, default_artifact_root, async_req=True)
3018
+ >>> thread = api.get_artifact_version_by_id_get_with_http_info(id, async_req=True)
3200
3019
  >>> result = thread.get()
3201
3020
 
3202
3021
  :param id: (required)
3203
3022
  :type id: str
3204
- :param backend_store_uri:
3205
- :type backend_store_uri: object
3206
- :param default_artifact_root:
3207
- :type default_artifact_root: object
3208
3023
  :param async_req: Whether to execute the request asynchronously.
3209
3024
  :type async_req: bool, optional
3210
3025
  :param _preload_content: if False, the ApiResponse.data will
@@ -3232,7 +3047,7 @@ class MlfoundryArtifactsApi:
3232
3047
 
3233
3048
  _params = locals()
3234
3049
 
3235
- _all_params = ["id", "backend_store_uri", "default_artifact_root"]
3050
+ _all_params = ["id"]
3236
3051
  _all_params.extend(
3237
3052
  [
3238
3053
  "async_req",
@@ -3265,14 +3080,6 @@ class MlfoundryArtifactsApi:
3265
3080
  if _params.get("id") is not None: # noqa: E501
3266
3081
  _query_params.append(("id", _params["id"]))
3267
3082
 
3268
- if _params.get("backend_store_uri") is not None: # noqa: E501
3269
- _query_params.append(("backend_store_uri", _params["backend_store_uri"]))
3270
-
3271
- if _params.get("default_artifact_root") is not None: # noqa: E501
3272
- _query_params.append(
3273
- ("default_artifact_root", _params["default_artifact_root"])
3274
- )
3275
-
3276
3083
  # process the header parameters
3277
3084
  _header_params = dict(_params.get("_headers", {}))
3278
3085
  # process the form parameters
@@ -3319,8 +3126,6 @@ class MlfoundryArtifactsApi:
3319
3126
  artifact_name: StrictStr,
3320
3127
  version: Optional[StrictInt] = None,
3321
3128
  artifact_type: Optional[ArtifactType] = None,
3322
- backend_store_uri: Optional[Any] = None,
3323
- default_artifact_root: Optional[Any] = None,
3324
3129
  **kwargs,
3325
3130
  ) -> ArtifactVersionResponseDto: # noqa: E501
3326
3131
  """Get Artifact Version By Name # noqa: E501
@@ -3329,7 +3134,7 @@ class MlfoundryArtifactsApi:
3329
3134
  This method makes a synchronous HTTP request by default. To make an
3330
3135
  asynchronous HTTP request, please pass async_req=True
3331
3136
 
3332
- >>> thread = api.get_artifact_version_by_name_get(experiment_id, artifact_name, version, artifact_type, backend_store_uri, default_artifact_root, async_req=True)
3137
+ >>> thread = api.get_artifact_version_by_name_get(experiment_id, artifact_name, version, artifact_type, async_req=True)
3333
3138
  >>> result = thread.get()
3334
3139
 
3335
3140
  :param experiment_id: (required)
@@ -3340,10 +3145,6 @@ class MlfoundryArtifactsApi:
3340
3145
  :type version: int
3341
3146
  :param artifact_type:
3342
3147
  :type artifact_type: ArtifactType
3343
- :param backend_store_uri:
3344
- :type backend_store_uri: object
3345
- :param default_artifact_root:
3346
- :type default_artifact_root: object
3347
3148
  :param async_req: Whether to execute the request asynchronously.
3348
3149
  :type async_req: bool, optional
3349
3150
  :param _request_timeout: timeout setting for this request.
@@ -3360,13 +3161,7 @@ class MlfoundryArtifactsApi:
3360
3161
  message = "Error! Please call the get_artifact_version_by_name_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
3361
3162
  raise ValueError(message)
3362
3163
  return self.get_artifact_version_by_name_get_with_http_info(
3363
- experiment_id,
3364
- artifact_name,
3365
- version,
3366
- artifact_type,
3367
- backend_store_uri,
3368
- default_artifact_root,
3369
- **kwargs,
3164
+ experiment_id, artifact_name, version, artifact_type, **kwargs
3370
3165
  ) # noqa: E501
3371
3166
 
3372
3167
  @validate_arguments
@@ -3376,8 +3171,6 @@ class MlfoundryArtifactsApi:
3376
3171
  artifact_name: StrictStr,
3377
3172
  version: Optional[StrictInt] = None,
3378
3173
  artifact_type: Optional[ArtifactType] = None,
3379
- backend_store_uri: Optional[Any] = None,
3380
- default_artifact_root: Optional[Any] = None,
3381
3174
  **kwargs,
3382
3175
  ) -> ApiResponse: # noqa: E501
3383
3176
  """Get Artifact Version By Name # noqa: E501
@@ -3386,7 +3179,7 @@ class MlfoundryArtifactsApi:
3386
3179
  This method makes a synchronous HTTP request by default. To make an
3387
3180
  asynchronous HTTP request, please pass async_req=True
3388
3181
 
3389
- >>> thread = api.get_artifact_version_by_name_get_with_http_info(experiment_id, artifact_name, version, artifact_type, backend_store_uri, default_artifact_root, async_req=True)
3182
+ >>> thread = api.get_artifact_version_by_name_get_with_http_info(experiment_id, artifact_name, version, artifact_type, async_req=True)
3390
3183
  >>> result = thread.get()
3391
3184
 
3392
3185
  :param experiment_id: (required)
@@ -3397,10 +3190,6 @@ class MlfoundryArtifactsApi:
3397
3190
  :type version: int
3398
3191
  :param artifact_type:
3399
3192
  :type artifact_type: ArtifactType
3400
- :param backend_store_uri:
3401
- :type backend_store_uri: object
3402
- :param default_artifact_root:
3403
- :type default_artifact_root: object
3404
3193
  :param async_req: Whether to execute the request asynchronously.
3405
3194
  :type async_req: bool, optional
3406
3195
  :param _preload_content: if False, the ApiResponse.data will
@@ -3428,14 +3217,7 @@ class MlfoundryArtifactsApi:
3428
3217
 
3429
3218
  _params = locals()
3430
3219
 
3431
- _all_params = [
3432
- "experiment_id",
3433
- "artifact_name",
3434
- "version",
3435
- "artifact_type",
3436
- "backend_store_uri",
3437
- "default_artifact_root",
3438
- ]
3220
+ _all_params = ["experiment_id", "artifact_name", "version", "artifact_type"]
3439
3221
  _all_params.extend(
3440
3222
  [
3441
3223
  "async_req",
@@ -3477,14 +3259,6 @@ class MlfoundryArtifactsApi:
3477
3259
  if _params.get("artifact_type") is not None: # noqa: E501
3478
3260
  _query_params.append(("artifact_type", _params["artifact_type"].value))
3479
3261
 
3480
- if _params.get("backend_store_uri") is not None: # noqa: E501
3481
- _query_params.append(("backend_store_uri", _params["backend_store_uri"]))
3482
-
3483
- if _params.get("default_artifact_root") is not None: # noqa: E501
3484
- _query_params.append(
3485
- ("default_artifact_root", _params["default_artifact_root"])
3486
- )
3487
-
3488
3262
  # process the header parameters
3489
3263
  _header_params = dict(_params.get("_headers", {}))
3490
3264
  # process the form parameters
@@ -6736,27 +6510,17 @@ class MlfoundryArtifactsApi:
6736
6510
  )
6737
6511
 
6738
6512
  @validate_arguments
6739
- def resolve_agent_get(
6740
- self,
6741
- fqn: StrictStr,
6742
- backend_store_uri: Optional[Any] = None,
6743
- default_artifact_root: Optional[Any] = None,
6744
- **kwargs,
6745
- ) -> ResolveAgentAppResponseDto: # noqa: E501
6513
+ def resolve_agent_get(self, fqn: StrictStr, **kwargs) -> ResolveAgentAppResponseDto: # noqa: E501
6746
6514
  """Resolve Agent # noqa: E501
6747
6515
 
6748
6516
  This method makes a synchronous HTTP request by default. To make an
6749
6517
  asynchronous HTTP request, please pass async_req=True
6750
6518
 
6751
- >>> thread = api.resolve_agent_get(fqn, backend_store_uri, default_artifact_root, async_req=True)
6519
+ >>> thread = api.resolve_agent_get(fqn, async_req=True)
6752
6520
  >>> result = thread.get()
6753
6521
 
6754
6522
  :param fqn: (required)
6755
6523
  :type fqn: str
6756
- :param backend_store_uri:
6757
- :type backend_store_uri: object
6758
- :param default_artifact_root:
6759
- :type default_artifact_root: object
6760
6524
  :param async_req: Whether to execute the request asynchronously.
6761
6525
  :type async_req: bool, optional
6762
6526
  :param _request_timeout: timeout setting for this request.
@@ -6772,32 +6536,20 @@ class MlfoundryArtifactsApi:
6772
6536
  if "_preload_content" in kwargs:
6773
6537
  message = "Error! Please call the resolve_agent_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
6774
6538
  raise ValueError(message)
6775
- return self.resolve_agent_get_with_http_info(
6776
- fqn, backend_store_uri, default_artifact_root, **kwargs
6777
- ) # noqa: E501
6539
+ return self.resolve_agent_get_with_http_info(fqn, **kwargs) # noqa: E501
6778
6540
 
6779
6541
  @validate_arguments
6780
- def resolve_agent_get_with_http_info(
6781
- self,
6782
- fqn: StrictStr,
6783
- backend_store_uri: Optional[Any] = None,
6784
- default_artifact_root: Optional[Any] = None,
6785
- **kwargs,
6786
- ) -> ApiResponse: # noqa: E501
6542
+ def resolve_agent_get_with_http_info(self, fqn: StrictStr, **kwargs) -> ApiResponse: # noqa: E501
6787
6543
  """Resolve Agent # noqa: E501
6788
6544
 
6789
6545
  This method makes a synchronous HTTP request by default. To make an
6790
6546
  asynchronous HTTP request, please pass async_req=True
6791
6547
 
6792
- >>> thread = api.resolve_agent_get_with_http_info(fqn, backend_store_uri, default_artifact_root, async_req=True)
6548
+ >>> thread = api.resolve_agent_get_with_http_info(fqn, async_req=True)
6793
6549
  >>> result = thread.get()
6794
6550
 
6795
6551
  :param fqn: (required)
6796
6552
  :type fqn: str
6797
- :param backend_store_uri:
6798
- :type backend_store_uri: object
6799
- :param default_artifact_root:
6800
- :type default_artifact_root: object
6801
6553
  :param async_req: Whether to execute the request asynchronously.
6802
6554
  :type async_req: bool, optional
6803
6555
  :param _preload_content: if False, the ApiResponse.data will
@@ -6825,7 +6577,7 @@ class MlfoundryArtifactsApi:
6825
6577
 
6826
6578
  _params = locals()
6827
6579
 
6828
- _all_params = ["fqn", "backend_store_uri", "default_artifact_root"]
6580
+ _all_params = ["fqn"]
6829
6581
  _all_params.extend(
6830
6582
  [
6831
6583
  "async_req",
@@ -6858,14 +6610,6 @@ class MlfoundryArtifactsApi:
6858
6610
  if _params.get("fqn") is not None: # noqa: E501
6859
6611
  _query_params.append(("fqn", _params["fqn"]))
6860
6612
 
6861
- if _params.get("backend_store_uri") is not None: # noqa: E501
6862
- _query_params.append(("backend_store_uri", _params["backend_store_uri"]))
6863
-
6864
- if _params.get("default_artifact_root") is not None: # noqa: E501
6865
- _query_params.append(
6866
- ("default_artifact_root", _params["default_artifact_root"])
6867
- )
6868
-
6869
6613
  # process the header parameters
6870
6614
  _header_params = dict(_params.get("_headers", {}))
6871
6615
  # process the form parameters