Damien/sagemaker tgi (#2888)

2025-06-13 11:35:42 -07:00
parent b1d2a02f66
commit a7ad73197d
6 changed files with 409 additions and 0 deletions
@@ -490,6 +490,17 @@ async fn test_snowflake_provider() -> Result<()> {
    .await
 }

+#[tokio::test]
+async fn test_sagemaker_tgi_provider() -> Result<()> {
+    test_provider(
+        "SageMakerTgi",
+        &["SAGEMAKER_ENDPOINT_NAME"],
+        None,
+        goose::providers::sagemaker_tgi::SageMakerTgiProvider::default,
+    )
+    .await
+}
+
 // Print the final test report
 #[ctor::dtor]
 fn print_test_report() {