@@ -95,28 +95,27 @@ func (r InferencePipelineDataStreamParams) MarshalJSON() (data []byte, err error
9595// Configuration for the data stream. Depends on your **Openlayer project task
9696// type**.
9797type InferencePipelineDataStreamParamsConfig struct {
98- CategoricalFeatureNames param.Field [interface {}] `json:"categoricalFeatureNames,required"`
99- ClassNames param.Field [interface {}] `json:"classNames,required"`
100- FeatureNames param.Field [interface {}] `json:"featureNames,required"`
101- InputVariableNames param.Field [interface {}] `json:"inputVariableNames,required"`
102- Metadata param.Field [interface {}] `json:"metadata,required"`
103- Prompt param.Field [interface {}] `json:"prompt,required"`
98+ CategoricalFeatureNames param.Field [interface {}] `json:"categoricalFeatureNames"`
99+ ClassNames param.Field [interface {}] `json:"classNames"`
104100 // Name of the column with the context retrieved. Applies to RAG use cases.
105101 // Providing the context enables RAG-specific metrics.
106102 ContextColumnName param.Field [string ] `json:"contextColumnName"`
107103 // Name of the column with the cost associated with each row.
108- CostColumnName param.Field [string ] `json:"costColumnName"`
104+ CostColumnName param.Field [string ] `json:"costColumnName"`
105+ FeatureNames param.Field [interface {}] `json:"featureNames"`
109106 // Name of the column with the ground truths.
110107 GroundTruthColumnName param.Field [string ] `json:"groundTruthColumnName"`
111108 // Name of the column with the inference ids. This is useful if you want to update
112109 // rows at a later point in time. If not provided, a unique id is generated by
113110 // Openlayer.
114- InferenceIDColumnName param.Field [string ] `json:"inferenceIdColumnName"`
111+ InferenceIDColumnName param.Field [string ] `json:"inferenceIdColumnName"`
112+ InputVariableNames param.Field [interface {}] `json:"inputVariableNames"`
115113 // Name of the column with the labels. The data in this column must be
116114 // **zero-indexed integers**, matching the list provided in `classNames`.
117115 LabelColumnName param.Field [string ] `json:"labelColumnName"`
118116 // Name of the column with the latencies.
119- LatencyColumnName param.Field [string ] `json:"latencyColumnName"`
117+ LatencyColumnName param.Field [string ] `json:"latencyColumnName"`
118+ Metadata param.Field [interface {}] `json:"metadata"`
120119 // Name of the column with the total number of tokens.
121120 NumOfTokenColumnName param.Field [string ] `json:"numOfTokenColumnName"`
122121 // Name of the column with the model outputs.
@@ -125,7 +124,8 @@ type InferencePipelineDataStreamParamsConfig struct {
125124 PredictionsColumnName param.Field [string ] `json:"predictionsColumnName"`
126125 // Name of the column with the model's predictions as **lists of class
127126 // probabilities**.
128- PredictionScoresColumnName param.Field [string ] `json:"predictionScoresColumnName"`
127+ PredictionScoresColumnName param.Field [string ] `json:"predictionScoresColumnName"`
128+ Prompt param.Field [interface {}] `json:"prompt"`
129129 // Name of the column with the questions. Applies to RAG use cases. Providing the
130130 // question enables RAG-specific metrics.
131131 QuestionColumnName param.Field [string ] `json:"questionColumnName"`
0 commit comments