Skip to content

Commit 927f518

Browse files
author
lmh
committed
feat: add file input
1 parent 55375f8 commit 927f518

File tree

3 files changed

+110
-0
lines changed

3 files changed

+110
-0
lines changed

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/responses/constant/ResponsesConstants.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ public class ResponsesConstants {
3939
public static final String CONTENT_ITEM_TYPE_INPUT_TEXT = "input_text";
4040
public static final String CONTENT_ITEM_TYPE_INPUT_IMAGE = "input_image";
4141
public static final String CONTENT_ITEM_TYPE_INPUT_VIDEO = "input_video";
42+
public static final String CONTENT_ITEM_TYPE_INPUT_FILE = "input_file";
4243
public static final String CONTENT_ITEM_TYPE_OUTPUT_TEXT = "output_text";
4344
public static final String CONTENT_ITEM_TYPE_SUMMARY_TEXT = "summary_text";
4445

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/responses/content/InputContentItem.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
@JsonSubTypes.Type(value = InputContentItemText.class, name = ResponsesConstants.CONTENT_ITEM_TYPE_INPUT_TEXT),
1717
@JsonSubTypes.Type(value = InputContentItemImage.class, name = ResponsesConstants.CONTENT_ITEM_TYPE_INPUT_IMAGE),
1818
@JsonSubTypes.Type(value = InputContentItemVideo.class, name = ResponsesConstants.CONTENT_ITEM_TYPE_INPUT_VIDEO),
19+
@JsonSubTypes.Type(value = InputContentItemFile.class, name = ResponsesConstants.CONTENT_ITEM_TYPE_INPUT_FILE),
1920
})
2021
public abstract class InputContentItem {
2122
@JsonProperty("type")
Lines changed: 108 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,108 @@
1+
package com.volcengine.ark.runtime.model.responses.content;
2+
3+
import com.fasterxml.jackson.annotation.JsonProperty;
4+
import com.volcengine.ark.runtime.model.responses.constant.ResponsesConstants;
5+
6+
public class InputContentItemFile extends InputContentItem{
7+
8+
public InputContentItemFile() {
9+
super(ResponsesConstants.CONTENT_ITEM_TYPE_INPUT_FILE);
10+
}
11+
12+
public String getFileName() {
13+
return fileName;
14+
}
15+
16+
public void setFileName(String fileName) {
17+
this.fileName = fileName;
18+
}
19+
20+
public String getFileData() {
21+
return fileData;
22+
}
23+
24+
public void setFileData(String fileData) {
25+
this.fileData = fileData;
26+
}
27+
28+
public String getFileId() {
29+
return fileId;
30+
}
31+
32+
public void setFileId(String fileId) {
33+
this.fileId = fileId;
34+
}
35+
36+
public String getFileUrl() {
37+
return fileUrl;
38+
}
39+
40+
public void setFileUrl(String fileUrl) {
41+
this.fileUrl = fileUrl;
42+
}
43+
44+
@JsonProperty("file_url")
45+
private String fileUrl;
46+
47+
@JsonProperty("file_id")
48+
private String fileId;
49+
50+
@JsonProperty("file_data")
51+
private String fileData;
52+
53+
@JsonProperty("filename")
54+
private String fileName;
55+
56+
@Override
57+
public String toString() {
58+
return "InputContentItemFile{" +
59+
"fileUrl='" + fileUrl + '\'' +
60+
", fileId='" + fileId + '\'' +
61+
", fileData='" + fileData + '\'' +
62+
", fileName='" + fileName + '\'' +
63+
'}';
64+
}
65+
66+
public static final class InputContentItemFileBuilder {
67+
private String fileUrl;
68+
private String fileId;
69+
private String fileData;
70+
private String fileName;
71+
72+
private InputContentItemFileBuilder() {
73+
}
74+
75+
public static InputContentItemFileBuilder anInputContentItemFile() {
76+
return new InputContentItemFileBuilder();
77+
}
78+
79+
public InputContentItemFileBuilder fileUrl(String fileUrl) {
80+
this.fileUrl = fileUrl;
81+
return this;
82+
}
83+
84+
public InputContentItemFileBuilder fileId(String fileId) {
85+
this.fileId = fileId;
86+
return this;
87+
}
88+
89+
public InputContentItemFileBuilder fileData(String fileData) {
90+
this.fileData = fileData;
91+
return this;
92+
}
93+
94+
public InputContentItemFileBuilder fileName(String fileName) {
95+
this.fileName = fileName;
96+
return this;
97+
}
98+
99+
public InputContentItemFile build() {
100+
InputContentItemFile inputContentItemFile = new InputContentItemFile();
101+
inputContentItemFile.setFileUrl(fileUrl);
102+
inputContentItemFile.setFileId(fileId);
103+
inputContentItemFile.setFileData(fileData);
104+
inputContentItemFile.setFileName(fileName);
105+
return inputContentItemFile;
106+
}
107+
}
108+
}

0 commit comments

Comments
 (0)