diff options
Diffstat (limited to '.venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/cohere.py')
-rw-r--r-- | .venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/cohere.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/cohere.py b/.venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/cohere.py new file mode 100644 index 00000000..2fbe6fbf --- /dev/null +++ b/.venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/cohere.py @@ -0,0 +1,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" |