com.alibaba.cloud.ai.model.RerankRequest Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-ai-alibaba-core Show documentation

Building AI applications with Spring Boot

There is a newer version: 1.0.0-M2.1

Show newest version

/*
* Copyright 2024 the original author or authors.
*
* Licensed 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
*
*      https://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 com.alibaba.cloud.ai.model;

import org.springframework.ai.document.Document;
import org.springframework.ai.model.ModelRequest;

import java.util.List;

/**
 * Title rerank request.

 * Description rerank request.

 *
 * @author yuanci.ytb
 * @since 1.0.0-M2
 */

public class RerankRequest implements ModelRequest> {

	private final String query;

	private final List documents;

	private final RerankOptions options;

	public RerankRequest(String query, List documents) {
		this(query, documents, null);
	}

	public RerankRequest(String query, List documents, RerankOptions options) {
		this.query = query;
		this.documents = documents;
		this.options = options;
	}

	@Override
	public List getInstructions() {
		return this.documents;
	}

	@Override
	public RerankOptions getOptions() {
		return this.options;
	}

	public String getQuery() {
		return query;
	}

}