Fix AzureOpenAIModel.

ailabteam · Jul 15, 2024 · 97ca850 · 97ca850
1 parent 9cf9eef
commit 97ca850
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/knowledge_storm/lm.py b/knowledge_storm/lm.py
@@ -109,8 +109,8 @@ class AzureOpenAIModel(dspy.AzureOpenAI):
  """A wrapper class for dspy.AzureOpenAI."""
  def __init__(
  self,
- api_base: str,
- api_version: str,
+ api_base: Optional[str] = None,
+ api_version: Optional[str] = None,
  model: str = "gpt-3.5-turbo-instruct",
  api_key: Optional[str] = None,
  model_type: Literal["chat", "text"] = "chat",
@@ -143,8 +143,6 @@ def get_usage_and_reset(self):
  return usage
 
 
-
-
 class ClaudeModel(dspy.dsp.modules.lm.LM):
  """Copied from dspy/dsp/modules/anthropic.py with the addition of tracking token usage."""