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 e5490befe9..95cfd24fd6 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 3521a182ab..9c8f33b717 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";