langcode/src/plugins/vectorSearch/vectorSearchPlugin.ts

A Plugin-Based Framework for Managing and Using LangChain

import { Document } from "@langchain/core/documents";
import { VectorSearchInitConfig, VectorSearchRunArgs,Plugin, VectorSearchExpose, PluginType } from "../../types";


export default class VectorSearchPlugin
  implements Plugin<VectorSearchInitConfig, VectorSearchRunArgs, VectorSearchExpose, Document[]>
{
  name = "vectorSearch";
  description = "Generic vector search with dynamic retriever input.";
  type=PluginType.VectorSearch;
  RunConfigExample={
    query: "",
    retriever: undefined
  }
  private retriever!:VectorSearchExpose["retriever"];

  InitConfigExample = {}; // init'e config verilmeyecek

  expose(): VectorSearchExpose {
    return {
      name: this.name,
      description: this.description,
      type:this.type,
      InitConfigExample: this.InitConfigExample,
      RunConfigExample:this.RunConfigExample,
      retriever: this.retriever,
    };
  }
  
  async init(_: VectorSearchInitConfig) {
    // hiçbir şey yapılmaz
  }

  async run(args: VectorSearchRunArgs): Promise<Document[]> {
    const { retriever, query } = args;
  //  if (!retriever || !query) {throw new Error("retriever instance is required")};
    this.retriever = retriever
    return await retriever.invoke(query);
  }

 
}