API 키 만들기와 SDK 시작

API 키 만들기와 SDK 시작

AI21 API를 호출하려면 먼저 API 키를 만들고, 원하는 언어의 SDK를 설치해야 해요. 파이썬 기준으로 설치부터 첫 호출까지 잡아볼게요.

출처: https://docs.ai21.com/docs/sdk

사전 준비

  1. AI21 홈페이지에서 계정을 만들고 API 키를 발급받아요.
  2. 환경변수에 키를 등록합니다.
export AI21_API_KEY=<YOUR_API_KEY>

SDK 설치

AI21은 Python과 TypeScript를 공식 지원해요. 파이썬 SDK는 ai21 패키지입니다.

pip install ai21

첫 호출 (Chat Completions)

ai21 SDK의 AI21Client를 사용하면 파이썬 코드에서 바로 모델을 호출할 수 있어요.

from ai21 import AI21Client
from ai21.models.chat import ChatMessage

client = AI21Client()

response = client.chat.completions.create(
    model="jamba-1.6-large",
    messages=[ChatMessage(role="user", content="안녕하세요!")],
)
print(response.choices[0].message.content)

SDK는 동기·비동기 호출 모두를 지원해서, asyncio 기반 앱에서는 AsyncAI21Client를 쓸 수 있어요.

더 알아보기

  • https://docs.ai21.com/reference/jamba-1-6-api-ref — REST API 레퍼런스
  • https://docs.ai21.com/docs/sdk — SDK 설치·사용 가이드