모델에서 온톨로지 데이터 사용

모델 어댑터는 온톨로지 객체와 객체 집합을 입력으로 받거나 추론 중 SDK를 통해 객체, 함수 및 언어 모델을 호출할 수 있습니다. 온톨로지의 업무 데이터와 기존 모델링 로직을 같은 추론 과정에서 사용할 때 필요한 연결 방식입니다.

객체와 객체 집합 입력

Developer Console 또는 Code Workspaces에서 사용할 온톨로지의 Python SDK를 생성합니다. 아래 Employee와 EmployeeObjectSet은 예시 객체 및 집합 클래스입니다. 실제로 생성한 ontology_sdk 패키지의 클래스와 속성으로 바꿉니다.

어댑터의 api()는 입력에 객체 타입을 지정합니다. 다음 입력/출력 정의를 api()에서 (inputs, outputs)로 반환합니다.

import palantir_models as pm
from ontology_sdk.ontology.objects import Employee
from ontology_sdk.ontology.object_sets import EmployeeObjectSet

inputs = {
    "object_input": pm.Object(Employee),
    "object_set_input": pm.ObjectSet(Employee),
}
outputs = {"param_out": pm.Parameter(type=str)}

predict의 입력 이름은 object_input과 object_set_input입니다. 첫 값은 Employee 인스턴스이고 두 번째는 해당 타입의 EmployeeObjectSet입니다. 어댑터가 실제 업무 로직을 실행하여 반환한 문자열을 param_out으로 제공합니다. Object 입력은 해당 타입의 기본 키로, ObjectSet 입력은 객체 집합 RID로 전달할 수도 있습니다.

직접 온라인 배포가 실행 중이면 모델 Query 페이지에서 온톨로지의 객체와 집합을 선택하여 입력할 수 있습니다. 왼쪽 사이드바에서 이 입력을 사용하는 모델 함수를 게시할 수도 있습니다.

추론 중 SDK 조회

predict 또는 transform 실행 안에서 FoundryClient()를 생성하여 온톨로지 및 Platform SDK를 사용할 수 있습니다. 다음 함수는 Employee에 필수 문자열 first_name 속성이 있는 예시 SDK의 실제 조회 경로입니다. 모델의 predict에서 기본 키를 받아 이 함수를 호출하면 조회한 이름을 반환합니다.

from ontology_sdk import FoundryClient

def get_employee_first_name(employee_primary_key: str) -> str:
    client = FoundryClient()
    employee = client.ontology.objects.Employee.get(employee_primary_key)
    return employee.first_name

객체 조회 결과를 모델의 특성으로 사용하거나 온톨로지 함수 및 언어 모델과 결합하는 로직도 같은 추론 컨텍스트에서 작성할 수 있습니다. 생성한 SDK의 실제 객체 타입과 반환 속성에 맞춰 모델 API 및 추론 입력을 정의합니다.