Spaces:
Runtime error
Runtime error
File size: 2,007 Bytes
105b369 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
from os import getenv
from typing import Optional, Dict, Any
from phi.utils.log import logger
from phi.llm.openai.like import OpenAILike
try:
from openai import AzureOpenAI as AzureOpenAIClient
except ImportError:
logger.error("`azure openai` not installed")
raise
class AzureOpenAIChat(OpenAILike):
name: str = "AzureOpenAIChat"
model: str
api_key: Optional[str] = getenv("AZURE_OPENAI_API_KEY")
api_version: str = getenv("AZURE_OPENAI_API_VERSION", "2023-12-01-preview")
azure_endpoint: Optional[str] = getenv("AZURE_OPENAI_ENDPOINT")
azure_deployment: Optional[str] = getenv("AZURE_DEPLOYMENT")
base_url: Optional[str] = None
azure_ad_token: Optional[str] = None
azure_ad_token_provider: Optional[Any] = None
openai_client: Optional[AzureOpenAIClient] = None
@property
def get_client(self) -> AzureOpenAIClient: # type: ignore
if self.openai_client:
return self.openai_client
_client_params: Dict[str, Any] = {}
if self.api_key:
_client_params["api_key"] = self.api_key
if self.api_version:
_client_params["api_version"] = self.api_version
if self.organization:
_client_params["organization"] = self.organization
if self.azure_endpoint:
_client_params["azure_endpoint"] = self.azure_endpoint
if self.azure_deployment:
_client_params["azure_deployment"] = self.azure_deployment
if self.base_url:
_client_params["base_url"] = self.base_url
if self.azure_ad_token:
_client_params["azure_ad_token"] = self.azure_ad_token
if self.azure_ad_token_provider:
_client_params["azure_ad_token_provider"] = self.azure_ad_token_provider
if self.http_client:
_client_params["http_client"] = self.http_client
if self.client_params:
_client_params.update(self.client_params)
return AzureOpenAIClient(**_client_params)
|