BJ_ZTEITS-Developers / parking-cloud-yudao

Commit b77e8f16e9e8948a05ee332a59cb433d00f7cf08

Authored by cherishsince
1 parent f1f003b7

增加 mj Imagine 测试类

Inline Side-by-side
Showing 1 changed file with 34 additions and 0 deletions
yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/image/MjTests.java 0 → 100644
  View file @b77e8f1
  1 +package cn.iocoder.yudao.framework.ai.image;
  2 +
  3 +import cn.hutool.core.io.FileUtil;
  4 +import cn.iocoder.yudao.framework.ai.midjourney.MidjourneyConfig;
  5 +import cn.iocoder.yudao.framework.ai.midjourney.interactions.MjImagineInteractions;
  6 +import org.junit.Test;
  7 +
  8 +import java.io.File;
  9 +import java.util.HashMap;
  10 +import java.util.List;
  11 +import java.util.Map;
  12 +
  13 +/**
  14 + * mj
  15 + *
  16 + * author: fansili
  17 + * time: 2024/4/4 18:59
  18 + */
  19 +public class MjTests {
  20 +
  21 + @Test
  22 + public void mjImage() {
  23 + String token = "NTY5MDY4NDAxNzEyOTU1Mzky.G4-Fu0.MzD-7ll-ElbXTTgDPHF-WS_UyhMAfbKN3WyyBc";
  24 + Map<String, String> requestTemplates = new HashMap<>();
  25 + List<File> files = FileUtil.loopFiles("/Users/fansili/projects/github/ruoyi-vue-pro/yudao-module-ai/yudao-spring-boot-starter-ai/src/main/resources/http-body");
  26 + for (File file : files) {
  27 + requestTemplates.put(file.getName().replace(".json", ""), FileUtil.readUtf8String(file));
  28 + }
  29 + MidjourneyConfig midjourneyConfig = new MidjourneyConfig(
  30 + token, "1221445697157468200", "1221445862962630706", requestTemplates);
  31 + MjImagineInteractions mjImagineInteractions = new MjImagineInteractions(midjourneyConfig);
  32 + mjImagineInteractions.execute("童话里应该是什么样子？");
  33 + }
  34 +}
... ...