This repository was archived by the owner on May 11, 2024. It is now read-only.

Commit 3be05bb

committed

修改单元测试时无法加载apikey和prompt的问题

1 parent ea73207 commit 3be05bbCopy full SHA for 3be05bb

File tree

+11

-1

lines changed

+11

-1

lines changed

Lines changed: 7 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -56,21 +56,27 @@ public void chatOneShot(){`
`56`	`56`
`57`	`57`	`@Test`
`58`	`58`	`public void chatSession(){`
	`59`	`+ adminApiKeyService.load();`
`59`	`60`	`ChatGPTReq chatGPTReq = ChatGPTReq.builder().build();`
`60`	`61`	`ChatGPTResp resp = chatService.sessionChat(`
`61`	`62`	`"", 1, chatGPTReq,"我刚刚问了什么",`
`62`	`63`	`adminApiKeyService.roundRobinGetByType(ApiType.OPENAI), SessionType.NORMAL_CHAT);`
`63`	`64`	`System.out.println(resp.getMessage());`
`64`	`65`	`}`
`65`	`66`
	`67`	`+ /**`
	`68`	`+ * 改方法已废弃`
	`69`	`+ */`
`66`	`70`	`@Test`
`67`	`71`	`public void testCreditGrants(){`
	`72`	`+ adminApiKeyService.load();`
`68`	`73`	`CreditGrantsResp openAiCreditGrantsResp = chatService.creditGrants(adminApiKeyService.roundRobinGetByType(ApiType.OPENAI));`
`69`	`74`	`System.out.println(openAiCreditGrantsResp);`
`70`	`75`	`}`
`71`	`76`
`72`	`77`	`@Test`
`73`	`78`	`public void testStreamSessionReq(){`
	`79`	`+ adminApiKeyService.load();`
`74`	`80`	`String msg = "你好";`
`75`	`81`	`ChatGPTReq chatGPTReq = ChatGPTReq.builder()`
`76`	`82`	`.messages(ImmutableList.of(new ContextMessage(Role.USER.name, msg)))`
`@@ -93,6 +99,7 @@ public void testStreamSessionReq(){`
`93`	`99`
`94`	`100`	`@Test`
`95`	`101`	`public void testStreamOneShotReq(){`
	`102`	`+ adminApiKeyService.load();`
`96`	`103`	`String msg = "你好呀";`
`97`	`104`	`ChatGPTReq chatGPTReq = ChatGPTReq.builder()`
`98`	`105`	`.messages(ImmutableList.of(new ContextMessage(Role.USER.name, msg)))`

Lines changed: 4 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -33,10 +33,13 @@ public class TestFileChatService {`
`33`	`33`	`private FileChatService fileChatService;`
`34`	`34`	`@Resource`
`35`	`35`	`private AdminApiKeyService adminApiKeyService;`
	`36`	`+ @Resource`
	`37`	`+ private PromptService promptService;`
`36`	`38`
`37`	`39`	`@Test`
`38`	`40`	`public void uploadFile(){`
`39`		`-`
	`41`	`+ adminApiKeyService.load();`
	`42`	`+ promptService.load();`
`40`	`43`	`//生成File文件`
`41`	`44`	`File file = new File("D:\\!!笔记\\000学习\\6-专业前沿学习\\003-人工智能\\09-NLP相关论文\\14-chatgpt\\ChatGPT-SpringBoot\\src\\test\\java\\com\\gzhu\\funai\\service\\ICTAI2022_9_28___终稿.pdf");`
`42`	`45`

Comments

(0)