内置模型表单新增 EB4 Turbo (#837)

baidubce · Oct 16, 2024 · b45bd1e · b45bd1e
1 parent 9c7d006
commit b45bd1e
Showing 1 changed file with 48 additions and 0 deletions.
diff --git a/python/qianfan/resources/llm/chat_completion.py b/python/qianfan/resources/llm/chat_completion.py
@@ -125,6 +125,54 @@ def _local_models(self) -> Dict[str, QfLLMInfo]:
                 input_price_per_1k_tokens=0.12,
                 output_price_per_1k_tokens=0.12,
             ),
+            "ERNIE-4.0-Turbo-8K-Latest": QfLLMInfo(
+                endpoint="/chat/ernie-4.0-turbo-8k-latest",
+                required_keys={"messages"},
+                optional_keys={
+                    "stream",
+                    "temperature",
+                    "top_p",
+                    "penalty_score",
+                    "user_id",
+                    "system",
+                    "stop",
+                    "enable_system_memory",
+                    "system_memory_id",
+                    "disable_search",
+                    "enable_citation",
+                    "enable_trace",
+                    "max_output_tokens",
+                    "response_format",
+                },
+                max_input_chars=20000,
+                max_input_tokens=5120,
+                input_price_per_1k_tokens=0.02,
+                output_price_per_1k_tokens=0.06,
+            ),
+            "ERNIE-4.0-Turbo-8K-Preview": QfLLMInfo(
+                endpoint="/chat/ernie-4.0-turbo-8k-preview",
+                required_keys={"messages"},
+                optional_keys={
+                    "stream",
+                    "temperature",
+                    "top_p",
+                    "penalty_score",
+                    "user_id",
+                    "system",
+                    "stop",
+                    "enable_system_memory",
+                    "system_memory_id",
+                    "disable_search",
+                    "enable_citation",
+                    "enable_trace",
+                    "max_output_tokens",
+                    "response_format",
+                },
+                max_input_chars=20000,
+                max_input_tokens=5120,
+                input_price_per_1k_tokens=0.02,
+                output_price_per_1k_tokens=0.06,
+            ),
             "ERNIE-Lite-AppBuilder-8K-0614": QfLLMInfo(
                 endpoint="/chat/ai_apaas_lite",
                 required_keys={"messages"},