Skip to content

Commit

Permalink
chore(llm): fix azure token provider signature
Browse files Browse the repository at this point in the history
  • Loading branch information
mspronesti committed Dec 10, 2023
1 parent 250c5fe commit 297b3a9
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions pandasai/llm/azure_openai.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
"""

import os
from typing import Any, Dict, Optional, Union
from typing import Any, Dict, Optional, Union, Callable

import openai
from ..helpers import load_dotenv
Expand All @@ -39,7 +39,7 @@ class AzureOpenAI(BaseOpenAI):
"""A function that returns an Azure Active Directory token.
Will be invoked on every request.
"""
azure_ad_token_provider: Union[str, None] = None
azure_ad_token_provider: Union[Callable[[], str], None] = None
deployment_name: str
api_version: str = ""
"""Legacy, for openai<1.0.0 support."""
Expand All @@ -52,7 +52,7 @@ def __init__(
api_token: Optional[str] = None,
azure_endpoint: Union[str, None] = None,
azure_ad_token: Union[str, None] = None,
azure_ad_token_provider: Union[str, None] = None,
azure_ad_token_provider: Union[Callable[[], str], None] = None,
api_base: Optional[str] = None,
api_version: Optional[str] = None,
deployment_name: str = None,
Expand Down

0 comments on commit 297b3a9

Please sign in to comment.