nit: LLM -> BaseLLM

langchain-ai · Apr 8, 2024 · cd5c1da · cd5c1da
1 parent 2e0ed62
commit cd5c1da
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/libs/langchain-community/src/llms/layerup_security.ts b/libs/langchain-community/src/llms/layerup_security.ts
@@ -1,12 +1,16 @@
-import { LLM, type BaseLLMParams } from "@langchain/core/language_models/llms";
+import {
+  LLM,
+  BaseLLM,
+  type BaseLLMParams,
+} from "@langchain/core/language_models/llms";
 import {
   GuardrailResponse,
   LayerupSecurity as LayerupSecuritySDK,
   LLMMessage,
 } from "@layerup/layerup-security";
 
 export interface LayerupSecurityOptions extends BaseLLMParams {
-  llm: LLM;
+  llm: BaseLLM;
   layerupApiKey?: string;
   layerupApiBaseUrl?: string;
   promptGuardrails?: string[];