Skip to content

Commit 8e35cdc

Browse files
committed
Fix HfInference conversational (#2985)
* Fix HfInference Conversational * fix test
1 parent a8467cd commit 8e35cdc

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/huggingface_hub/inference/_providers/hf_inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def _prepare_payload_as_bytes(
8282

8383
class HFInferenceConversational(HFInferenceTask):
8484
def __init__(self):
85-
super().__init__("text-generation")
85+
super().__init__("conversational")
8686

8787
def _prepare_payload_as_dict(self, inputs: Any, parameters: Dict, mapped_model: str) -> Optional[Dict]:
8888
payload_model = parameters.get("model") or mapped_model

tests/test_inference_providers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -470,7 +470,7 @@ def test_prepare_request_conversational(self, mocker):
470470
assert (
471471
request.url == "https://router.huggingface.co/hf-inference/models/username/repo_name/v1/chat/completions"
472472
)
473-
assert request.task == "text-generation"
473+
assert request.task == "conversational"
474474
assert request.model == "username/repo_name"
475475
assert request.json == {
476476
"model": "username/repo_name",

0 commit comments

Comments
 (0)