Class ChatOpenAI

src/optimizer/textgrad/engine/openai.py:158–196

class ChatOpenAI(BaseOpenAIEngine):
    def __init__(
        self,
        model_string: str = "gpt-3.5-turbo-0613",
        system_prompt: str = BaseOpenAIEngine.DEFAULT_SYSTEM_PROMPT,
        is_multimodal: bool = False,
        base_url: str = None,
        **kwargs,
    ):
        """
        :param model_string:
        :param system_prompt:
        :param base_url: Used to support Ollama
        """
        root = platformdirs.user_cache_dir("textgrad")
        cache_path = os.path.join(root, f"cache_openai_{model_string}.db")

        super().__init__(cache_path, system_prompt, model_string, is_multimodal)

        self.base_url = base_url

        if not base_url:
            if os.getenv("OPENAI_API_KEY") is None:
                raise ValueError(
                    "Please set the OPENAI_API_KEY environment variable if you'd like to use OpenAI models."
                )

            # Check if OPENAI_API_BASE is set in environment variables
            api_base = os.getenv("OPENAI_API_BASE")
            if api_base:
                self.client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"), base_url=api_base)
            else:
                self.client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
        elif base_url and base_url == OLLAMA_BASE_URL:
            self.client = OpenAI(base_url=base_url, api_key="ollama")
        else:
            raise ValueError(
                "Invalid base URL provided. Please use the default OLLAMA base URL or None."
            )


class AzureChatOpenAI(BaseOpenAIEngine):