From 0d50b22dbfbf2e1b16c83124845a1416bdfeb234 Mon Sep 17 00:00:00 2001
From: Yi Ding <yi.s.ding@gmail.com>
Date: Sun, 28 Apr 2024 18:43:14 -0700
Subject: [PATCH] fix(core): add more exports on `llm/index` (#780)
---
packages/core/e2e/fixtures/llm/anthropic.ts | 3 +++
packages/core/src/llm/index.ts | 11 +++++++++--
2 files changed, 12 insertions(+), 2 deletions(-)
diff --git a/packages/core/e2e/fixtures/llm/anthropic.ts b/packages/core/e2e/fixtures/llm/anthropic.ts
index e5490befe..95cfd24fd 100644
--- a/packages/core/e2e/fixtures/llm/anthropic.ts
+++ b/packages/core/e2e/fixtures/llm/anthropic.ts
@@ -1,3 +1,6 @@
import { OpenAI } from "./openai.js";
export class Anthropic extends OpenAI {}
+export const ALL_AVAILABLE_ANTHROPIC_LEGACY_MODELS = {};
+export const ALL_AVAILABLE_ANTHROPIC_MODELS = {};
+export const ALL_AVAILABLE_V3_MODELS = {};
diff --git a/packages/core/src/llm/index.ts b/packages/core/src/llm/index.ts
index 3521a182a..9c8f33b71 100644
--- a/packages/core/src/llm/index.ts
+++ b/packages/core/src/llm/index.ts
@@ -1,5 +1,11 @@
-export { Anthropic } from "./anthropic.js";
+export {
+ ALL_AVAILABLE_ANTHROPIC_LEGACY_MODELS,
+ ALL_AVAILABLE_ANTHROPIC_MODELS,
+ ALL_AVAILABLE_V3_MODELS,
+ Anthropic,
+} from "./anthropic.js";
export { FireworksLLM } from "./fireworks.js";
+export { GEMINI_MODEL, Gemini } from "./gemini.js";
export { Groq } from "./groq.js";
export {
ALL_AVAILABLE_MISTRAL_MODELS,
@@ -10,12 +16,13 @@ export * from "./openai.js";
export { Portkey } from "./portkey.js";
export * from "./replicate_ai.js";
// Note: The type aliases for replicate are to simplify usage for Llama 2 (we're using replicate for Llama 2 support)
-export { GEMINI_MODEL, Gemini } from "./gemini.js";
export {
+ ALL_AVAILABLE_REPLICATE_MODELS,
DeuceChatStrategy,
LlamaDeuce,
ReplicateChatStrategy,
ReplicateLLM,
+ ReplicateSession,
} from "./replicate_ai.js";
export { TogetherLLM } from "./together.js";
export * from "./types.js";
--
GitLab