(feat) add setting input_type for cohere

2025-04-27 03:34:10 +00:00 · 2023-11-02 10:16:35 -07:00 · 2023-11-02 10:16:35 -07:00 · 36a2266382
commit 36a2266382
parent d100bfdc2e
2 changed files with 7 additions and 0 deletions
--- a/litellm/main.py
+++ b/litellm/main.py
@ -1764,6 +1764,7 @@ def embedding(
            response = cohere.embedding(
                model=model,
                input=input,
+                optional_params=kwargs,
                encoding=encoding,
                api_key=cohere_key,
                logging_obj=logging,