싱글턴

import google.generativeai as genai
import os
model = genai.GenerativeModel('gemini-pro')
response = model.generate_content("너가 누군지 한 문장으로 설명해줘")

print(response.text)

Untitled

멀티턴

import google.generativeai as genai 

model = genai.GenerativeModel('gemini-pro')
chat_session = model.start_chat(history=[])

while True:
    user_query = input("[사용자]: ")
    
    if user_query.lower() == '종료':
        break
    
    response = chat_session.send_message(user_query)
    print(f'[모델]: {response.text}')

Untitled

위와 같이 ChatSession 객체를 사용하는 것이 간편하고, 불필요한 오버헤드도 줄일 수 있다.