임베딩

export VOYAGE_API_KEY="<your secret key>"

pip install -U voyageai

import voyageai

vo = voyageai.Client()
# 이는 자동으로 환경 변수 VOYAGE_API_KEY를 사용합니다.
# 또는 vo = voyageai.Client(api_key="<your secret key>")를 사용할 수 있습니다

texts = ["샘플 텍스트 1", "샘플 텍스트 2"]

result = vo.embed(texts, model="voyage-3", input_type="document")
print(result.embeddings[0])
print(result.embeddings[1])

[0.02012746, 0.01957859, ...]  # "샘플 텍스트 1"의 임베딩
[0.01429677, 0.03077182, ...]  # "샘플 텍스트 2"의 임베딩

curl https://api.voyageai.com/v1/embeddings \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer $VOYAGE_API_KEY" \
  -d '{
    "input": ["샘플 텍스트 1", "샘플 텍스트 2"],
    "model": "voyage-3"
  }'

{
  "object": "list",
  "data": [
    {
      "embedding": [0.02012746, 0.01957859, ...],
      "index": 0
    },
    {
      "embedding": [0.01429677, 0.03077182, ...],
      "index": 1
    }
  ],
  "model": "voyage-3",
  "usage": {
    "total_tokens": 10
  }
}

documents = [
    "지중해식 식단은 생선, 올리브 오일, 채소를 강조하며, 만성 질환을 줄이는 것으로 알려져 있습니다.",
    "식물의 광합성은 빛 에너지를 포도당으로 전환하고 필수적인 산소를 생산합니다.",
    "20세기 혁신은 라디오에서 스마트폰에 이르기까지 전자 기술의 발전을 중심으로 이루어졌습니다.",
    "강은 물, 관개, 수생 생물의 서식지를 제공하며, 생태계에 매우 중요합니다.",
    "Apple의 4분기 재무 결과와 사업 업데이트를 논의하기 위한 컨퍼런스 콜이 2023년 11월 2일 목요일 오후 2시(PT) / 오후 5시(ET)로 예정되어 있습니다.",
    "셰익스피어의 작품인 '햄릿'과 '한 여름 밤의 꿈'은 문학에서 영원히 남을 것입니다."
]

import voyageai

vo = voyageai.Client()

# 문서 임베딩
doc_embds = vo.embed(
    documents, model="voyage-3", input_type="document"
).embeddings

query = "Apple의 컨퍼런스 콜은 언제로 예정되어 있나요?"

import numpy as np

# 쿼리 임베딩
query_embd = vo.embed(
    [query], model="voyage-3", input_type="query"
).embeddings[0]

# 유사도 계산
# Voyage 임베딩은 길이가 1로 정규화되어 있으므로 내적과
# 코사인 유사도는 동일합니다.
similarities = np.dot(doc_embds, query_embd)

retrieved_id = np.argmax(similarities)
print(documents[retrieved_id])

Apple의 4분기 재무 결과와 사업 업데이트를 논의하기 위한 컨퍼런스 콜이 2023년 11월 2일 목요일 오후 2시(PT) / 오후 5시(ET)로 예정되어 있습니다.