fix: llm teset

LarchLiu · Apr 3, 2024 · eb2df50 · eb2df50
1 parent 7561532
commit eb2df50
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 11 deletions.
diff --git a/packages/core/test/googleai.test.ts b/packages/core/test/googleai.test.ts
@@ -1,6 +1,6 @@
 import { describe, expect, it } from 'vitest'
 
-// import { summarizeContent } from '../src/llm/googleai'
+import { summarizeContent } from '../src/llm/googleai'
 import type { WebInfoData } from '../src/types'
 
 const webInfo: WebInfoData = {
@@ -15,17 +15,15 @@ const webInfo: WebInfoData = {
 }
 describe('googleai', () => {
   it('summarize in zh-CN', async () => {
-    // const res = await summarizeContent({ apiKey: import.meta.env.VITE_GOOGLE_API_KEY, lang: 'zh-CN' }, webInfo)
-    // expect(res.summary).toBeDefined()
-    // expect(res.categories).toBeDefined()
+    const res = await summarizeContent({ apiKey: import.meta.env.VITE_GOOGLE_API_KEY, lang: 'zh-CN' }, webInfo)
+    expect(res.summary).toBeDefined()
+    expect(res.categories).toBeDefined()
     // expect(res)?.toMatchSnapshot()
-    expect(webInfo.title).toBeDefined()
   })
   it('summarize in en', async () => {
-    // const res = await summarizeContent({ apiKey: import.meta.env.VITE_GOOGLE_API_KEY, lang: 'en' }, webInfo)
-    // expect(res.summary).toBeDefined()
-    // expect(res.categories).toBeDefined()
+    const res = await summarizeContent({ apiKey: import.meta.env.VITE_GOOGLE_API_KEY, lang: 'en' }, webInfo)
+    expect(res.summary).toBeDefined()
+    expect(res.categories).toBeDefined()
     // expect(res)?.toMatchSnapshot()
-    expect(webInfo.title).toBeDefined()
   })
 })
diff --git a/packages/core/test/openai.test.ts b/packages/core/test/openai.test.ts
@@ -14,13 +14,13 @@ const webInfo: WebInfoData = {
 }
 describe('openai', () => {
   it('summarize in zh-CN', async () => {
-    const res = await summarizeContent(import.meta.env.VITE_OPENAI_API_KEY, webInfo, 'zh-CN', 'https://aihubmix.com/v1')
+    const res = await summarizeContent(import.meta.env.VITE_OPENAI_API_KEY, webInfo, 'zh-CN', 'https://aihubmix.com')
     expect(res.summary).toBeDefined()
     expect(res.categories).toBeDefined()
     // expect(res)?.toMatchSnapshot()
   })
   it('summarize in en', async () => {
-    const res = await summarizeContent(import.meta.env.VITE_OPENAI_API_KEY, webInfo, 'en', 'https://aihubmix.com/v1')
+    const res = await summarizeContent(import.meta.env.VITE_OPENAI_API_KEY, webInfo, 'en', 'https://aihubmix.com')
     expect(res.summary).toBeDefined()
     expect(res.categories).toBeDefined()
     // expect(res)?.toMatchSnapshot()