提交 ecc403e1 编写于 作者: 知源笔记's avatar 知源笔记

feat: stop chars

上级 72f402a2
run = "update-java-alternatives --set java-1.8.0-openjdk-amd64 && mvn -DskipTests clean package && java -jar target/askdata-backend-1.0-SNAPSHOT.jar --spring.profiles.active=prod"
run = "update-java-alternatives --set java-1.8.0-openjdk-amd64 && mvn -DskipTests package && java -jar target/askdata-backend-1.0-SNAPSHOT.jar --spring.profiles.active=prod"
......@@ -132,6 +132,8 @@ public class BiVenusController {
log.info("User msg: {}", userMsg);
GptRequest completion = GptRequest.newRequest("Translate natural language to SQL queries.");
completion.setStop(Arrays.asList("#", ";"));
completion.setTemperature(0.0);
completion.addUserMsg(userMsg);
completion.setApiKey(System.getenv("INSCODE_API_KEY"));
String reply = gptClient.chatCompletion(completion);
......
......@@ -40,6 +40,7 @@ public class GptClient {
public String chatCompletion(GptRequest completion) {
String json = JSON.toJSONString(completion);
log.debug("GptRequest: {}", json);
RequestBody requestBody = RequestBody.create(MediaType.parse("application/json; charset=utf-8"), json);
Request request = new Request.Builder()
......
......@@ -3,6 +3,8 @@ package org.enthusa.askdata.ext.inscode;
import com.alibaba.fastjson.annotation.JSONField;
import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.Data;
import lombok.Getter;
import lombok.Setter;
import java.util.ArrayList;
import java.util.List;
......@@ -17,7 +19,21 @@ public class GptRequest {
System, Assistant, User
}
@Getter
private List<Message> messages;
/**
* 在0和2之间使用什么采样温度?较高的值如0.8会使输出更随机,而较低的值如0.2会使其更加集中和确定性。
* 默认值: 1
*/
@Setter
@Getter
private Double temperature;
@Setter
@Getter
private List<String> stop;
@JsonProperty("apikey")
@JSONField(name = "apikey")
private String apiKey;
......
......@@ -23,7 +23,7 @@
</encoder>
</appender>
<logger name="com.nowcoder" level="DEBUG"/>
<logger name="org.enthusa" level="DEBUG"/>
<root level="INFO">
<appender-ref ref="STDOUT"/>
......
......@@ -6,6 +6,7 @@ import org.enthusa.askdata.ext.inscode.GptRequest;
import org.junit.Test;
import javax.annotation.Resource;
import java.util.Arrays;
/**
* @author henry
......@@ -19,6 +20,7 @@ public class ChatGptTest extends AbstractTest {
public void test() throws InterruptedException {
GptRequest request = GptRequest.newRequest("Translate natural language to SQL queries.");
request.addUserMsg("你能做什么?");
request.setStop(Arrays.asList("#", ";"));
request.setApiKey(System.getenv("INSCODE_API_KEY"));
System.out.println(gptClient.chatCompletion(request));
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册