gemini-pro python 에서 사용해 보기

 

gemini-pro python 에서 사용해 보기

 

키발급

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

로그인 - Google 계정

Get API Key 선택합니다.

API 키 만들기 를 클릭합니다.

확인

프로젝트 만든 적이 없으므로 새 프로젝트에서 API 키 만들기 선택합니다.

키가 생성되고 복사합니다.

 

키 설정

위 발급받은 키를 GOOGLE_API_KEY 의 값으로 설정해야합니다.

아래 포스팅을 참고합니다.

2024.06.08 - [코딩/Python_AI] - .env 파일에서 OPENAI_API_KEY 처리하기

 

코딩해서 사용해 보기

* 패키지 설치

pip install langchain-google-genai pillow

 

* 코드

from dotenv import load_dotenv

# GOOGLE_API_KEY
load_dotenv()

from langchain_google_genai import ChatGoogleGenerativeAI

# ChatGoogleGenerativeAI 언어 모델을 "gemini-pro" 모델로 초기화합니다.
llm = ChatGoogleGenerativeAI(model="gemini-pro")
# 언어 모델에 프롬프트를 전달하여 결과를 생성합니다.
result = llm.invoke("AI LLM에 대해서 간략히 설명해 줘")
# 생성된 결과의 내용을 출력합니다.
print(result.content)

 

* 결과