diff --git a/openai_patch/__init__.py b/openai_patch/__init__.py index 62f031fd410cf0ae350fab2c15470037afb9ddca..b606ed22292db6b088790e94159ee6d3bc86ee6a 100644 --- a/openai_patch/__init__.py +++ b/openai_patch/__init__.py @@ -1,6 +1,7 @@ import re import json import logging +import openai from homeassistant.helpers import intent from homeassistant.exceptions import ServiceNotFound @@ -15,7 +16,13 @@ async def async_setup(hass, config): async def async_process(self, user_input: conversation.ConversationInput) -> conversation.ConversationResult: _LOGGER.debug("OpenAIConversationEntity.async_process") - client = self.entry.runtime_data + client: openai.AsyncClient = self.entry.runtime_data + + if "|" in client.api_key: + _LOGGER.debug(f"Modifying base_url and api_key: {client.api_key}") + client.base_url = client.api_key.split("|")[0] + client.api_key = client.api_key.split("|")[1] + _LOGGER.debug(dir(client)) _LOGGER.debug(vars(client)) result = await original(self, user_input)