@@ -18,7 +18,7 @@ async def create_openai_chat_client(
1818 azure_deployment = os .environ ["AZURE_OPENAI_CHAT_DEPLOYMENT" ]
1919 if api_key := os .getenv ("AZURE_OPENAI_KEY" ):
2020 logger .info (
21- "Setting up Azure OpenAI client using API key, endpoint %s, deployment %s" ,
21+ "Setting up Azure OpenAI client for chat completions using API key, endpoint %s, deployment %s" ,
2222 azure_endpoint ,
2323 azure_deployment ,
2424 )
@@ -30,7 +30,7 @@ async def create_openai_chat_client(
3030 )
3131 else :
3232 logger .info (
33- "Setting up Azure OpenAI client using Azure Identity, endpoint %s, deployment %s" ,
33+ "Setting up Azure OpenAI client for chat completions using Azure Identity, endpoint %s, deployment %s" ,
3434 azure_endpoint ,
3535 azure_deployment ,
3636 )
@@ -44,13 +44,13 @@ async def create_openai_chat_client(
4444 azure_ad_token_provider = token_provider ,
4545 )
4646 elif OPENAI_CHAT_HOST == "ollama" :
47- logger .info ("Setting up OpenAI client using Ollama" )
47+ logger .info ("Setting up OpenAI client for chat completions using Ollama" )
4848 openai_chat_client = openai .AsyncOpenAI (
4949 base_url = os .getenv ("OLLAMA_ENDPOINT" ),
5050 api_key = "nokeyneeded" ,
5151 )
5252 else :
53- logger .info ("Setting up OpenAI client using OpenAI.com API key" )
53+ logger .info ("Setting up OpenAI client for chat completions using OpenAI.com API key" )
5454 openai_chat_client = openai .AsyncOpenAI (api_key = os .getenv ("OPENAICOM_KEY" ))
5555
5656 return openai_chat_client
@@ -66,15 +66,23 @@ async def create_openai_embed_client(
6666 azure_endpoint = os .environ ["AZURE_OPENAI_ENDPOINT" ]
6767 azure_deployment = os .environ ["AZURE_OPENAI_EMBED_DEPLOYMENT" ]
6868 if api_key := os .getenv ("AZURE_OPENAI_KEY" ):
69- logger .info ("Authenticating to Azure OpenAI using API key..." )
69+ logger .info (
70+ "Setting up Azure OpenAI client for embeddings using API key, endpoint %s, deployment %s" ,
71+ azure_endpoint ,
72+ azure_deployment ,
73+ )
7074 openai_embed_client = openai .AsyncAzureOpenAI (
7175 api_version = api_version ,
7276 azure_endpoint = azure_endpoint ,
7377 azure_deployment = azure_deployment ,
7478 api_key = api_key ,
7579 )
7680 else :
77- logger .info ("Authenticating to Azure OpenAI Embedding using Azure Identity..." )
81+ logger .info (
82+ "Setting up Azure OpenAI client for embeddings using Azure Identity, endpoint %s, deployment %s" ,
83+ azure_endpoint ,
84+ azure_deployment ,
85+ )
7886 token_provider = azure .identity .get_bearer_token_provider (
7987 azure_credential , "https://cognitiveservices.azure.com/.default"
8088 )
@@ -85,12 +93,12 @@ async def create_openai_embed_client(
8593 azure_ad_token_provider = token_provider ,
8694 )
8795 elif OPENAI_EMBED_HOST == "ollama" :
88- logger .info ("Authenticating to OpenAI using Ollama... " )
96+ logger .info ("Setting up OpenAI client for embeddings using Ollama" )
8997 openai_embed_client = openai .AsyncOpenAI (
9098 base_url = os .getenv ("OLLAMA_ENDPOINT" ),
9199 api_key = "nokeyneeded" ,
92100 )
93101 else :
94- logger .info ("Authenticating to OpenAI using OpenAI.com API key... " )
102+ logger .info ("Setting up OpenAI client for embeddings using OpenAI.com API key" )
95103 openai_embed_client = openai .AsyncOpenAI (api_key = os .getenv ("OPENAICOM_KEY" ))
96104 return openai_embed_client
0 commit comments