(gemini) gemini API OpenAI SDK 호환 가능

gemini API를 OpenAI SDK를 통해 활용 가능

documentation :  https://ai.google.dev/gemini-api/docs/openai?hl=ko

 

from openai import OpenAI

def _request_google(messages, model: str):
    client = OpenAI(
        api_key=os.getenv("GOOGLE_API_KEY"),
        base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
    )
    return client.chat.completions.create(model=model, messages=messages)


def request_google(
    message: str, model: str = "gemini-2.0-flash", system_instruction: str = None
):
    messages = []
    if system_instruction:
        messages = [{"role" : "system", "content": system_instruction}]

    messages.append({"role": "user", "content": message})

    return _request_google(messages, model).choices[0].message.content