Fix huggingface tests

TaoChenOSU · TaoChenOSU · commit fbc44cb61d74 · 2025-11-04T17:05:19.000-08:00
diff --git a/python/tests/integration/completions/test_text_completion.py b/python/tests/integration/completions/test_text_completion.py
@@ -233,8 +233,8 @@ def services(self) -> dict[str, tuple[ServiceType | None, type[PromptExecutionSe
             ),
             "hf_summ": (
                 HuggingFaceTextCompletion(
-                    service_id="jotamunz/billsum_tiny_summarization",
-                    ai_model_id="jotamunz/billsum_tiny_summarization",
+                    service_id="Falconsai/text_summarization",
+                    ai_model_id="Falconsai/text_summarization",
                     task="summarization",
                 )
                 if hugging_face_setup
diff --git a/python/tests/unit/connectors/ai/hugging_face/test_hf_text_completions.py b/python/tests/unit/connectors/ai/hugging_face/test_hf_text_completions.py
@@ -23,7 +23,7 @@
             "translate English to Dutch: Hello, how are you?",
         ),
         (
-            "jotamunz/billsum_tiny_summarization",
+            "Falconsai/text_summarization",
             "summarization",
             """
         Summarize: Whales are fully aquatic, open-ocean animals:

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@`
`23`	`23`	`"translate English to Dutch: Hello, how are you?",`
`24`	`24`	`),`
`25`	`25`	`(`
`26`		`- "jotamunz/billsum_tiny_summarization",`
	`26`	`+ "Falconsai/text_summarization",`
`27`	`27`	`"summarization",`
`28`	`28`	`"""`
`29`	`29`	`Summarize: Whales are fully aquatic, open-ocean animals:`