1
1
import * as dotenv from "dotenv" ;
2
2
3
- < < < << << HEAD
4
- import { CompletionOptions } from ".." ;
5
- import { BaseLLM } from "../llm" ;
6
-
7
- jest . setTimeout ( 100_000 ) ;
8
- === === =
9
3
import { CompletionOptions } from "../index.js" ;
10
4
import { BaseLLM } from "../llm/index.js" ;
11
5
import OpenAI from "../llm/llms/OpenAI.js" ;
12
6
13
7
// jest.setTimeout(100_000);
14
- >>> >>> > v0 .9 .184 - vscode
15
8
16
9
dotenv . config ( ) ;
17
10
@@ -40,10 +33,6 @@ function testLLM(llm: BaseLLM) {
40
33
}
41
34
42
35
expect ( total . length ) . toBeGreaterThan ( 0 ) ;
43
- < < < << << HEAD
44
- console . log ( total ) ;
45
- === === =
46
- >>> >>> > v0 .9 .184 - vscode
47
36
return ;
48
37
} ) ;
49
38
@@ -54,21 +43,13 @@ function testLLM(llm: BaseLLM) {
54
43
}
55
44
56
45
expect ( total . length ) . toBeGreaterThan ( 0 ) ;
57
- < < < << << HEAD
58
- console . log ( total ) ;
59
- === === =
60
- >>> >>> > v0 .9 .184 - vscode
61
46
return ;
62
47
} ) ;
63
48
64
49
test ( "Complete works" , async ( ) => {
65
50
const completion = await llm . complete ( "Hi" ) ;
66
51
67
52
expect ( completion . length ) . toBeGreaterThan ( 0 ) ;
68
- < < < << << HEAD
69
- console . log ( completion ) ;
70
- === === =
71
- >>> >>> > v0 .9 .184 - vscode
72
53
return ;
73
54
} ) ;
74
55
} ) ;
@@ -78,29 +59,17 @@ describe("LLM", () => {
78
59
// testLLM(
79
60
// new FreeTrial({
80
61
// model: "gpt-3.5-turbo",
81
- < < < << << HEAD
82
- // })
83
- = === ===
84
62
// }),
85
- >>> >>> > v0 .9 .184 - vscode
86
63
// );
87
64
// testLLM(
88
65
// new Anthropic({
89
66
// model: "claude-2",
90
67
// apiKey: process.env.ANTHROPIC_API_KEY,
91
- < < < << << HEAD
92
- // })
93
- // );
94
- // testLLM(
95
- // new OpenAI({ apiKey: process.env.OPENAI_API_KEY, model: "gpt-3.5-turbo" })
96
- // );
97
- = === ===
98
68
// }),
99
69
// );
100
70
testLLM (
101
71
new OpenAI ( { apiKey : process . env . OPENAI_API_KEY , model : "gpt-3.5-turbo" } ) ,
102
72
) ;
103
- >>> >>> > v0 .9 .184 - vscode
104
73
// TODO: Fix Replicate
105
74
// testLLM(
106
75
// new Replicate({
@@ -123,11 +92,7 @@ describe("LLM", () => {
123
92
// testLLM(new HuggingFaceTGI({ model: "codellama-7b" }));
124
93
// testLLM(new HuggingFaceInferenceAPI({ model: "codellama-7b" }));
125
94
// testLLM(
126
- < < < << << HEAD
127
- // new GooglePalm({
128
- = === ===
129
95
// new Gemini({
130
- >>> >>> > v0 .9 .184 - vscode
131
96
// model: "gemini-pro",
132
97
// // model: "chat-bison-001",
133
98
// apiKey: process.env.GOOGLE_PALM_API_KEY,
0 commit comments