From 333221ff8d7db17587eabf63854817d10d51ebcb Mon Sep 17 00:00:00 2001 From: Calvinn Ng Date: Thu, 25 Jul 2024 11:17:55 +0800 Subject: [PATCH] merge core/test/llm.test.ts --- core/test/llm.test.ts | 35 ----------------------------------- 1 file changed, 35 deletions(-) diff --git a/core/test/llm.test.ts b/core/test/llm.test.ts index bb6ebffde..e3f427731 100644 --- a/core/test/llm.test.ts +++ b/core/test/llm.test.ts @@ -1,17 +1,10 @@ import * as dotenv from "dotenv"; -<<<<<<< HEAD -import { CompletionOptions } from ".."; -import { BaseLLM } from "../llm"; - -jest.setTimeout(100_000); -======= import { CompletionOptions } from "../index.js"; import { BaseLLM } from "../llm/index.js"; import OpenAI from "../llm/llms/OpenAI.js"; // jest.setTimeout(100_000); ->>>>>>> v0.9.184-vscode dotenv.config(); @@ -40,10 +33,6 @@ function testLLM(llm: BaseLLM) { } expect(total.length).toBeGreaterThan(0); -<<<<<<< HEAD - console.log(total); -======= ->>>>>>> v0.9.184-vscode return; }); @@ -54,10 +43,6 @@ function testLLM(llm: BaseLLM) { } expect(total.length).toBeGreaterThan(0); -<<<<<<< HEAD - console.log(total); -======= ->>>>>>> v0.9.184-vscode return; }); @@ -65,10 +50,6 @@ function testLLM(llm: BaseLLM) { const completion = await llm.complete("Hi"); expect(completion.length).toBeGreaterThan(0); -<<<<<<< HEAD - console.log(completion); -======= ->>>>>>> v0.9.184-vscode return; }); }); @@ -78,29 +59,17 @@ describe("LLM", () => { // testLLM( // new FreeTrial({ // model: "gpt-3.5-turbo", -<<<<<<< HEAD - // }) -======= // }), ->>>>>>> v0.9.184-vscode // ); // testLLM( // new Anthropic({ // model: "claude-2", // apiKey: process.env.ANTHROPIC_API_KEY, -<<<<<<< HEAD - // }) - // ); - // testLLM( - // new OpenAI({ apiKey: process.env.OPENAI_API_KEY, model: "gpt-3.5-turbo" }) - // ); -======= // }), // ); testLLM( new OpenAI({ apiKey: process.env.OPENAI_API_KEY, model: "gpt-3.5-turbo" }), ); ->>>>>>> v0.9.184-vscode // TODO: Fix Replicate // testLLM( // new Replicate({ @@ -123,11 +92,7 @@ describe("LLM", () => { // testLLM(new HuggingFaceTGI({ model: "codellama-7b" })); // testLLM(new HuggingFaceInferenceAPI({ model: "codellama-7b" })); // testLLM( -<<<<<<< HEAD - // new GooglePalm({ -======= // new Gemini({ ->>>>>>> v0.9.184-vscode // model: "gemini-pro", // // model: "chat-bison-001", // apiKey: process.env.GOOGLE_PALM_API_KEY,