Skip to content

Commit 5e1d853

Browse files
fix: generateImages requires apikey
1 parent 1ca9440 commit 5e1d853

File tree

4 files changed

+7
-3
lines changed

4 files changed

+7
-3
lines changed

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/interceptor/ArkResourceStsAuthenticationInterceptor.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ public Response intercept(Chain chain) throws IOException {
5454
String requestResourceType = getRequestResourceType(request);
5555
String requestResourceId = getRequestResourceId(request);
5656

57-
if (request.url().url().getPath().contains("contents/generations")) {
57+
if (request.url().url().getPath().contains("contents/generations") || request.url().url().getPath().contains("images/generations")) {
5858
throw new ArkException("content generation currently does not support ak&sk authentication, use api_key instead.");
5959
}
6060

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/service/ArkBaseServiceImpl.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@
1212
import com.volcengine.ark.runtime.model.context.chat.ContextChatCompletionRequest;
1313
import com.volcengine.ark.runtime.model.embeddings.EmbeddingRequest;
1414
import com.volcengine.ark.runtime.model.embeddings.EmbeddingResult;
15+
import com.volcengine.ark.runtime.model.images.generation.GenerateImagesRequest;
16+
import com.volcengine.ark.runtime.model.images.generation.ImagesResponse;
1517
import com.volcengine.ark.runtime.model.multimodalembeddings.MultimodalEmbeddingRequest;
1618
import com.volcengine.ark.runtime.model.multimodalembeddings.MultimodalEmbeddingResult;
1719
import com.volcengine.ark.runtime.model.tokenization.TokenizationRequest;
@@ -43,6 +45,8 @@ public interface ArkBaseServiceImpl {
4345

4446
MultimodalEmbeddingResult createMultiModalEmbeddings(MultimodalEmbeddingRequest request);
4547

48+
ImagesResponse generateImages(GenerateImagesRequest request);
49+
4650
CreateContentGenerationTaskResult createContentGenerationTask(CreateContentGenerationTaskRequest request);
4751

4852
GetContentGenerationTaskResponse getContentGenerationTask(GetContentGenerationTaskRequest request);

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/service/ArkService.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ public TokenizationResult createTokenization(TokenizationRequest request, Map<St
269269
return execute(api.createTokenization(request, request.getModel(), customHeaders));
270270
}
271271

272-
public ImagesResponse GenerateImages(GenerateImagesRequest request) {
272+
public ImagesResponse generateImages(GenerateImagesRequest request) {
273273
return execute(api.generateImages(request, request.getModel(), new HashMap<>()));
274274
}
275275

volcengine-java-sdk-ark-runtime/test/java/com/volcengine/ark/runtime/ImageGenerationsExample.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ public static void main(String[] args) {
4444

4545
System.out.println(generateRequest.toString());
4646
// send create request
47-
ImagesResponse imagesResponse = service.GenerateImages(generateRequest);
47+
ImagesResponse imagesResponse = service.generateImages(generateRequest);
4848
System.out.println(imagesResponse.getData().get(0).getUrl());
4949

5050
service.shutdownExecutor();

0 commit comments

Comments
 (0)