From 847a34cc7612b30ae5f33ccfdc08161a63213a4b Mon Sep 17 00:00:00 2001
From: JasonLiuWeihang <138853984+JasonLiuWeihang@users.noreply.github.com>
Date: Wed, 25 Sep 2024 17:18:08 +0800
Subject: [PATCH] Update model_prices_and_context_window.json

add Baidu's llm ERNIE-bot
---
 model_prices_and_context_window.json | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json
index 5aef89d45e..2134f87e94 100644
--- a/model_prices_and_context_window.json
+++ b/model_prices_and_context_window.json
@@ -1,4 +1,16 @@
 {
+    "ERNIE-bot":{
+        "max_tokens": 4096,
+        "max_input_tokens": 128000,
+        "max_output_tokens": 4096,
+        "input_cost_per_token": 0.000005,
+        "output_cost_per_token": 0.000015,
+        "litellm_provider": "Baidu",
+        "mode": "chat",
+        "supports_function_calling": true,
+        "supports_parallel_function_calling": true,
+        "supports_vision": true
+    },
     "sample_spec": {
         "max_tokens": "set to max_output_tokens if provider specifies it. IF not set to max_tokens provider specifies", 
         "max_input_tokens": "max input tokens, if the provider specifies it. if not default to max_tokens",