[실습] google gemini 테스트 [python] 10분컷

Topic. google vertext AI gemini 를 테스트해봅니다.

---

1. API key 발급

아래 사이트에서 발급받으세요.

https://makersuite.google.com/app/apikey

로그인 - Google 계정

---

2. 라이브러리 설치

$ pip install google-generativeai

---

3. 소스 작성

import google.generativeai as genai
GOOGLE_API_KEY = '[위에서 발급받은 API KEY]'
genai.configure(api_key=GOOGLE_API_KEY)

# 구글에서 제공하는 Generate AI API 목록 출력
for m in genai.list_models():
    if 'generateContent' in m.supported_generation_methods:
        print(m.name)

model = genai.GenerativeModel('gemini-pro')

prompt = "안녕하세요? 여기에 질문하면 됩니다."

result = ""

response = model.generate_content(prompt, stream=True)
for chunk in response:
    result += chunk.text
    print(chunk.text)
print("[result] : ",result)