modules/multi2vec-bind/nearVideo/graphql_extract.go

//                           _       _
// __      _____  __ ___   ___  __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
//  \ V  V /  __/ (_| |\ V /| | (_| | ||  __/
//   \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
//  Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
//  CONTACT: [email protected]
//

package nearVideo

import "github.com/weaviate/weaviate/usecases/modulecomponents/nearVideo"

// extractNearVideoFn arguments, such as "video" and "certainty"
func extractNearVideoFn(source map[string]interface{}) interface{} {
	var args nearVideo.NearVideoParams

	video, ok := source["video"].(string)
	if ok {
		args.Video = video
	}

	certainty, ok := source["certainty"]
	if ok {
		args.Certainty = certainty.(float64)
	}

	distance, ok := source["distance"]
	if ok {
		args.Distance = distance.(float64)
		args.WithDistance = true
	}

	return &args
}