From 79178519fb612747ce6b45d27d3b51be095663c8 Mon Sep 17 00:00:00 2001
From: Jaydeep Mor <jaydeep@mbarclabs.com>
Date: Mon, 16 Dec 2024 18:22:16 +0530
Subject: [PATCH] Make azure_api_key parameter optional for init_openai_model
 def

---
 knowledge_storm/storm_wiki/engine.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/knowledge_storm/storm_wiki/engine.py b/knowledge_storm/storm_wiki/engine.py
index 9a94acff..f038a5b5 100644
--- a/knowledge_storm/storm_wiki/engine.py
+++ b/knowledge_storm/storm_wiki/engine.py
@@ -38,8 +38,8 @@ def __init__(self):
     def init_openai_model(
         self,
         openai_api_key: str,
-        azure_api_key: str,
         openai_type: Literal["openai", "azure"],
+        azure_api_key: Optional[str] = None,
         api_base: Optional[str] = None,
         api_version: Optional[str] = None,
         temperature: Optional[float] = 1.0,