@@ -74,20 +74,20 @@ async def run_completion_request(
7474 else :
7575 return message .content
7676
77- async def run_embedding_request (self , batch : list [str ]):
78- token_provider , api_key = self .get_authentication_properties ()
77+ # async def run_embedding_request(self, batch: list[str]):
78+ # token_provider, api_key = self.get_authentication_properties()
7979
80- model_deployment = os .environ ["OpenAI__EmbeddingModel" ]
81- async with AsyncAzureOpenAI (
82- azure_deployment = model_deployment ,
83- api_version = os .environ ["OpenAI__ApiVersion" ],
84- azure_endpoint = os .environ ["OpenAI__Endpoint" ],
85- azure_ad_token_provider = token_provider ,
86- api_key = api_key ,
87- ) as open_ai_client :
88- embeddings = await open_ai_client .embeddings .create (
89- model = os .environ ["OpenAI__EmbeddingModel" ],
90- input = batch ,
91- )
80+ # model_deployment = os.environ["OpenAI__EmbeddingModel"]
81+ # async with AsyncAzureOpenAI(
82+ # azure_deployment=model_deployment,
83+ # api_version=os.environ["OpenAI__ApiVersion"],
84+ # azure_endpoint=os.environ["OpenAI__Endpoint"],
85+ # azure_ad_token_provider=token_provider,
86+ # api_key=api_key,
87+ # ) as open_ai_client:
88+ # embeddings = await open_ai_client.embeddings.create(
89+ # model=os.environ["OpenAI__EmbeddingModel"],
90+ # input=batch,
91+ # )
9292
93- return embeddings
93+ # return embeddings
0 commit comments