blob: 2fbe6fbf77ac8ccfa248692abe4c3c96460342fd (
about) (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
from huggingface_hub.inference._providers._common import (
BaseConversationalTask,
)
_PROVIDER = "cohere"
_BASE_URL = "https://api.cohere.com"
class CohereConversationalTask(BaseConversationalTask):
def __init__(self):
super().__init__(provider=_PROVIDER, base_url=_BASE_URL)
def _prepare_route(self, mapped_model: str) -> str:
return "/compatibility/v1/chat/completions"
|
