iSharkFly-Open
/
WeChat-Service
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

USVT-136 GPT 超时时间设置为 30s

This commit is contained in:
YuCheng Hu 2023-02-18 08:01:45 -05:00
parent 79481d5118
commit 9cdaa97974
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/northtecom/visatrack/api/service/impl/ChatGPTService.java
View File

@ -107,7 +107,7 @@ public class ChatGPTService {
*/
private String getChatGPTContent(String content) {
log.debug(">>>>>>>> ChatGPT Content Call");
OpenAiService service = new OpenAiService("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");
OpenAiService service = new OpenAiService("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", 30);
CompletionRequest completionRequest = new CompletionRequestBuilder().setModel("text-davinci-003")
.setPrompt(content)
.setMaxTokens(500)
@ -120,7 +120,6 @@ public class ChatGPTService {
content = service.createCompletion(completionRequest).getChoices().get(0).getText();
return content;
}

6
src/test/java/com/northtecom/visatrack/api/OpenAiTest.java
View File

@ -43,12 +43,12 @@ class OpenAiTest {
}
// @Test
@Test
public void testToWeChatMessage() {
OpenAiService service = new OpenAiService("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");
OpenAiService service = new OpenAiService("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", 30);
CompletionRequest completionRequest = new CompletionRequestBuilder()
.setPrompt("Python")
.setPrompt("雷达与热成像等其他技术监测火灾效果对比")
.setModel("text-davinci-003")
.setMaxTokens(1000)
.setEcho(true)