devbox
爱喝兽奶帝天荒
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

SpringBoot整合chatGPT_com.theokanning.openai-gpt3-java

作者：爱喝兽奶帝天荒 | 2024-08-06 13:17:16

赞

踩

com.theokanning.openai-gpt3-java

1 添加依赖

  1.         <!-- 导入openai依赖 -->
  2.         <dependency>
  3.             <groupId>com.theokanning.openai-gpt3-java</groupId>
  4.             <artifactId>client</artifactId>
  5.             <version>0.8.1</version>
  6.         </dependency>

2 创建相关文件

2.1 实体类:OpenAi.java

  1.  
  2. package com.wkf.workrecord.tools.openai;
  3.  
  4. import lombok.AllArgsConstructor;
  5. import lombok.Data;
  6. import lombok.NoArgsConstructor;
  7.  
  8. /**
  9.  * @author wuKeFan
  10.  * @date 2023-02-10 15:40:22
  11.  */
  12. @Data
  13. @NoArgsConstructor
  14. @AllArgsConstructor
  15. public class OpenAi {
  16.  
  17.     String id;
  18.  
  19.     String name;
  20.  
  21.     String desc;
  22.  
  23.     String model;
  24.  
  25.     // 提示模板
  26.     String prompt;
  27.  
  28.     // 创新采样
  29.     Double temperature;
  30.  
  31.     // 情绪采样
  32.     Double topP;
  33.  
  34.     // 结果条数
  35.     Double n = 1d;
  36.  
  37.     // 频率处罚系数
  38.     Double frequencyPenalty;
  39.  
  40.     // 重复处罚系数
  41.     Double presencePenalty;
  42.  
  43.     // 停用词
  44.     String stop;
  45.  
  46. }

2.2 配置类:OpenAiProperties.java

  1.  
  2. package com.wkf.workrecord.tools.openai;
  3.  
  4. import lombok.Data;
  5. import org.springframework.beans.factory.InitializingBean;
  6. import org.springframework.boot.context.properties.ConfigurationProperties;
  7.  
  8. /**
  9.  * @author wuKeFan
  10.  * @date 2023-02-10 15:25:32
  11.  */
  12.  
  13. @Data
  14. @ConfigurationProperties(prefix = "openai")
  15. public class OpenAiProperties implements InitializingBean {
  16.     // 秘钥
  17.     String token;
  18.     // 超时时间
  19.     Integer timeout;
  20.  
  21.     // 设置属性时同时设置给OpenAiUtils
  22.     @Override
  23.     public void afterPropertiesSet() throws Exception {
  24.         OpenAiUtils.OPENAPI_TOKEN = token;
  25.         OpenAiUtils.TIMEOUT = timeout;
  26.     }
  27. }

2.3 核心业务逻辑OpenAiUtils.java

  1.  
  2. package com.wkf.workrecord.tools.openai;
  3.  
  4. import com.theokanning.openai.OpenAiService;
  5. import com.theokanning.openai.completion.CompletionChoice;
  6. import com.theokanning.openai.completion.CompletionRequest;
  7. import org.springframework.util.StringUtils;
  8. import java.util.*;
  9.  
  10. /**
  11.  * @author wuKeFan
  12.  * @date 2023-02-10 15:32:18
  13.  */
  14. public class OpenAiUtils {
  15.     public static final Map<String, OpenAi> PARMS = new HashMap<>();
  16.  
  17.     static {
  18.         PARMS.put("OpenAi01", new OpenAi("OpenAi01", "问&答", "依据现有知识库问&答", "text-davinci-003", "Q: %s\nA:", 0.0, 1.0, 1.0, 0.0, 0.0, "\n"));
  19.         PARMS.put("OpenAi02", new OpenAi("OpenAi02", "语法纠正", "将句子转换成标准的英语，输出结果始终是英文", "text-davinci-003", "%s", 0.0, 1.0, 1.0, 0.0, 0.0, ""));
  20.         PARMS.put("OpenAi03", new OpenAi("OpenAi03", "内容概况", "将一段话，概况中心", "text-davinci-003", "Summarize this for a second-grade student:\n%s", 0.7, 1.0, 1.0, 0.0, 0.0, ""));
  21.         PARMS.put("OpenAi04", new OpenAi("OpenAi04", "生成OpenAi的代码", "一句话生成OpenAi的代码", "code-davinci-002", "\"\"\"\nUtil exposes the following:\nutil.openai() -> authenticates & returns the openai module, which has the following functions:\nopenai.Completion.create(\n    prompt=\"<my prompt>\", # The prompt to start completing from\n    max_tokens=123, # The max number of tokens to generate\n    temperature=1.0 # A measure of randomness\n    echo=True, # Whether to return the prompt in addition to the generated completion\n)\n\"\"\"\nimport util\n\"\"\"\n%s\n\"\"\"\n\n", 0.0, 1.0, 1.0, 0.0, 0.0, "\"\"\""));
  22.         PARMS.put("OpenAi05", new OpenAi("OpenAi05", "程序命令生成", "一句话生成程序的命令，目前支持操作系统指令比较多", "text-davinci-003", "Convert this text to a programmatic command:\n\nExample: Ask Constance if we need some bread\nOutput: send-msg `find constance` Do we need some bread?\n\n%s", 0.0, 1.0, 1.0, 0.2, 0.0, ""));
  23.         PARMS.put("OpenAi06", new OpenAi("OpenAi06", "语言翻译", "把一种语法翻译成其它几种语言", "text-davinci-003", "Translate this into %s:\n%s", 0.3, 1.0, 1.0, 0.0, 0.0, ""));
  24.         PARMS.put("OpenAi07", new OpenAi("OpenAi07", "Stripe国际API生成", "一句话生成Stripe国际支付API", "code-davinci-002", "\"\"\"\nUtil exposes the following:\n\nutil.stripe() -> authenticates & returns the stripe module; usable as stripe.Charge.create etc\n\"\"\"\nimport util\n\"\"\"\n%s\n\"\"\"", 0.0, 1.0, 1.0, 0.0, 0.0, "\"\"\""));
  25.         PARMS.put("OpenAi08", new OpenAi("OpenAi08", "SQL语句生成", "依据上下文中的表信息，生成SQL语句", "code-davinci-002", "### %s SQL tables, 表字段信息如下:\n%s\n#\n### %s\n %s", 0.0, 1.0, 1.0, 0.0, 0.0, "# ;"));
  26.         PARMS.put("OpenAi09", new OpenAi("OpenAi09", "结构化生成", "对于非结构化的数据抽取其中的特征生成结构化的表格", "text-davinci-003", "A table summarizing, use Chinese:\n%s\n", 0.0, 1.0, 1.0, 0.0, 0.0, ""));
  27.         PARMS.put("OpenAi10", new OpenAi("OpenAi10", "信息分类", "把一段信息继续分类", "text-davinci-003", "%s\n分类:", 0.0, 1.0, 1.0, 0.0, 0.0, ""));
  28.         PARMS.put("OpenAi11", new OpenAi("OpenAi11", "Python代码解释", "把代码翻译成文字，用来解释程序的作用", "code-davinci-002", "# %s \n %s \n\n# 解释代码作用\n\n#", 0.0, 1.0, 1.0, 0.0, 0.0, ""));
  29.         PARMS.put("OpenAi12", new OpenAi("OpenAi12", "文字转表情符号", "将文本编码成表情服务", "text-davinci-003", "转换文字为表情。\n%s：", 0.8, 1.0, 1.0, 0.0, 0.0, "\n"));
  30.         PARMS.put("OpenAi13", new OpenAi("OpenAi13", "时间复杂度计算", "求一段代码的时间复杂度", "text-davinci-003", "%s\n\"\"\"\n函数的时间复杂度是", 0.0, 1.0, 1.0, 0.0, 0.0, "\n"));
  31.         PARMS.put("OpenAi14", new OpenAi("OpenAi14", "程序代码翻译", "把一种语言的代码翻译成另外一种语言的代码", "code-davinci-002", "##### 把这段代码从%s翻译成%s\n### %s\n    \n   %s\n    \n### %s", 0.0, 1.0, 1.0, 0.0, 0.0, "###"));
  32.         PARMS.put("OpenAi15", new OpenAi("OpenAi15", "高级情绪评分", "支持批量列表的方式检查情绪", "text-davinci-003", "对下面内容进行情感分类:\n%s\"\n情绪评级:", 0.0, 1.0, 1.0, 0.0, 0.0, ""));
  33.         PARMS.put("OpenAi16", new OpenAi("OpenAi16", "代码解释", "对一段代码进行解释", "code-davinci-002", "代码:\n%s\n\"\"\"\n上面的代码在做什么:\n1. ", 0.0, 1.0, 1.0, 0.0, 0.0, "\"\"\""));
  34.         PARMS.put("OpenAi17", new OpenAi("OpenAi17", "关键字提取", "提取一段文本中的关键字", "text-davinci-003", "抽取下面内容的关键字:\n%s", 0.5, 1.0, 1.0, 0.8, 0.0, ""));
  35.         PARMS.put("OpenAi18", new OpenAi("OpenAi18", "问题解答", "类似解答题", "text-davinci-003", "Q: %s\nA: ?", 0.0, 1.0, 1.0, 0.0, 0.0, ""));
  36.         PARMS.put("OpenAi19", new OpenAi("OpenAi19", "广告设计", "给一个产品设计一个广告", "text-davinci-003", "为下面的产品创作一个创业广告，用于投放到抖音上：\n产品:%s.", 0.5, 1.0, 1.0, 0.0, 0.0, ""));
  37.         PARMS.put("OpenAi20", new OpenAi("OpenAi20", "产品取名", "依据产品描述和种子词语，给一个产品取一个好听的名字", "text-davinci-003", "产品描述: %s.\n种子词: %s.\n产品名称: ", 0.8, 1.0, 1.0, 0.0, 0.0, ""));
  38.         PARMS.put("OpenAi21", new OpenAi("OpenAi21", "句子简化", "把一个长句子简化成一个短句子", "text-davinci-003", "%s\nTl;dr: ", 0.7, 1.0, 1.0, 0.0, 1.0, ""));
  39.         PARMS.put("OpenAi22", new OpenAi("OpenAi22", "修复代码Bug", "自动修改代码中的bug", "code-davinci-002", "##### 修复下面代码的bug\n### %s\n %s\n###  %s\n", 0.0, 1.0, 1.0, 0.0, 0.0, "###"));
  40.         PARMS.put("OpenAi23", new OpenAi("OpenAi23", "表格填充数据", "自动为一个表格生成数据", "text-davinci-003", "spreadsheet ,%s rows:\n%s\n", 0.5, 1.0, 1.0, 0.0, 0.0, ""));
  41.         PARMS.put("OpenAi24", new OpenAi("OpenAi24", "语言聊天机器人", "各种开发语言的两天机器人", "code-davinci-002", "You: %s\n%s机器人:", 0.0, 1.0, 1.0, 0.5, 0.0, "You: "));
  42.         PARMS.put("OpenAi25", new OpenAi("OpenAi25", "机器学习机器人", "机器学习模型方面的机器人", "text-davinci-003", "You: %s\nML机器人:", 0.3, 1.0, 1.0, 0.5, 0.0, "You: "));
  43.         PARMS.put("OpenAi26", new OpenAi("OpenAi26", "清单制作", "可以列出各方面的分类列表，比如歌单", "text-davinci-003", "列出10%s:", 0.5, 1.0, 1.0, 0.52, 0.5, "11.0"));
  44.         PARMS.put("OpenAi27", new OpenAi("OpenAi27", "文本情绪分析", "对一段文字进行情绪分析", "text-davinci-003", "推断下面文本的情绪是积极的, 中立的, 还是消极的.\n文本: \"%s\"\n观点:", 0.0, 1.0, 1.0, 0.5, 0.0, ""));
  45.         PARMS.put("OpenAi28", new OpenAi("OpenAi28", "航空代码抽取", "抽取文本中的航空diam信息", "text-davinci-003", "抽取下面文本中的航空代码：\n文本：\"%s\"\n航空代码：", 0.0, 1.0, 1.0, 0.0, 0.0, "\n"));
  46.         PARMS.put("OpenAi29", new OpenAi("OpenAi29", "生成SQL语句", "无上下文，语句描述生成SQL", "text-davinci-003", "%s", 0.3, 1.0, 1.0, 0.0, 0.0, ""));
  47.         PARMS.put("OpenAi30", new OpenAi("OpenAi30", "抽取联系信息", "从文本中抽取联系方式", "text-davinci-003", "从下面文本中抽取%s:\n%s", 0.0, 1.0, 1.0, 0.0, 0.0, ""));
  48.         PARMS.put("OpenAi31", new OpenAi("OpenAi31", "程序语言转换", "把一种语言转成另外一种语言", "code-davinci-002", "#%s to %s:\n%s:%s\n\n%s:", 0.0, 1.0, 1.0, 0.0, 0.0, ""));
  49.         PARMS.put("OpenAi32", new OpenAi("OpenAi32", "好友聊天", "模仿好友聊天", "text-davinci-003", "You: %s\n好友:", 0.5, 1.0, 1.0, 0.5, 0.0, "You:"));
  50.         PARMS.put("OpenAi33", new OpenAi("OpenAi33", "颜色生成", "依据描述生成对应颜色", "text-davinci-003", "%s:\nbackground-color: ", 0.0, 1.0, 1.0, 0.0, 0.0, ";"));
  51.         PARMS.put("OpenAi34", new OpenAi("OpenAi34", "程序文档生成", "自动为程序生成文档", "code-davinci-002", "# %s\n \n%s\n# 上述代码的详细、高质量文档字符串：\n\"\"\"", 0.0, 1.0, 1.0, 0.0, 0.0, "#\"\"\""));
  52.         PARMS.put("OpenAi35", new OpenAi("OpenAi35", "段落创作", "依据短语生成相关文短", "text-davinci-003", "为下面短语创建一个中文段:\n%s:\n", 0.5, 1.0, 1.0, 0.0, 0.0, ""));
  53.         PARMS.put("OpenAi36", new OpenAi("OpenAi36", "代码压缩", "把多行代码简单的压缩成一行", "code-davinci-002", "将下面%s代码转成一行:\n%s\n%s一行版本:", 0.0, 1.0, 1.0, 0.0, 0.0, ";"));
  54.         PARMS.put("OpenAi37", new OpenAi("OpenAi37", "故事创作", "依据一个主题创建一个故事", "text-davinci-003", "主题: %s\n故事创作:", 0.8, 1.0, 1.0, 0.5, 0.0, ""));
  55.         PARMS.put("OpenAi38", new OpenAi("OpenAi38", "人称转换", "第一人称转第3人称", "text-davinci-003", "把下面内容从第一人称转为第三人称 (性别女):\n%s\n", 0.0, 1.0, 1.0, 0.0, 0.0, ""));
  56.         PARMS.put("OpenAi39", new OpenAi("OpenAi39", "摘要说明", "依据笔记生成摘要说明", "text-davinci-003", "将下面内容转换成将下%s摘要：\n%s", 0.0, 1.0, 1.0, 0.0, 0.0, ""));
  57.         PARMS.put("OpenAi40", new OpenAi("OpenAi40", "头脑风暴", "给定一个主题，让其生成一些主题相关的想法", "text-davinci-003", "头脑风暴一些关于%s的想法：", 0.6, 1.0, 1.0, 1.0, 1.0, ""));
  58.         PARMS.put("OpenAi41", new OpenAi("OpenAi41", "ESRB文本分类", "按照ESRB进行文本分类", "text-davinci-003", "Provide an ESRB rating for the following text:\\n\\n\\\"%s\"\\n\\nESRB rating:", 0.3, 1.0, 1.0, 0.0, 0.0, "\n"));
  59.         PARMS.put("OpenAi42", new OpenAi("OpenAi42", "提纲生成", "按照提示为相关内容生成提纲", "text-davinci-003", "为%s提纲:", 0.3, 1.0, 1.0, 0.0, 0.0, ""));
  60.         PARMS.put("OpenAi43", new OpenAi("OpenAi43", "美食制作（后果自负）", "依据美食名称和材料生成美食的制作步骤", "text-davinci-003", "依据下面成分和美食，生成制作方法:\n%s\n成分:\n%s\n制作方法:", 0.3, 1.0, 1.0, 0.0, 0.0, ""));
  61.         PARMS.put("OpenAi44", new OpenAi("OpenAi44", "AI聊天", "与AI机器进行聊天", "text-davinci-003", "Human: %s", 0.9, 1.0, 1.0, 0.0, 0.6, "Human:AI:"));
  62.         PARMS.put("OpenAi45", new OpenAi("OpenAi45", "摆烂聊天", "与讽刺机器进行聊天", "text-davinci-003", "Marv不情愿的回答问题.\nYou:%s\nMarv:", 0.5, 0.3, 1.0, 0.5, 0.0, ""));
  63.         PARMS.put("OpenAi46", new OpenAi("OpenAi46", "分解步骤", "把一段文本分解成几步来完成", "text-davinci-003", "为下面文本生成次序列表，并增加列表数子: \n%s\n", 0.3, 1.0, 1.0, 0.0, 0.0, ""));
  64.         PARMS.put("OpenAi47", new OpenAi("OpenAi47", "点评生成", "依据文本内容自动生成点评", "text-davinci-003", "依据下面内容，进行点评:\n%s\n点评:", 0.5, 1.0, 1.0, 0.0, 0.0, ""));
  65.         PARMS.put("OpenAi48", new OpenAi("OpenAi48", "知识学习", "可以为学习知识自动解答", "text-davinci-003", "%s", 0.3, 1.0, 1.0, 0.0, 0.0, ""));
  66.         PARMS.put("OpenAi49", new OpenAi("OpenAi49", "面试", "生成面试题", "text-davinci-003", "创建10道%s相关的面试题（中文）：\n", 0.5, 1.0, 10.0, 0.0, 0.0, ""));
  67.     }
  68.  
  69.     public static String OPENAPI_TOKEN = "";
  70.     public static Integer TIMEOUT = null;
  71.  
  72.     /**
  73.      * 获取ai
  74.      *
  75.      * @param openAi
  76.      * @param prompt
  77.      * @return
  78.      */
  79.     public static List<CompletionChoice> getAiResult(OpenAi openAi, String prompt) {
  80.         if (TIMEOUT == null || TIMEOUT < 1000) {
  81.             TIMEOUT = 3000;
  82.         }
  83.         OpenAiService service = new OpenAiService(OPENAPI_TOKEN, TIMEOUT);
  84.         CompletionRequest.CompletionRequestBuilder builder = CompletionRequest.builder()
  85.                 .model(openAi.getModel())
  86.                 .prompt(prompt)
  87.                 .temperature(openAi.getTemperature())
  88.                 .maxTokens(1000)
  89.                 .topP(openAi.getTopP())
  90.                 .frequencyPenalty(openAi.getFrequencyPenalty())
  91.                 .presencePenalty(openAi.getPresencePenalty());
  92.         if (!StringUtils.isEmpty(openAi.getStop())) {
  93.             builder.stop(Arrays.asList(openAi.getStop().split(",")));
  94.         }
  95.         CompletionRequest completionRequest = builder.build();
  96.         return service.createCompletion(completionRequest).getChoices();
  97.     }
  98.  
  99.     /**
  100.      * 问答
  101.      *
  102.      * @param question
  103.      * @return
  104.      */
  105.     public static List<CompletionChoice> getQuestionAnswer(String question) {
  106.         OpenAi openAi = PARMS.get("OpenAi01");
  107.         return getAiResult(openAi, String.format(openAi.getPrompt(), question));
  108.     }
  109.  
  110.     /**
  111.      * 语法纠错
  112.      *
  113.      * @param text
  114.      * @return
  115.      */
  116.     public static List<CompletionChoice> getGrammarCorrection(String text) {
  117.         OpenAi openAi = PARMS.get("OpenAi02");
  118.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  119.     }
  120.  
  121.     /**
  122.      * 将一段话，概况中心
  123.      *
  124.      * @param text
  125.      * @return
  126.      */
  127.     public static List<CompletionChoice> getSummarize(String text) {
  128.         OpenAi openAi = PARMS.get("OpenAi03");
  129.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  130.     }
  131.  
  132.     /**
  133.      * 一句话生成OpenAi的代码
  134.      *
  135.      * @param text
  136.      * @return
  137.      */
  138.     public static List<CompletionChoice> getOpenAiApi(String text) {
  139.         OpenAi openAi = PARMS.get("OpenAi04");
  140.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  141.     }
  142.  
  143.     /**
  144.      * 一句话生成程序的命令，目前支持操作系统指令比较多
  145.      *
  146.      * @param text
  147.      * @return
  148.      */
  149.     public static List<CompletionChoice> getTextToCommand(String text) {
  150.         OpenAi openAi = PARMS.get("OpenAi05");
  151.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  152.     }
  153.  
  154.     /**
  155.      * 把一种语法翻译成其它几种语言
  156.      *
  157.      * @param text
  158.      * @return
  159.      */
  160.     public static List<CompletionChoice> getTranslatesLanguages(String text, String translatesLanguages) {
  161.         if (StringUtils.isEmpty(translatesLanguages)) {
  162.             translatesLanguages = "  1. French, 2. Spanish and 3. English";
  163.         }
  164.         OpenAi openAi = PARMS.get("OpenAi06");
  165.         return getAiResult(openAi, String.format(openAi.getPrompt(), translatesLanguages, text));
  166.     }
  167.  
  168.     /**
  169.      * 一句话生成Stripe国际支付API
  170.      *
  171.      * @param text
  172.      * @return
  173.      */
  174.     public static List<CompletionChoice> getStripeApi(String text) {
  175.         OpenAi openAi = PARMS.get("OpenAi07");
  176.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  177.     }
  178.  
  179.  
  180.     /**
  181.      * 依据上下文中的表信息，生成SQL语句
  182.      *
  183.      * @param databaseType 数据库类型
  184.      * @param tables       上午依赖的表和字段 Employee(id, name, department_id)
  185.      * @param text         SQL描述
  186.      * @param sqlType      sql类型，比如SELECT
  187.      * @return
  188.      */
  189.     public static List<CompletionChoice> getStripeApi(String databaseType, List<String> tables, String text, String sqlType) {
  190.         OpenAi openAi = PARMS.get("OpenAi08");
  191.         StringJoiner joiner = new StringJoiner("\n");
  192.         for (int i = 0; i < tables.size(); i++) {
  193.             joiner.add("# " + tables);
  194.         }
  195.         return getAiResult(openAi, String.format(openAi.getPrompt(), databaseType, joiner.toString(), text, sqlType));
  196.     }
  197.  
  198.     /**
  199.      * 对于非结构化的数据抽取其中的特征生成结构化的表格
  200.      *
  201.      * @param text 非结构化的数据
  202.      * @return
  203.      */
  204.     public static List<CompletionChoice> getUnstructuredData(String text) {
  205.         OpenAi openAi = PARMS.get("OpenAi09");
  206.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  207.     }
  208.  
  209.     /**
  210.      * 把一段信息继续分类
  211.      *
  212.      * @param text 要分类的文本
  213.      * @return
  214.      */
  215.     public static List<CompletionChoice> getTextCategory(String text) {
  216.         OpenAi openAi = PARMS.get("OpenAi10");
  217.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  218.     }
  219.  
  220.     /**
  221.      * 把一段信息继续分类
  222.      *
  223.      * @param codeType 代码类型，比如Python
  224.      * @param code     要解释的代码
  225.      * @return
  226.      */
  227.     public static List<CompletionChoice> getCodeExplain(String codeType, String code) {
  228.         OpenAi openAi = PARMS.get("OpenAi11");
  229.         return getAiResult(openAi, String.format(openAi.getPrompt(), codeType, code));
  230.     }
  231.  
  232.     /**
  233.      * 将文本编码成表情服务
  234.      *
  235.      * @param text 文本
  236.      * @return
  237.      */
  238.     public static List<CompletionChoice> getTextEmoji(String text) {
  239.         OpenAi openAi = PARMS.get("OpenAi12");
  240.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  241.     }
  242.  
  243.     /**
  244.      * 求一段代码的时间复杂度
  245.      *
  246.      * @param code 代码
  247.      * @return
  248.      */
  249.     public static List<CompletionChoice> getTimeComplexity(String code) {
  250.         OpenAi openAi = PARMS.get("OpenAi13");
  251.         return getAiResult(openAi, String.format(openAi.getPrompt(), code));
  252.     }
  253.  
  254.  
  255.     /**
  256.      * 把一种语言的代码翻译成另外一种语言的代码
  257.      *
  258.      * @param fromLanguage 要翻译的代码语言
  259.      * @param toLanguage   要翻译成的代码语言
  260.      * @param code         代码
  261.      * @return
  262.      */
  263.     public static List<CompletionChoice> getTranslateProgramming(String fromLanguage, String toLanguage, String code) {
  264.         OpenAi openAi = PARMS.get("OpenAi14");
  265.         return getAiResult(openAi, String.format(openAi.getPrompt(), fromLanguage, toLanguage, fromLanguage, code, toLanguage));
  266.     }
  267.  
  268.     /**
  269.      * 支持批量列表的方式检查情绪
  270.      *
  271.      * @param texts 文本
  272.      * @return
  273.      */
  274.     public static List<CompletionChoice> getBatchTweetClassifier(List<String> texts) {
  275.         OpenAi openAi = PARMS.get("OpenAi15");
  276.         StringJoiner stringJoiner = new StringJoiner("\n");
  277.         for (int i = 0; i < texts.size(); i++) {
  278.             stringJoiner.add((i + 1) + ". " + texts.get(i));
  279.         }
  280.         return getAiResult(openAi, String.format(openAi.getPrompt(), stringJoiner.toString()));
  281.     }
  282.  
  283.     /**
  284.      * 对一段代码进行解释
  285.      *
  286.      * @param code 文本
  287.      * @return
  288.      */
  289.     public static List<CompletionChoice> getExplainCOde(String code) {
  290.         OpenAi openAi = PARMS.get("OpenAi16");
  291.         return getAiResult(openAi, String.format(openAi.getPrompt(), code));
  292.     }
  293.  
  294.     /**
  295.      * 提取一段文本中的关键字
  296.      *
  297.      * @param text 文本
  298.      * @return
  299.      */
  300.     public static List<CompletionChoice> getTextKeywords(String text) {
  301.         OpenAi openAi = PARMS.get("OpenAi17");
  302.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  303.     }
  304.  
  305.     /**
  306.      * 事实回答答题
  307.      *
  308.      * @param text 文本
  309.      * @return
  310.      */
  311.     public static List<CompletionChoice> getFactualAnswering(String text) {
  312.         OpenAi openAi = PARMS.get("OpenAi18");
  313.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  314.     }
  315.  
  316.     /**
  317.      * 给一个产品设计一个广告
  318.      *
  319.      * @param text 文本
  320.      * @return
  321.      */
  322.     public static List<CompletionChoice> getAd(String text) {
  323.         OpenAi openAi = PARMS.get("OpenAi19");
  324.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  325.     }
  326.  
  327.     /**
  328.      * 依据产品描述和种子词语，给一个产品取一个好听的名字
  329.      *
  330.      * @param productDescription 产品描述
  331.      * @param seedWords          种子词语
  332.      * @return
  333.      */
  334.     public static List<CompletionChoice> getProductName(String productDescription, String seedWords) {
  335.         OpenAi openAi = PARMS.get("OpenAi20");
  336.         return getAiResult(openAi, String.format(openAi.getPrompt(), productDescription, seedWords));
  337.     }
  338.  
  339.     /**
  340.      * 把一个长句子简化成一个短句子
  341.      *
  342.      * @param text 长句子
  343.      * @return
  344.      */
  345.     public static List<CompletionChoice> getProductName(String text) {
  346.         OpenAi openAi = PARMS.get("OpenAi21");
  347.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  348.     }
  349.  
  350.     /**
  351.      * 自动修改代码中的bug
  352.      *
  353.      * @param codeType 语言类型
  354.      * @param code     代码
  355.      * @return
  356.      */
  357.     public static List<CompletionChoice> getBugFixer(String codeType, String code) {
  358.         OpenAi openAi = PARMS.get("OpenAi22");
  359.         return getAiResult(openAi, String.format(openAi.getPrompt(), codeType, code, codeType));
  360.     }
  361.  
  362.     /**
  363.      * 自动为一个表格生成数据
  364.      *
  365.      * @param rows    生成的行数
  366.      * @param headers 数据表头，格式如：姓名| 年龄|性别|生日
  367.      * @return
  368.      */
  369.     public static List<CompletionChoice> getFillData(int rows, String headers) {
  370.         OpenAi openAi = PARMS.get("OpenAi23");
  371.         return getAiResult(openAi, String.format(openAi.getPrompt(), rows, headers));
  372.     }
  373.  
  374.     /**
  375.      * 各种开发语言的两天机器人
  376.      *
  377.      * @param question             你的问题
  378.      * @param programmingLanguages 语言 比如Java JavaScript
  379.      * @return
  380.      */
  381.     public static List<CompletionChoice> getProgrammingLanguageChatbot(String question, String programmingLanguages) {
  382.         OpenAi openAi = PARMS.get("OpenAi24");
  383.         return getAiResult(openAi, String.format(openAi.getPrompt(), question, programmingLanguages));
  384.     }
  385.  
  386.     /**
  387.      * 机器学习模型方面的机器人
  388.      *
  389.      * @param question 你的问题
  390.      * @return
  391.      */
  392.     public static List<CompletionChoice> getMLChatbot(String question) {
  393.         OpenAi openAi = PARMS.get("OpenAi25");
  394.         return getAiResult(openAi, String.format(openAi.getPrompt(), question));
  395.     }
  396.  
  397.     /**
  398.      * 可以列出各方面的分类列表，比如歌单
  399.      *
  400.      * @param text 清单描述
  401.      * @return
  402.      */
  403.     public static List<CompletionChoice> getListMaker(String text) {
  404.         OpenAi openAi = PARMS.get("OpenAi26");
  405.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  406.     }
  407.  
  408.     /**
  409.      * 对一段文字进行情绪分析
  410.      *
  411.      * @param text
  412.      * @return
  413.      */
  414.     public static List<CompletionChoice> getTweetClassifier(String text) {
  415.         OpenAi openAi = PARMS.get("OpenAi27");
  416.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  417.     }
  418.  
  419.     /**
  420.      * 抽取文本中的航空代码信息
  421.      *
  422.      * @param text
  423.      * @return
  424.      */
  425.     public static List<CompletionChoice> getAirportCodeExtractor(String text) {
  426.         OpenAi openAi = PARMS.get("OpenAi28");
  427.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  428.     }
  429.  
  430.     /**
  431.      * 无上下文，语句描述生成SQL
  432.      *
  433.      * @param text
  434.      * @return
  435.      */
  436.     public static List<CompletionChoice> getSQL(String text) {
  437.         OpenAi openAi = PARMS.get("OpenAi29");
  438.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  439.     }
  440.  
  441.     /**
  442.      * 从文本中抽取联系方式
  443.      *
  444.      * @param extractContent 抽取内容描述
  445.      * @param text
  446.      * @return 从下面文本中抽取邮箱和电话:\n教育行业A股IPO第一股（股票代码 003032）\n全国咨询/投诉热线：400-618-4000    举报邮箱:mc@itcast.cn
  447.      */
  448.     public static List<CompletionChoice> getExtractContactInformation(String extractContent, String text) {
  449.         OpenAi openAi = PARMS.get("OpenAi30");
  450.         return getAiResult(openAi, String.format(openAi.getPrompt(), extractContent, text));
  451.     }
  452.  
  453.     /**
  454.      * 把一种语言转成另外一种语言代码
  455.      *
  456.      * @param fromCodeType 当前代码类型
  457.      * @param toCodeType   转换的代码类型
  458.      * @param code
  459.      * @return
  460.      */
  461.     public static List<CompletionChoice> getTransformationCode(String fromCodeType, String toCodeType, String code) {
  462.         OpenAi openAi = PARMS.get("OpenAi31");
  463.         return getAiResult(openAi, String.format(openAi.getPrompt(), fromCodeType, toCodeType, fromCodeType, code, toCodeType));
  464.     }
  465.  
  466.     /**
  467.      * 模仿好友聊天
  468.      *
  469.      * @param question
  470.      * @return
  471.      */
  472.     public static List<CompletionChoice> getFriendChat(String question) {
  473.         OpenAi openAi = PARMS.get("OpenAi32");
  474.         return getAiResult(openAi, String.format(openAi.getPrompt(), question));
  475.     }
  476.  
  477.     /**
  478.      * 依据描述生成对应颜色
  479.      *
  480.      * @param text
  481.      * @return
  482.      */
  483.     public static List<CompletionChoice> getMoodToColor(String text) {
  484.         OpenAi openAi = PARMS.get("OpenAi33");
  485.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  486.     }
  487.  
  488.     /**
  489.      * 自动为程序生成文档
  490.      *
  491.      * @param codeType 语言
  492.      * @param code
  493.      * @return
  494.      */
  495.     public static List<CompletionChoice> getCodeDocument(String codeType, String code) {
  496.         OpenAi openAi = PARMS.get("OpenAi34");
  497.         return getAiResult(openAi, String.format(openAi.getPrompt(), codeType, code));
  498.     }
  499.  
  500.     /**
  501.      * 依据短语生成相关文短
  502.      *
  503.      * @param text 短语
  504.      * @return
  505.      */
  506.     public static List<CompletionChoice> getCreateAnalogies(String text) {
  507.         OpenAi openAi = PARMS.get("OpenAi35");
  508.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  509.     }
  510.  
  511.     /**
  512.      * 把多行代码简单的压缩成一行
  513.      *
  514.      * @param codeType 语言
  515.      * @param code
  516.      * @return
  517.      */
  518.     public static List<CompletionChoice> getCodeLine(String codeType, String code) {
  519.         OpenAi openAi = PARMS.get("OpenAi36");
  520.         return getAiResult(openAi, String.format(openAi.getPrompt(), codeType, code, codeType));
  521.     }
  522.  
  523.     /**
  524.      * 依据一个主题创建一个故事
  525.      *
  526.      * @param topic 创作主题
  527.      * @return
  528.      */
  529.     public static List<CompletionChoice> getStory(String topic) {
  530.         OpenAi openAi = PARMS.get("OpenAi37");
  531.         return getAiResult(openAi, String.format(openAi.getPrompt(), topic));
  532.     }
  533.  
  534.     /**
  535.      * 第一人称转第3人称
  536.      *
  537.      * @param text
  538.      * @return
  539.      */
  540.     public static List<CompletionChoice> getStoryCreator(String text) {
  541.         OpenAi openAi = PARMS.get("OpenAi38");
  542.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  543.     }
  544.  
  545.     /**
  546.      * 依据笔记生成摘要说明
  547.      *
  548.      * @param scene 生成的摘要场景
  549.      * @param note  记录的笔记
  550.      * @return
  551.      */
  552.     public static List<CompletionChoice> getNotesToSummary(String scene, String note) {
  553.         OpenAi openAi = PARMS.get("OpenAi39");
  554.         return getAiResult(openAi, String.format(openAi.getPrompt(), note));
  555.     }
  556.  
  557.     /**
  558.      * 给定一个主题，让其生成一些主题相关的想法
  559.      *
  560.      * @param topic 头脑风暴关键词
  561.      * @return
  562.      */
  563.     public static List<CompletionChoice> getIdeaGenerator(String topic) {
  564.         OpenAi openAi = PARMS.get("OpenAi40");
  565.         return getAiResult(openAi, String.format(openAi.getPrompt(), topic));
  566.     }
  567.  
  568.     /**
  569.      * 按照ESRB进行文本分类
  570.      *
  571.      * @param text 文本
  572.      * @return
  573.      */
  574.     public static List<CompletionChoice> getESRBRating(String text) {
  575.         OpenAi openAi = PARMS.get("OpenAi41");
  576.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  577.     }
  578.  
  579.     /**
  580.      * 按照提示为相关内容生成提纲
  581.      *
  582.      * @param text 场景，比如 数据库软件生成大学毕业论文
  583.      * @return
  584.      */
  585.     public static List<CompletionChoice> getEssayOutline(String text) {
  586.         OpenAi openAi = PARMS.get("OpenAi42");
  587.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  588.     }
  589.  
  590.     /**
  591.      * 依据美食名称和材料生成美食的制作步骤
  592.      *
  593.      * @param name        美食名称
  594.      * @param ingredients 美食食材
  595.      * @return
  596.      */
  597.     public static List<CompletionChoice> getRecipeCreator(String name, List<String> ingredients) {
  598.         OpenAi openAi = PARMS.get("OpenAi43");
  599.         StringJoiner joiner = new StringJoiner("\n");
  600.         for (String ingredient : ingredients) {
  601.             joiner.add(ingredient);
  602.         }
  603.         return getAiResult(openAi, String.format(openAi.getPrompt(), name, joiner.toString()));
  604.     }
  605.  
  606.     /**
  607.      * 与AI机器进行聊天
  608.      *
  609.      * @param question
  610.      * @return
  611.      */
  612.     public static List<CompletionChoice> getAiChatbot(String question) {
  613.         OpenAi openAi = PARMS.get("OpenAi44");
  614.         return getAiResult(openAi, String.format(openAi.getPrompt(), question));
  615.     }
  616.  
  617.     /**
  618.      * 与讽刺机器进行聊天，聊天的机器人是一种消极情绪
  619.      *
  620.      * @param question
  621.      * @return
  622.      */
  623.     public static List<CompletionChoice> getMarvChatbot(String question) {
  624.         OpenAi openAi = PARMS.get("OpenAi45");
  625.         return getAiResult(openAi, String.format(openAi.getPrompt(), question));
  626.     }
  627.  
  628.     /**
  629.      * 把一段文本分解成几步来完成
  630.      *
  631.      * @param text
  632.      * @return
  633.      */
  634.     public static List<CompletionChoice> getTurnDirection(String text) {
  635.         OpenAi openAi = PARMS.get("OpenAi46");
  636.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  637.     }
  638.  
  639.     /**
  640.      * 依据文本内容自动生成点评
  641.      *
  642.      * @param text
  643.      * @return
  644.      */
  645.     public static List<CompletionChoice> getReviewCreator(String text) {
  646.         OpenAi openAi = PARMS.get("OpenAi47");
  647.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  648.     }
  649.  
  650.     /**
  651.      * 可以为学习知识自动解答
  652.      *
  653.      * @param text
  654.      * @return
  655.      */
  656.     public static List<CompletionChoice> getStudyNote(String text) {
  657.         OpenAi openAi = PARMS.get("OpenAi48");
  658.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  659.     }
  660.  
  661.     /**
  662.      * 生成面试题
  663.      *
  664.      * @param text
  665.      * @return
  666.      */
  667.     public static List<CompletionChoice> getInterviewQuestion(String text) {
  668.         OpenAi openAi = PARMS.get("OpenAi49");
  669.         System.out.println(String.format(openAi.getPrompt(), text));
  670.         return getAiResult(openAi, String.format(openAi.getPrompt(), text));
  671.     }
  672.  
  673. }

2.4 自动配置类OpenAiAutoConfiguration.java

  1.  
  2. package com.wkf.workrecord.tools.openai;
  3.  
  4. import org.springframework.boot.context.properties.EnableConfigurationProperties;
  5. import org.springframework.context.annotation.Configuration;
  6.  
  7. /**
  8.  * 自动配置类
  9.  * @author wuKeFan
  10.  * @date 2023-02-10 15:34:01
  11.  */
  12. @Configuration
  13. @EnableConfigurationProperties(OpenAiProperties.class)
  14. public class OpenAiAutoConfiguration {
  15. }

2.5 在resources文件夹下的META-INF/spring.factories文件中增加配置

org.springframework.boot.autoconfigure.EnableAutoConfiguration=com.wkf.workrecord.tools.openai.OpenAiAutoConfiguration

2.6 在yml文件上配置token

  1. openai:
  2.   token: 你的token
  3.   timeout: 5000

3 编写测试类

  1.  
  2. package com.wkf.workrecord.study;
  3.  
  4. import com.theokanning.openai.completion.CompletionChoice;
  5. import com.wkf.workrecord.tools.openai.OpenAiUtils;
  6. import lombok.extern.slf4j.Slf4j;
  7. import org.junit.jupiter.api.Test;
  8. import org.springframework.boot.test.context.SpringBootTest;
  9.  
  10. import java.util.List;
  11.  
  12. /**
  13.  * openAi测试类
  14.  * @author wuKeFan
  15.  * @date 2023-02-10 15:37:01
  16.  */
  17. @Slf4j
  18. @SpringBootTest
  19. public class OpenAiTest {
  20.  
  21.     /**
  22.      * openAi接口请求API
  23.      */
  24.     @Test
  25.     public void test() {
  26.         List<CompletionChoice> questionAnswer = OpenAiUtils.getQuestionAnswer("使用SpringBoot框架进行Http请求的详细代码");
  27.         for (CompletionChoice completionChoice : questionAnswer) {
  28.             System.out.println(completionChoice.getText());
  29.         }
  30.         List<CompletionChoice> openAiApi = OpenAiUtils.getOpenAiApi("使用SpringBoot框架进行Http请求");
  31.         for (CompletionChoice completionChoice : openAiApi) {
  32.             System.out.println(completionChoice.getText());
  33.         }
  34.     }
  35.  
  36. }

4 补充

如果使用上述方法出现超时错误的,可以使用这个方法

4.1 添加依赖

  1.         <!-- openAi 最新版依赖 -->
  2.         <dependency>
  3.             <groupId>com.unfbx</groupId>
  4.             <artifactId>chatgpt-java</artifactId>
  5.             <version>1.0.6</version>
  6.         </dependency>

4.2 添加代码

  1.  
  2.         Proxy proxy = new Proxy(Proxy.Type.HTTP, new InetSocketAddress("localhost", 7890));
  3.         //日志输出可以不添加
  4.         //HttpLoggingInterceptor httpLoggingInterceptor = new HttpLoggingInterceptor(new OpenAILogger());
  5.         //httpLoggingInterceptor.setLevel(HttpLoggingInterceptor.Level.BODY);
  6.         OpenAiClient openAiClient = OpenAiClient.builder()
  7.                 .apiKey("sk-***********************************************")
  8.                 .connectTimeout(50)
  9.                 .writeTimeout(50)
  10.                 .readTimeout(50)
  11.                 .proxy(proxy)
  12.                 //.interceptor(Collections.singletonList(httpLoggingInterceptor))
  13.                 .apiHost("https://api.openai.com/")
  14.                 .build();
  15.         CompletionResponse completions = openAiClient.completions("你是openAi吗");
  16.         Arrays.stream(completions.getChoices()).forEach(System.out::println);

5 总结

到此这篇关于SpringBoot整合chatGPT的文章就介绍到这了

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/爱喝兽奶帝天荒/article/detail/937582
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号