Update base.py

run-llama · Oct 9, 2024 · c5601b7 · c5601b7
1 parent 48cac06
commit c5601b7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/llama-index-integrations/llms/llama-index-llms-zhipuai/llama_index/llms/zhipuai/base.py b/llama-index-integrations/llms/llama-index-llms-zhipuai/llama_index/llms/zhipuai/base.py
@@ -116,6 +116,7 @@ class ZhipuAI(FunctionCallingLLM):
     def __init__(
         self,
         model: str,
+        api_key: str,
         temperature: float = 0.95,
         max_tokens: int = 1024,
         timeout: float = DEFAULT_REQUEST_TIMEOUT,