Yandex CloudCompatible with Milvus 2.6.x

이 항목에서는 Milvus에서 Yandex Cloud 임베딩 기능을 구성하고 사용하는 방법을 설명합니다.

임베딩 모델 선택

Milvus는 yc 제공자를 통해 Yandex Cloud AI Studio 텍스트 벡터화 모델을 지원합니다. 함수 매개변수에서 ` model_name `를 Milvus가 호출해야 할 Yandex Cloud 모델 URI로 설정하십시오.

예를 들어, 문서를 위한 Yandex 텍스트 임베딩은 emb://<folder_ID>/text-search-doc/latest 와 같은 모델 URI를 사용하며 256차원 벡터를 반환합니다. 사용 가능한 모델 URI 및 차원에 대해서는 ‘텍스트 벡터화 모델’을 참조하십시오.

인증 정보 구성

Milvus가 임베딩을 요청하려면 Yandex Cloud API 키를 알아야 합니다. API 키는 milvus.yaml 에서 설정하거나 환경 변수를 통해 구성할 수 있습니다.

옵션 1: 구성 파일

API 키를 milvus.yaml 에 저장하고 Yandex Cloud 제공자를 자격 증명 레이블로 지정하십시오.

# milvus.yaml
credential:
  yandex_apikey:
    apikey: <YOUR_YC_API_KEY>

function:
  textEmbedding:
    providers:
      yc:
        credential: yandex_apikey
        # url: https://llm.api.cloud.yandex.net/foundationModels/v1/textEmbedding

옵션 2: 환경 변수

milvus.yaml 에 일치하는 자격 증명이 설정되어 있지 않은 경우, Milvus는 다음 환경 변수에서 Yandex Cloud API 키를 읽어올 수 있습니다:

변수	필수?	설명
MILVUS_YC_API_KEY	예	Milvus 서비스가 Yandex Cloud AI Studio를 호출하는 데 사용하는 Yandex Cloud API 키입니다.

임베딩 함수 사용

인증 정보를 구성한 후, 입력 텍스트 필드와 출력 벡터 필드가 포함된 스키마를 정의하고, 해당 스키마에 Yandex Cloud 임베딩 함수를 추가합니다.

from pymilvus import MilvusClient, DataType, Function, FunctionType

client = MilvusClient(uri="http://localhost:19530")

schema = client.create_schema()
schema.add_field("id", DataType.INT64, is_primary=True, auto_id=False)
schema.add_field("document", DataType.VARCHAR, max_length=9000)
schema.add_field("dense", DataType.FLOAT_VECTOR, dim=256)

text_embedding_function = Function(
    name="yandex_cloud_embedding",
    function_type=FunctionType.TEXTEMBEDDING,
    input_field_names=["document"],
    output_field_names=["dense"],
    params={
        "provider": "yc",
        "model_name": "emb://<folder_ID>/text-search-doc/latest",
        "credential": "yandex_apikey",
        "dim": "256",
    },
)

schema.add_function(text_embedding_function)

Yandex Cloud 전용 매개변수

매개변수	필수 항목인가요?	설명	값 / 예
provider	예	사용할 임베딩 모델 제공자.	"yc"
model_name	예	호출할 Yandex Cloud 모델 URI입니다.	"emb://<folder_ID>/text-search-doc/latest"
credential	아니요	milvus.yaml 의 최상위 credential: 섹션에 정의된 자격 증명의 레이블입니다.	"yandex_apikey"
dim	아니요	출력 벡터의 차원입니다. 설정하는 경우, 이 값은 출력 벡터 필드의 차원과 일치해야 합니다.	"256"

다음 단계

임베딩 함수를 구성한 후에는 ‘임베딩 함수 개요’를 참조하여 인덱스 생성, 데이터 삽입 및 시맨틱 검색 실행에 대한 지침을 확인하십시오.