Java調(diào)用ChatGPT的實(shí)現(xiàn)代碼

更新時(shí)間：2023年02月18日 14:13:08 作者：mikey桑

這篇文章主要介紹了Java調(diào)用ChatGPT的實(shí)現(xiàn)代碼,本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下

Java調(diào)用ChatGPT的小插件

1. ChatGPT賬號(hào)準(zhǔn)備

很多博文有介紹怎么獲取賬號(hào)，具體的可自行搜索。
準(zhǔn)備好ChatGPT帳號(hào)之后打開(kāi)openai的官網(wǎng)去創(chuàng)建API KEYS，鏈接：https://platform.openai.com/account/api-keys。
注意：這里的API KEYS創(chuàng)建好以后一定要妥善保存，創(chuàng)建以后，第二次就無(wú)法再查看了，想要再看，只能刪除了API KEYS然后重新創(chuàng)建。

這里的API KEYS妥善保管后面會(huì)用到。

2. 配置階段

2.1 依賴引入

pom.xml中引入依賴

        <dependency>
            <groupId>io.github.asleepyfish</groupId>
            <artifactId>chatgpt</artifactId>
            <version>1.0.3</version>
        </dependency>

2.2 配置application.yml文件

在application.yml文件中配置chatgpt相關(guān)參數(shù)

chatgpt:
  model: text-davinci-003
  token: sk-xxxxxxxxxxxxxxxxxxx
  retries: 10

這里的model是選擇chatgpt哪個(gè)模型，默認(rèn)填好的是最優(yōu)的模型了，token就是上面申請(qǐng)的API KEYS，retries指的是當(dāng)chatgpt第一次請(qǐng)求回答失敗時(shí)，重新請(qǐng)求的次數(shù)（增加該參數(shù)的原因是因?yàn)榇罅吭L問(wèn)的原因，在某一個(gè)時(shí)刻，chatgpt服務(wù)將處于無(wú)法訪問(wèn)的情況）

2.3 @EnableChatGPT注解

啟動(dòng)類上加入@EnableChatGPT注解則將ChatGPT服務(wù)注入到Spring中。

在這里插入圖片描述

3. 使用

提供了工具類OpenAiUtils，里面提供了相關(guān)方法進(jìn)行調(diào)用。
其中最簡(jiǎn)單的使用方法是：

OpenAiUtils.createCompletion(prompt);

入?yún)?code>prompt即輸入的問(wèn)題的字符串。

還提供一個(gè)通用的靜態(tài)方法是

public static List<String> createCompletion(CompletionRequest completionRequest) {...}

入?yún)?code>CompletionRequest 里包含模型的一些可調(diào)參數(shù)。

OpenAiUtils類中還提供了多個(gè)可供選擇的靜態(tài)方法，可以自行查看。

上述方法的返回參數(shù)是一個(gè)list，是因?yàn)檎{(diào)整參數(shù)返回答案n可以一次性返回多條不同的解答（n為CompletionRequest類中一個(gè)參數(shù)）。

4. 測(cè)試

demo的Git地址：

https://github.com/asleepyfish/chatgpt-demo

測(cè)試代碼：

@SpringBootTest
public class SpringTest {
    @Test
    public void chatGPTTest() {
        OpenAiUtils.createCompletion("use c++ write QuickSort").forEach(System.out::println);
    }
}

ChatGPT輸出結(jié)果：

#include <iostream>?
using namespace std;?

// A utility function to swap two elements?
void swap(int* a, int* b)?
{?
?? ?int t = *a;?
?? ?*a = *b;?
?? ?*b = t;?
}?

/* This function takes last element as pivot, places?
the pivot element at its correct position in sorted?
array, and places all smaller (smaller than pivot)?
to left of pivot and all greater elements to right?
of pivot */
int partition (int arr[], int low, int high)?
{?
?? ?int pivot = arr[high]; // pivot?
?? ?int i = (low - 1); // Index of smaller element?

?? ?for (int j = low; j <= high - 1; j++)?
?? ?{?
?? ??? ?// If current element is smaller than the pivot?
?? ??? ?if (arr[j] < pivot)?
?? ??? ?{?
?? ??? ??? ?i++; // increment index of smaller element?
?? ??? ??? ?swap(&arr[i], &arr[j]);?
?? ??? ?}?
?? ?}?
?? ?swap(&arr[i + 1], &arr[high]);?
?? ?return (i + 1);?
}?

/* The main function that implements QuickSort?
arr[] --> Array to be sorted,?
low --> Starting index,?
high --> Ending index */
void quickSort(int arr[], int low, int high)?
{?
?? ?if (low < high)?
?? ?{?
?? ??? ?/* pi is partitioning index, arr[p] is now?
?? ??? ?at right place */
?? ??? ?int pi = partition(arr, low, high);?

?? ??? ?// Separately sort elements before?
?? ??? ?// partition and after partition?
?? ??? ?quickSort(arr, low, pi - 1);?
?? ??? ?quickSort(arr, pi + 1, high);?
?? ?}?
}?

/* Function to print an array */
void printArray(int arr[], int size)?
{?
?? ?int i;?
?? ?for (i = 0; i < size; i++)?
?? ??? ?cout << arr[i] << " ";?
?? ?cout << endl;?
}?

// Driver Code?
int main()?
{?
?? ?int arr[] = {10, 7, 8, 9, 1, 5};?
?? ?int n = sizeof(arr) / sizeof(arr[0]);?
?? ?quickSort(arr, 0, n - 1);?
?? ?cout << "Sorted array: " << endl;?
?? ?printArray(arr, n);?
?? ?return 0;?
}

到此這篇關(guān)于Java調(diào)用ChatGPT的實(shí)現(xiàn)代碼的文章就介紹到這了,更多相關(guān)Java調(diào)用ChatGPT內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: