Gemini API 시작하기 on VSCode

Hwoo·2024년 6월 22일
1

GoogleAI

목록 보기
1/1

기본 요건
이 빠른 시작은 Google Colab에서 실행할 수 있으며, 브라우저에서 직접 이 노트북을 실행하고 추가 환경 구성이 필요하지 않습니다.

또는 이 빠른 시작을 로컬에서 완료하려면 개발 환경이 다음 요구사항을 충족하는지 확인하세요.

Python 3.9 이상
노트북 실행을 위한 jupyter 설치

가상환경 구성을 위한 아나콘다 설치,
local 환경에서 google-colab을 설치 순으로 진행한다.

1. 아나콘다 설치

wiki백과 :
아나콘다는 패키지 관리와 디플로이를 단순케 할 목적으로 과학 계산을 위한 파이썬과 R 프로그래밍 언어의 자유-오픈 소스 배포판이다. 패키지 버전들은 패키지 관리 시스템 conda를 통해 관리된다.

% brew install --cask anaconda
% cd /opt/homebrew/anaconda3/bin/
% ./conda init zsh
no change     /opt/homebrew/anaconda3/condabin/conda
no change     /opt/homebrew/anaconda3/bin/conda
no change     /opt/homebrew/anaconda3/bin/conda-env
no change     /opt/homebrew/anaconda3/bin/activate
no change     /opt/homebrew/anaconda3/bin/deactivate
no change     /opt/homebrew/anaconda3/etc/profile.d/conda.sh
no change     /opt/homebrew/anaconda3/etc/fish/conf.d/conda.fish
no change     /opt/homebrew/anaconda3/shell/condabin/Conda.psm1
modified      /opt/homebrew/anaconda3/shell/condabin/conda-hook.ps1
no change     /opt/homebrew/anaconda3/lib/python3.11/site-packages/xontrib/conda.xsh
no change     /opt/homebrew/anaconda3/etc/profile.d/conda.csh
modified      /Users/hwooair/.zshrc
==> For changes to take effect, close and re-open your current shell. <==

아나콘다가 제대로 설치되고 PATH(환경변수가) 적용되면 가상환경이 기본적으로 활성화가 되어, (base) 가 붙는다.

% source ~/.zshrc
(base) %
(base) % conda --version
conda 24.5.0

기본 가상환경을 false로 변경

% conda config --set auto_activate_base false

가상환경 해제

% conda deactivate

가상환경 리스트확인

% conda info --envs

가상환경 생성

conda create -n [가상환경 이름] python=[파이썬 버전]

% conda create -n test_conda python=3.9.6

가상환경에서 jupyter 커널 생성하기

# 패키지 설치 확인
% pip freeze |grep -i jupyter
% pip freeze |grep -i ipykernel
# 설치가 안된 경우에 설치
% pip install jupyter
% pip install ipykernel
# python -m ipykernel install --user --name [conda name] --display-name "[jupyter kernel name]
python -m ipykernel install --user --name test_conda --display-name "test_conda"

VSCode 에서 jupyter kernel 연결

  • ipynb 새파일 생성
  • 오른쪽 위에 연결된 커널 클릭 - 다른 커널 선택 - Jupyter 커널 선택



참고 : https://hyunki99.tistory.com/90

2. google-colab tutorial on VScode

https://ai.google.dev/gemini-api/docs/get-started/tutorial?lang=python&hl=ko

이제 vscode 위에서 google-colab을 써보자.

위 링크에서 tutorial을 순서대로 진행한다.

local 환경에서 %pip install google-colab을 시도하면 실패한다.
아래 방법으로 진행해보자.

!pip install -e git+https://github.com/googlecolab/colabtools#egg=google.colab

pip install -e git+https://github.com/googlecolab/colabtools#egg=google.colab
ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
jupyter-console 6.6.3 requires ipykernel>=6.14, but you have ipykernel 5.5.6 which is incompatible.
jupyterlab 4.2.2 requires ipykernel>=6.5.0, but you have ipykernel 5.5.6 which is incompatible.
pip install ipykernel --upgrade

커널 재시작후 다시 실행하면 설치 done.

  Running setup.py develop for google.colab
Successfully installed google.colab

아나콘다로 설치하는 방법도 있다.
(https://anaconda.org/conda-forge/google-colab)

conda install conda-forge::google-colab

https://stackoverflow.com/questions/77110771/error-when-pip-install-google-colab-use-pep517

이제 google.colab을 가져와보자.

from google.colab import userdata

아래와 같은 에러가 난다.

AttributeError                            Traceback (most recent call last)
<ipython-input-2-cdbfe1e84901> in <cell line: 2>()
      1 # Used to securely store your API key
----> 2 from google.colab import userdata

~/study-ai/GeminiAI/src/google-colab/google/colab/__init__.py in <module>
     17 from google.colab import _import_magics
     18 from google.colab import _installation_commands
---> 19 from google.colab import _interactive_table_hint_button
     20 from google.colab import _reprs
     21 from google.colab import _shell_customizations

~/study-ai/GeminiAI/src/google-colab/google/colab/_interactive_table_hint_button.py in <module>
     16 import uuid as _uuid
     17 import weakref as _weakref
---> 18 from google.colab import data_table as _data_table
     19 from google.colab import output as _output
     20 import IPython as _IPython

~/study-ai/GeminiAI/src/google-colab/google/colab/data_table.py in <module>
    317 
    318 
--> 319 class _JavascriptModuleFormatter(_IPython.core.formatters.BaseFormatter):
    320   format_type = _traitlets.Unicode(_JAVASCRIPT_MODULE_MIME_TYPE)
...
--> 320   format_type = _traitlets.Unicode(_JAVASCRIPT_MODULE_MIME_TYPE)
    321   print_method = _traitlets.ObjectName('_repr_javascript_module_')
    322 

인터프리터 환경에서의 문제로, jupyter notebook이나 colab에서는 나지 않는다고 한다.

아래 파일에서 다음과 같이 수정하자.

~/study-ai/GeminiAI/src/google-colab/google/colab/data_table.py
~/study-ai/GeminiAI/src/google-colab/google/colab/_reprs.py

in data_table.py
...
with _warnings.catch_warnings():
  # Importing via IPython raises a spurious warning, but avoids a version
  # mismatch internally.
  _warnings.simplefilter('ignore')
  #from IPython.utils import traitlets as _traitlets
  # 아래처럼 from IPython.utils를 제거해준다.
  import traitlets as _traitlets
  
in _reprs.py
...
  #from IPython.utils import traitlets
  import traitlets

참고 : https://stackoverflow.com/questions/75666380/attributeerror-module-ipython-utils-traitlets-has-no-attribute-unicode

ModuleNotFoundError: No module named 'PIL'

pillow package 설치로 해결.

pip install pillow

드디어 google.colab을 제대로 가져오는데 성공!
이제부터는 tutorial 진행이 원활하게 된다

GoogleAIStudio 에서 API key값을 생성해서 넣어주자.
https://aistudio.google.com/app/apikey

# Or use `os.getenv('GOOGLE_API_KEY')` to fetch an environment variable.
#GOOGLE_API_KEY=userdata.get('GOOGLE_API_KEY')
GOOGLE_API_KEY="여기에 API Key값을 넣어주면 된다."
genai.configure(api_key=GOOGLE_API_KEY)

Google AI 모델들 리스트를 확인해보자.
여러 버전의 Gemini들을 볼 수 있다.

모델 선택해주고..

prompt를 입력하면 아래처럼 잘 동작한다.

profile
happy coding with my cats

0개의 댓글