Refresh Open AI API credentials by using Google Cloud authentication

This code sample demonstrates a utility class that refreshes credentials for OpenAI API calls using Google Cloud authentication. The `OpenAI Credentials Refresher` class retrieves and refreshes credentials from Google Cloud when necessary, ensuring the continued availability of a valid API key for OpenAI requests.

Explore further

For detailed documentation that includes this code sample, see the following:

• Call Vertex AI models by using the OpenAI library

Code sample

Python

Before trying this sample, follow the Python setup instructions in the Vertex AI quickstart using client libraries. For more information, see the Vertex AI Python API reference documentation.

To authenticate to Vertex AI, set up Application Default Credentials. For more information, see Set up authentication for a local development environment.

from typing import Any

import google.auth
import google.auth.transport.requests
import openai


class OpenAICredentialsRefresher:
    def __init__(self, **kwargs: Any) -> None:
        # Set a dummy key here
        self.client = openai.OpenAI(**kwargs, api_key="DUMMY")
        self.creds, self.project = google.auth.default(
            scopes=["https://www.googleapis.com/auth/cloud-platform"]
        )

    def __getattr__(self, name: str) -> Any:
        if not self.creds.valid:
            auth_req = google.auth.transport.requests.Request()
            self.creds.refresh(auth_req)

            if not self.creds.valid:
                raise RuntimeError("Unable to refresh auth")

            self.client.api_key = self.creds.token
        return getattr(self.client, name)

What's next

To search and filter code samples for other Google Cloud products, see the Google Cloud sample browser.