1234567891011121314151617181920212223242526272829303132333435363738394041 |
- /*
- * # Licensed to the LF AI & Data foundation under one
- * # or more contributor license agreements. See the NOTICE file
- * # distributed with this work for additional information
- * # regarding copyright ownership. The ASF licenses this file
- * # to you under the Apache License, Version 2.0 (the
- * # "License"); you may not use this file except in compliance
- * # with the License. You may obtain a copy of the License at
- * #
- * # http://www.apache.org/licenses/LICENSE-2.0
- * #
- * # Unless required by applicable law or agreed to in writing, software
- * # distributed under the License is distributed on an "AS IS" BASIS,
- * # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * # See the License for the specific language governing permissions and
- * # limitations under the License.
- */
- package function
- import (
- "fmt"
- "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
- )
- type FunctionRunner interface {
- BatchRun(inputs ...any) ([]any, error)
- GetSchema() *schemapb.FunctionSchema
- GetOutputFields() []*schemapb.FieldSchema
- }
- func NewFunctionRunner(coll *schemapb.CollectionSchema, schema *schemapb.FunctionSchema) (FunctionRunner, error) {
- switch schema.GetType() {
- case schemapb.FunctionType_BM25:
- return NewBM25FunctionRunner(coll, schema)
- default:
- return nil, fmt.Errorf("unknown functionRunner type %s", schema.GetType().String())
- }
- }
|