From 4a52a71956a8d46fcb7294ac71734504bb09bcc2 Mon Sep 17 00:00:00 2001 From: S. Solomon Darnell Date: Fri, 28 Mar 2025 21:52:21 -0500 Subject: two version of R2R are here --- .../huggingface_hub/inference/_providers/cohere.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 .venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/cohere.py (limited to '.venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/cohere.py') 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" -- cgit v1.2.3