파이썬 OpenRouter AI 프롬프트 사용해보기

OpenRouter AI 프롬프트 사용해보기

 

OpenRouter AI는 다양한 AI 모델을 제공하는 플랫폼으로, 개발자들이 쉽게 AI 기능을 자신의 애플리케이션에 통합할 수 있도록 합니다. 이 글에서는 OpenRouter AI를 사용하여 AI 모델을 호출하는 간단한 예제를 설명합니다. 특히, 이 예제에서는 "무엇이 인생의 의미인가?"라는 질문에 대한 AI의 대답을 요청하는 방법을 보여줍니다.

필요한 도구

먼저, requests 라이브러리가 필요합니다. 이 라이브러리가 아직 설치되어 있지 않다면, 다음 명령어로 설치할 수 있습니다:

pip install requests

코드 설명

다음은 OpenRouter AI의 API를 사용하여 AI 모델에 질문을 보내고 답변을 받는 방법을 보여주는 코드입니다.

import requests
import json

# OpenRouter AI의 API 엔드포인트에 POST 요청을 보냅니다.
response = requests.post(
  url="https://openrouter.ai/api/v1/chat/completions",
  headers={
    "Authorization": "Bearer <your_api_key>",
    # "HTTP-Referer": "{YOUR_SITE_URL}", # 선택사항, OpenRouter AI 랭킹에 앱을 포함시키려면 사용합니다.
    # "X-Title": "{YOUR_APP_NAME}", # 선택사항, OpenRouter AI 랭킹에서 표시됩니다.
  },
  data=json.dumps({
    "model": "openrouter/auto", # 모델 선택은 선택사항입니다.
    "messages": [
      {"role": "user", "content": "What is the meaning of life?"}
    ]
  })
)

# 응답을 텍스트로 출력합니다.
print(response.text)

이 코드는 requests.post 메소드를 사용하여 OpenRouter AI의 chat/completions API 엔드포인트에 요청을 보냅니다. Authorization 헤더에는 API를 사용하기 위한 토큰이 포함되어 있어야 합니다. 이 토큰은 OpenRouter AI에서 제공받아야 합니다.

요청 본문에는 질문을 포함하는 messages 배열이 포함되어 있습니다. 이 예제에서는 질문으로 "What is the meaning of life?"를 사용하고 있습니다. 선택사항으로, 특정 모델을 지정할 수 있으며, 이 경우 "openrouter/auto" 모델이 사용되고 있습니다.

사용 예제

위 코드를 실행하면, OpenRouter AI의 AI 모델이 제출된 질문에 대한 답변을 반환합니다. 응답은 JSON 형태로 반환되며, 이를 통해 AI 모델의 대답을 확인할 수 있습니다.

OpenRouter AI를 사용함으로써 개발자는 손쉽게 AI 기능을 자신의 애플리케이션에 통합할 수 있으며, 다양한 AI 모델을 실험해 볼 수 있는 기회를 가질 수 있습니다.