Skip to content

Commit

Permalink
Fix types
Browse files Browse the repository at this point in the history
  • Loading branch information
jacoblee93 committed Aug 1, 2024
1 parent df05fc9 commit 7658119
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion libs/langchain-community/src/llms/layerup_security.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ import {
LLM,
BaseLLM,
type BaseLLMParams,
BaseLLMCallOptions,
} from "@langchain/core/language_models/llms";
import {
GuardrailResponse,
Expand Down Expand Up @@ -101,7 +102,7 @@ export class LayerupSecurity extends LLM {
return "layerup_security";
}

async _call(input: string, options?: BaseLLMParams): Promise<string> {
async _call(input: string, options?: BaseLLMCallOptions): Promise<string> {
// Since LangChain LLMs only support string inputs, we will wrap each call to Layerup in a single-message
// array of messages, then extract the string element when we need to access it.
let messages: LLMMessage[] = [
Expand Down

0 comments on commit 7658119

Please sign in to comment.