程序包 org.apache.seatunnel.transform.llm.model.openai

类 OpenAIModel

java.lang.Object

org.apache.seatunnel.transform.llm.model.AbstractModel

org.apache.seatunnel.transform.llm.model.openai.OpenAIModel

所有已实现的接口:

Closeable, AutoCloseable, Model

public class OpenAIModel
extends AbstractModel

OpenAI model. Refer chat api

字段概要

从类继承的字段 org.apache.seatunnel.transform.llm.model.AbstractModel

OBJECT_MAPPER

构造器概要

构造器

构造器	说明
OpenAIModel(org.apache.seatunnel.api.table.type.SeaTunnelRowType rowType, org.apache.seatunnel.api.table.type.SqlType outputType, String prompt, String model, String apiKey, String apiPath)

方法概要

修饰符和类型	方法	说明
protected List<String>	chatWithModel(String prompt, String data)
void	close()
org.apache.seatunnel.shade.com.fasterxml.jackson.databind.node.ObjectNode	createJsonNodeFromData(String prompt, String data)

从类继承的方法 org.apache.seatunnel.transform.llm.model.AbstractModel

inference

从类继承的方法 java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

构造器详细资料

OpenAIModel

public OpenAIModel(org.apache.seatunnel.api.table.type.SeaTunnelRowType rowType,
                   org.apache.seatunnel.api.table.type.SqlType outputType,
                   String prompt,
                   String model,
                   String apiKey,
                   String apiPath)

方法详细资料

chatWithModel

protected List<String> chatWithModel(String prompt,
                                     String data)
                              throws IOException

指定者:

chatWithModel 在类中 AbstractModel

抛出:

IOException

createJsonNodeFromData

public org.apache.seatunnel.shade.com.fasterxml.jackson.databind.node.ObjectNode createJsonNodeFromData(String prompt,
                                                                                                        String data)

close

public void close()
           throws IOException

抛出:

IOException