ChatAI-v2

This commit is contained in:
slhaf
2024-09-28 19:10:41 +08:00
parent c042c3cc61
commit 34cb78adf9
12 changed files with 775 additions and 0 deletions

131
.gitignore vendored Normal file
View File

@@ -0,0 +1,131 @@
# User-specific stuff
.idea/
*.iml
*.ipr
*.iws
# IntelliJ
out/
# mpeltonen/sbt-idea plugin
.idea_modules/
# JIRA plugin
atlassian-ide-plugin.xml
# Compiled class file
*.class
# Log file
*.log
# BlueJ files
*.ctxt
# Package Files #
*.jar
*.war
*.nar
*.ear
*.zip
*.tar.gz
*.rar
# virtual machine crash logs, see http://www.java.com/en/download/help/error_hotspot.xml
hs_err_pid*
*~
# temporary files which can be created if a process still has a handle open of a deleted file
.fuse_hidden*
# KDE directory preferences
.directory
# Linux trash folder which might appear on any partition or disk
.Trash-*
# .nfs files are created when an open file is removed but is still being accessed
.nfs*
# General
.DS_Store
.AppleDouble
.LSOverride
# Icon must end with two \r
Icon
# Thumbnails
._*
# Files that might appear in the root of a volume
.DocumentRevisions-V100
.fseventsd
.Spotlight-V100
.TemporaryItems
.Trashes
.VolumeIcon.icns
.com.apple.timemachine.donotpresent
# Directories potentially created on remote AFP share
.AppleDB
.AppleDesktop
Network Trash Folder
Temporary Items
.apdisk
# Windows thumbnail cache files
Thumbs.db
Thumbs.db:encryptable
ehthumbs.db
ehthumbs_vista.db
# Dump file
*.stackdump
# Folder config file
[Dd]esktop.ini
# Recycle Bin used on file shares
$RECYCLE.BIN/
# Windows Installer files
*.cab
*.msi
*.msix
*.msm
*.msp
# Windows shortcuts
*.lnk
.gradle
build/
# Ignore Gradle GUI config
gradle-app.setting
# Cache of project
.gradletasknamecache
**/build/
# Common working directory
run/
# Avoid ignoring Gradle wrapper jar file (.jar files are usually ignored)
!gradle-wrapper.jar
# Local Test Launch point
src/test/kotlin/RunTerminal.kt
# Mirai console files with direct bootstrap
/config
/data
/plugins
/bots
# Local Test Launch Point working directory
/debug-sandbox

23
.run/RunTerminal.run.xml Normal file
View File

@@ -0,0 +1,23 @@
<component name="ProjectRunConfigurationManager">
<configuration default="false" name="Run Mirai Console" type="GradleRunConfiguration" factoryName="Gradle" folderName="Mirai">
<ExternalSystemSettings>
<option name="executionName" />
<option name="externalProjectPath" value="$PROJECT_DIR$" />
<option name="externalSystemIdString" value="GRADLE" />
<option name="scriptParameters" value="" />
<option name="taskDescriptions">
<list />
</option>
<option name="taskNames">
<list>
<option value=":runConsole" />
</list>
</option>
<option name="vmOptions" />
</ExternalSystemSettings>
<ExternalSystemDebugServerProcess>true</ExternalSystemDebugServerProcess>
<ExternalSystemReattachDebugProcess>true</ExternalSystemReattachDebugProcess>
<DebugAllEnabled>false</DebugAllEnabled>
<method v="2" />
</configuration>
</component>

29
build.gradle.kts Normal file
View File

@@ -0,0 +1,29 @@
plugins {
val kotlinVersion = "1.8.10"
kotlin("jvm") version kotlinVersion
kotlin("plugin.serialization") version kotlinVersion
id("net.mamoe.mirai-console") version "2.16.0"
}
group = "com.plugin"
version = "0.1.0"
dependencies{
implementation ("com.squareup.okhttp3:okhttp:5.0.0-alpha.12")
implementation ("org.yaml:snakeyaml:2.2")
implementation ("cn.hutool:hutool-all:5.8.28")
implementation("cn.bigmodel.openapi:oapi-java-sdk:release-V4-2.0.2")
implementation ("com.aliyun:ocr_api20210707:3.1.1")
implementation ("junit:junit:4.13.2")
}
repositories {
maven("https://maven.aliyun.com/repository/public")
mavenCentral()
}
mirai {
jvmTarget = JavaVersion.VERSION_17
}

1
gradle.properties Normal file
View File

@@ -0,0 +1 @@
kotlin.code.style=official

BIN
gradle/wrapper/gradle-wrapper.jar vendored Normal file

Binary file not shown.

View File

@@ -0,0 +1,5 @@
distributionBase=GRADLE_USER_HOME
distributionPath=wrapper/dists
distributionUrl=https://mirrors.cloud.tencent.com/gradle/gradle-7.3.1-all.zip
zipStoreBase=GRADLE_USER_HOME
zipStorePath=wrapper/dists

234
gradlew vendored Normal file
View File

@@ -0,0 +1,234 @@
\\#!/bin/sh
\\#
\\# Copyright © 2015-2021 the original authors.
\\#
\\# Licensed under the Apache License, Version 2.0 (the "License");
\\# you may not use this file except in compliance with the License.
\\# You may obtain a copy of the License at
\\#
\\# https://www.apache.org/licenses/LICENSE-2.0
\\#
\\# Unless required by applicable law or agreed to in writing, software
\\# distributed under the License is distributed on an "AS IS" BASIS,
\\# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
\\# See the License for the specific language governing permissions and
\\# limitations under the License.
\\#
\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#
\\#
\\# Gradle start up script for POSIX generated by Gradle.
\\#
\\# Important for running:
\\#
\\# (1) You need a POSIX-compliant shell to run this script. If your /bin/sh is
\\# noncompliant, but you have some other compliant shell such as ksh or
\\# bash, then to run this script, type that shell name before the whole
\\# command line, like:
\\#
\\# ksh Gradle
\\#
\\# Busybox and similar reduced shells will NOT work, because this script
\\# requires all of these POSIX shell features:
\\# * functions;
\\# * expansions «\#[[\#[[\$]]#]]\#var», «\#[[\#[[\$]]#]]\#{var}», «\#[[\#[[\$]]#]]\#{var:-default}», «\#[[\#[[\$]]#]]\#{var+SET}»,
\\# «\#[[\#[[\$]]#]]\#{var\\#prefix}», «\#[[\#[[\$]]#]]\#{var%suffix}», and «\#[[\#[[\$]]#]]\#( cmd )»;
\\# * compound commands having a testable exit status, especially «case»;
\\# * various built-in commands including «command», «set», and «ulimit».
\\#
\\# Important for patching:
\\#
\\# (2) This script targets any POSIX shell, so it avoids extensions provided
\\# by Bash, Ksh, etc; in particular arrays are avoided.
\\#
\\# The "traditional" practice of packing multiple parameters into a
\\# space-separated string is a well documented source of bugs and security
\\# problems, so this is (mostly) avoided, by progressively accumulating
\\# options in "\#[[\#[[\$]]#]]\#@", and eventually passing that to Java.
\\#
\\# Where the inherited environment variables (DEFAULT_JVM_OPTS, JAVA_OPTS,
\\# and GRADLE_OPTS) rely on word-splitting, this is performed explicitly;
\\# see the in-line comments for details.
\\#
\\# There are tweaks for specific operating systems such as AIX, CygWin,
\\# Darwin, MinGW, and NonStop.
\\#
\\# (3) This script is generated from the Groovy template
\\# https://github.com/gradle/gradle/blob/master/subprojects/plugins/src/main/resources/org/gradle/api/internal/plugins/unixStartScript.txt
\\# within the Gradle project.
\\#
\\# You can find Gradle at https://github.com/gradle/gradle/.
\\#
\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#
\\# Attempt to set APP_HOME
\\# Resolve links: \#[[\#[[\$]]#]]\#0 may be a link
app_path=\#[[\#[[\$]]#]]\#0
\\# Need this for daisy-chained symlinks.
while
APP_HOME=\#[[\#[[\$]]#]]\#{app_path%"\#[[\#[[\$]]#]]\#{app_path\\#\\#*/}"} \\# leaves a trailing /; empty if no leading path
[ -h "\#[[\#[[\$]]#]]\#app_path" ]
do
ls=\#[[\#[[\$]]#]]\#( ls -ld "\#[[\#[[\$]]#]]\#app_path" )
link=\#[[\#[[\$]]#]]\#{ls\\#*' -> '}
case \#[[\#[[\$]]#]]\#link in \\#(
/*) app_path=\#[[\#[[\$]]#]]\#link ;; \\#(
*) app_path=\#[[\#[[\$]]#]]\#APP_HOME\#[[\#[[\$]]#]]\#link ;;
esac
done
APP_HOME=\#[[\#[[\$]]#]]\#( cd "\#[[\#[[\$]]#]]\#{APP_HOME:-./}" && pwd -P ) || exit
APP_NAME="Gradle"
APP_BASE_NAME=\#[[\#[[\$]]#]]\#{0\\#\\#*/}
\\# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"'
\\# Use the maximum available, or set MAX_FD != -1 to use that value.
MAX_FD=maximum
warn () {
echo "\#[[\#[[\$]]#]]\#*"
} >&2
die () {
echo
echo "\#[[\#[[\$]]#]]\#*"
echo
exit 1
} >&2
\\# OS specific support (must be 'true' or 'false').
cygwin=false
msys=false
darwin=false
nonstop=false
case "\#[[\#[[\$]]#]]\#( uname )" in \\#(
CYGWIN* ) cygwin=true ;; \\#(
Darwin* ) darwin=true ;; \\#(
MSYS* | MINGW* ) msys=true ;; \\#(
NONSTOP* ) nonstop=true ;;
esac
CLASSPATH=\#[[\#[[\$]]#]]\#APP_HOME/gradle/wrapper/gradle-wrapper.jar
\\# Determine the Java command to use to start the JVM.
if [ -n "\#[[\#[[\$]]#]]\#JAVA_HOME" ] ; then
if [ -x "\#[[\#[[\$]]#]]\#JAVA_HOME/jre/sh/java" ] ; then
\\# IBM's JDK on AIX uses strange locations for the executables
JAVACMD=\#[[\#[[\$]]#]]\#JAVA_HOME/jre/sh/java
else
JAVACMD=\#[[\#[[\$]]#]]\#JAVA_HOME/bin/java
fi
if [ ! -x "\#[[\#[[\$]]#]]\#JAVACMD" ] ; then
die "ERROR: JAVA_HOME is set to an invalid directory: \#[[\#[[\$]]#]]\#JAVA_HOME
Please set the JAVA_HOME variable in your environment to match the
location of your Java installation."
fi
else
JAVACMD=java
which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
Please set the JAVA_HOME variable in your environment to match the
location of your Java installation."
fi
\\# Increase the maximum file descriptors if we can.
if ! "\#[[\#[[\$]]#]]\#cygwin" && ! "\#[[\#[[\$]]#]]\#darwin" && ! "\#[[\#[[\$]]#]]\#nonstop" ; then
case \#[[\#[[\$]]#]]\#MAX_FD in \\#(
max*)
MAX_FD=\#[[\#[[\$]]#]]\#( ulimit -H -n ) ||
warn "Could not query maximum file descriptor limit"
esac
case \#[[\#[[\$]]#]]\#MAX_FD in \\#(
'' | soft) :;; \\#(
*)
ulimit -n "\#[[\#[[\$]]#]]\#MAX_FD" ||
warn "Could not set maximum file descriptor limit to \#[[\#[[\$]]#]]\#MAX_FD"
esac
fi
\\# Collect all arguments for the java command, stacking in reverse order:
\\# * args from the command line
\\# * the main class name
\\# * -classpath
\\# * -D...appname settings
\\# * --module-path (only if needed)
\\# * DEFAULT_JVM_OPTS, JAVA_OPTS, and GRADLE_OPTS environment variables.
\\# For Cygwin or MSYS, switch paths to Windows format before running java
if "\#[[\#[[\$]]#]]\#cygwin" || "\#[[\#[[\$]]#]]\#msys" ; then
APP_HOME=\#[[\#[[\$]]#]]\#( cygpath --path --mixed "\#[[\#[[\$]]#]]\#APP_HOME" )
CLASSPATH=\#[[\#[[\$]]#]]\#( cygpath --path --mixed "\#[[\#[[\$]]#]]\#CLASSPATH" )
JAVACMD=\#[[\#[[\$]]#]]\#( cygpath --unix "\#[[\#[[\$]]#]]\#JAVACMD" )
\\# Now convert the arguments - kludge to limit ourselves to /bin/sh
for arg do
if
case \#[[\#[[\$]]#]]\#arg in \\#(
-*) false ;; \\# don't mess with options \\#(
/?*) t=\#[[\#[[\$]]#]]\#{arg\\#/} t=/\#[[\#[[\$]]#]]\#{t%%/*} \\# looks like a POSIX filepath
[ -e "\#[[\#[[\$]]#]]\#t" ] ;; \\#(
*) false ;;
esac
then
arg=\#[[\#[[\$]]#]]\#( cygpath --path --ignore --mixed "\#[[\#[[\$]]#]]\#arg" )
fi
\\# Roll the args list around exactly as many times as the number of
\\# args, so each arg winds up back in the position where it started, but
\\# possibly modified.
\\#
\\# NB: a `for` loop captures its iteration list before it begins, so
\\# changing the positional parameters here affects neither the number of
\\# iterations, nor the values presented in `arg`.
shift \\# remove old arg
set -- "\#[[\#[[\$]]#]]\#@" "\#[[\#[[\$]]#]]\#arg" \\# push replacement arg
done
fi
\\# Collect all arguments for the java command;
\\# * \#[[\#[[\$]]#]]\#DEFAULT_JVM_OPTS, \#[[\#[[\$]]#]]\#JAVA_OPTS, and \#[[\#[[\$]]#]]\#GRADLE_OPTS can contain fragments of
\\# shell script including quotes and variable substitutions, so put them in
\\# double quotes to make sure that they get re-expanded; and
\\# * put everything else in single quotes, so that it's not re-expanded.
set -- \
"-Dorg.gradle.appname=\#[[\#[[\$]]#]]\#APP_BASE_NAME" \
-classpath "\#[[\#[[\$]]#]]\#CLASSPATH" \
org.gradle.wrapper.GradleWrapperMain \
"\#[[\#[[\$]]#]]\#@"
\\# Use "xargs" to parse quoted args.
\\#
\\# With -n1 it outputs one arg per line, with the quotes and backslashes removed.
\\#
\\# In Bash we could simply go:
\\#
\\# readarray ARGS < <( xargs -n1 <<<"\#[[\#[[\$]]#]]\#var" ) &&
\\# set -- "\#[[\#[[\$]]#]]\#{ARGS[@]}" "\#[[\#[[\$]]#]]\#@"
\\#
\\# but POSIX shell has neither arrays nor command substitution, so instead we
\\# post-process each arg (as a line of input to sed) to backslash-escape any
\\# character that might be a shell metacharacter, then use eval to reverse
\\# that process (while maintaining the separation between arguments), and wrap
\\# the whole thing up as a single "set" statement.
\\#
\\# This will of course break if any of these variables contains a newline or
\\# an unmatched quote.
\\#
eval "set -- \#[[\#[[\$]]#]]\#(
printf '%s\n' "\#[[\#[[\$]]#]]\#DEFAULT_JVM_OPTS \#[[\#[[\$]]#]]\#JAVA_OPTS \#[[\#[[\$]]#]]\#GRADLE_OPTS" |
xargs -n1 |
sed ' s~[^-[:alnum:]+,./:=@_]~\\&~g; ' |
tr '\n' ' '
)" '"\#[[\#[[\$]]#]]\#@"'
exec "\#[[\#[[\$]]#]]\#JAVACMD" "\#[[\#[[\$]]#]]\#@"

89
gradlew.bat vendored Normal file
View File

@@ -0,0 +1,89 @@
@rem
@rem Copyright 2015 the original author or authors.
@rem
@rem Licensed under the Apache License, Version 2.0 (the "License");
@rem you may not use this file except in compliance with the License.
@rem You may obtain a copy of the License at
@rem
@rem https://www.apache.org/licenses/LICENSE-2.0
@rem
@rem Unless required by applicable law or agreed to in writing, software
@rem distributed under the License is distributed on an "AS IS" BASIS,
@rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@rem See the License for the specific language governing permissions and
@rem limitations under the License.
@rem
@if "%DEBUG%" == "" @echo off
@rem \\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#
@rem
@rem Gradle startup script for Windows
@rem
@rem \\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#\\#
@rem Set local scope for the variables with windows NT shell
if "%OS%"=="Windows_NT" setlocal
set DIRNAME=%~dp0
if "%DIRNAME%" == "" set DIRNAME=.
set APP_BASE_NAME=%~n0
set APP_HOME=%DIRNAME%
@rem Resolve any "." and ".." in APP_HOME to make it shorter.
for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi
@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m"
@rem Find java.exe
if defined JAVA_HOME goto findJavaFromJavaHome
set JAVA_EXE=java.exe
%JAVA_EXE% -version >NUL 2>&1
if "%ERRORLEVEL%" == "0" goto execute
echo.
echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
echo.
echo Please set the JAVA_HOME variable in your environment to match the
echo location of your Java installation.
goto fail
:findJavaFromJavaHome
set JAVA_HOME=%JAVA_HOME:"=%
set JAVA_EXE=%JAVA_HOME%/bin/java.exe
if exist "%JAVA_EXE%" goto execute
echo.
echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
echo.
echo Please set the JAVA_HOME variable in your environment to match the
echo location of your Java installation.
goto fail
:execute
@rem Setup the command line
set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
@rem Execute Gradle
"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %*
:end
@rem End local scope for the variables with windows NT shell
if "%ERRORLEVEL%"=="0" goto mainEnd
:fail
rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
rem the _cmd.exe /c_ return code!
if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
exit /b 1
:mainEnd
if "%OS%"=="Windows_NT" endlocal
:omega

7
settings.gradle.kts Normal file
View File

@@ -0,0 +1,7 @@
pluginManagement {
repositories {
maven("https://maven.aliyun.com/repository/gradle-plugin")
gradlePluginPortal()
}
}
rootProject.name = "ChatAI-InGroup-v2"

135
src/test/java/MyTest.java Normal file
View File

@@ -0,0 +1,135 @@
import com.alibaba.fastjson.JSONObject;
import lombok.Data;
import org.apache.http.Header;
import org.apache.http.HttpEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicHeader;
import org.apache.http.util.EntityUtils;
import org.junit.Test;
import java.io.IOException;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class MyTest {
public static void www(String[] args) {
String miraiCode = "看下这个题目[mirai:image:https://multimedia.nt.qq.com.cn/download?appid=1407&fileid=CgoyOTk4ODEzODgyEhQW4G7W3v2lT-NKItOqORRIbuK6rxiZzgcg_woo4u2zytCqiANQgL2jAQ&spec=0&rkey=CAMSKMa3OFokB_Tlf6uVSrtr8zFEBtyDtGIdmuztQCnfO9gIDN19LUsLquI]";
String url = null;
if (miraiCode.matches(".*[mirai:image:(https?://[\\w./?&amp;=]+)].*")) {
String regex = "\\[mirai:image:(.*?)]";
Pattern pattern = Pattern.compile(regex);
// 创建Matcher对象
Matcher matcher = pattern.matcher(miraiCode);
// 查找并提取链接
if (matcher.find()) {
url = matcher.group(1); // 提取第一个括号内的内容
}
}
System.out.println(url);
}
@Test
public void sseTest() throws IOException {
@Data
class Content {
private String type;
private String text;
public Content() {
}
public Content(String type, String text) {
this.type = type;
this.text = text;
}
}
@Data
class Message {
private String role;
private Content[] content;
}
CloseableHttpClient client = HttpClients.createDefault();
HttpPost httpPost = new HttpPost("https://open.bigmodel.cn/api/paas/v4/assistant");
httpPost.setHeaders(new Header[]{
new BasicHeader("content-type", "application/json"),
new BasicHeader("authorization", "abce8567e434a0e4b987c34763c23c73.NCwRwbrkW9Mi6t63")
});
JSONObject jsonObject = new JSONObject();
jsonObject.put("assistant_id", "65a265419d72d299a9230616");
jsonObject.put("model", "glm-4-assistant");
jsonObject.put("stream", "True");
Message message = new Message();
message.role = "user";
Content[] contents = new Content[]{new Content("text", "改写提示词现在你是一名Java开发工程师...")};
message.content = contents;
Message[] messages = new Message[]{message};
jsonObject.put("messages", messages);
HttpEntity httpEntity = new StringEntity(jsonObject.toString(), "UTF-8");
httpPost.setEntity(httpEntity);
CloseableHttpResponse response = client.execute(httpPost);
System.out.println(response.getStatusLine());
HttpEntity entity = response.getEntity();
String result = EntityUtils.toString(entity, "UTF-8");
String[] strs = result.split("\n\n");
for (String str : strs) {
pojo.Data data = JSONObject.parseObject(str.substring(6), pojo.Data.class);
if (data.getStatus().equals("completed")) {
break;
}
//解析返回结果
// System.out.println(data.getChoices().get(0).getDelta());
String delta = data.getChoices().get(0).getDelta();
JSONObject deltaJsonObject = JSONObject.parseObject(delta);
String role = deltaJsonObject.get("role").toString();
if ("assistant".equals(role)) {
//输出content
System.out.print(deltaJsonObject.get("content"));
} else if ("tool".equals(role)) {
//解析tool_call内容根据字段type
JSONObject toolCall = deltaJsonObject.getJSONArray("tool_calls").getJSONObject(0);
String type = toolCall.get("type").toString();
if ("function".equals(type)) {
//如果调用了函数
JSONObject functionJsonObject = toolCall.getJSONObject("function");
if (functionJsonObject.get("outputs") != null) {
String contentUrl = functionJsonObject.getJSONArray("outputs").getJSONObject(0).getJSONObject("content").get("url").toString();
System.out.println("\r\n" + contentUrl);
}
} else if ("drawing_tool".equals(type)) {
//如果是绘画工具
JSONObject drawingToolJsonObject = toolCall.getJSONObject("drawing_tool");
if (drawingToolJsonObject.get("outputs") != null) {
String imageUrl = drawingToolJsonObject.getJSONArray("outputs").getJSONObject(0).get("image").toString();
System.out.println(imageUrl);
}
} else if ("web_browser".equals(type)) {
JSONObject webBrowserJsonObject = toolCall.getJSONObject("web_browser");
if (webBrowserJsonObject.get("outputs") != null) {
for (Object o : webBrowserJsonObject.getJSONArray("outputs")) {
System.out.println(JSONObject.parseObject(o.toString()).get("title"));
System.out.println(JSONObject.parseObject(o.toString()).get("link"));
System.out.println(JSONObject.parseObject(o.toString()).get("content"));
}
}
}
}
}
response.close();
client.close();
}
}

View File

@@ -0,0 +1,28 @@
package pojo;
public class Choices {
/**
* index
*/
private int index;
/**
* delta
*/
private String delta;
public int getIndex() {
return index;
}
public void setIndex(int index) {
this.index = index;
}
public String getDelta() {
return delta;
}
public void setDelta(String delta) {
this.delta = delta;
}
}

View File

@@ -0,0 +1,93 @@
package pojo;
import java.util.List;
public class Data {
/**
* id
*/
private String id;
/**
* created
*/
private long created;
/**
* model
*/
private String model;
/**
* choices
*/
private List<Choices> choices;
/**
* status
*/
private String status;
/**
* assistant_id
*/
private String assistant_id;
/**
* conversation_id
*/
private String conversation_id;
public String getId() {
return id;
}
public void setId(String id) {
this.id = id;
}
public long getCreated() {
return created;
}
public void setCreated(long created) {
this.created = created;
}
public String getModel() {
return model;
}
public void setModel(String model) {
this.model = model;
}
public List<Choices> getChoices() {
return choices;
}
public void setChoices(List<Choices> choices) {
this.choices = choices;
}
public String getStatus() {
return status;
}
public void setStatus(String status) {
this.status = status;
}
public String getAssistant_id() {
return assistant_id;
}
public void setAssistant_id(String assistant_id) {
this.assistant_id = assistant_id;
}
public String getConversation_id() {
return conversation_id;
}
public void setConversation_id(String conversation_id) {
this.conversation_id = conversation_id;
}
}