Skip to content

Commit 1e48160

Browse files
author
Calvinn Ng
committed
merge core/test/llm.test.ts
1 parent fdc0ed4 commit 1e48160

File tree

1 file changed

+0
-35
lines changed

1 file changed

+0
-35
lines changed

core/test/llm.test.ts

Lines changed: 0 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,10 @@
11
import * as dotenv from "dotenv";
22

3-
<<<<<<< HEAD
4-
import { CompletionOptions } from "..";
5-
import { BaseLLM } from "../llm";
6-
7-
jest.setTimeout(100_000);
8-
=======
93
import { CompletionOptions } from "../index.js";
104
import { BaseLLM } from "../llm/index.js";
115
import OpenAI from "../llm/llms/OpenAI.js";
126

137
// jest.setTimeout(100_000);
14-
>>>>>>> v0.9.184-vscode
158

169
dotenv.config();
1710

@@ -40,10 +33,6 @@ function testLLM(llm: BaseLLM) {
4033
}
4134

4235
expect(total.length).toBeGreaterThan(0);
43-
<<<<<<< HEAD
44-
console.log(total);
45-
=======
46-
>>>>>>> v0.9.184-vscode
4736
return;
4837
});
4938

@@ -54,21 +43,13 @@ function testLLM(llm: BaseLLM) {
5443
}
5544

5645
expect(total.length).toBeGreaterThan(0);
57-
<<<<<<< HEAD
58-
console.log(total);
59-
=======
60-
>>>>>>> v0.9.184-vscode
6146
return;
6247
});
6348

6449
test("Complete works", async () => {
6550
const completion = await llm.complete("Hi");
6651

6752
expect(completion.length).toBeGreaterThan(0);
68-
<<<<<<< HEAD
69-
console.log(completion);
70-
=======
71-
>>>>>>> v0.9.184-vscode
7253
return;
7354
});
7455
});
@@ -78,29 +59,17 @@ describe("LLM", () => {
7859
// testLLM(
7960
// new FreeTrial({
8061
// model: "gpt-3.5-turbo",
81-
<<<<<<< HEAD
82-
// })
83-
=======
8462
// }),
85-
>>>>>>> v0.9.184-vscode
8663
// );
8764
// testLLM(
8865
// new Anthropic({
8966
// model: "claude-2",
9067
// 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-
=======
9868
// }),
9969
// );
10070
testLLM(
10171
new OpenAI({ apiKey: process.env.OPENAI_API_KEY, model: "gpt-3.5-turbo" }),
10272
);
103-
>>>>>>> v0.9.184-vscode
10473
// TODO: Fix Replicate
10574
// testLLM(
10675
// new Replicate({
@@ -123,11 +92,7 @@ describe("LLM", () => {
12392
// testLLM(new HuggingFaceTGI({ model: "codellama-7b" }));
12493
// testLLM(new HuggingFaceInferenceAPI({ model: "codellama-7b" }));
12594
// testLLM(
126-
<<<<<<< HEAD
127-
// new GooglePalm({
128-
=======
12995
// new Gemini({
130-
>>>>>>> v0.9.184-vscode
13196
// model: "gemini-pro",
13297
// // model: "chat-bison-001",
13398
// apiKey: process.env.GOOGLE_PALM_API_KEY,

0 commit comments

Comments
 (0)