@langchain/classic 1.0.17 → 1.0.18

package/dist/experimental/generative_agents/generative_agent.js.map

@@ -1 +1 @@
-{"version":3,"file":"generative_agent.js","names":["llm: BaseLanguageModelInterface","longTermMemory: GenerativeAgentMemory","config: GenerativeAgentConfig","text: string","lines: string[]","result: string[]","line: string","prompt: PromptTemplate","observation: string","runManager?: CallbackManagerForChainRun","entityName: string","values: ChainValues","chainInputs: ChainValues","text: string | undefined","now?: Date","callToActionTemplate: string","config?: {\n      now?: Date;\n      forceRefresh?: boolean;\n    }","config: {\n      now?: Date;\n      forceRefresh?: boolean;\n    }","memoryContent: string","metadata?: Record<string, unknown>","callbacks?: Callbacks"],"sources":["../../../src/experimental/generative_agents/generative_agent.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport {\n  CallbackManagerForChainRun,\n  Callbacks,\n} from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { GenerativeAgentMemory } from \"./generative_agent_memory.js\";\nimport { BaseChain } from \"../../chains/base.js\";\n\n/**\n * Configuration for the GenerativeAgent class. Defines the character's\n * name, optional age, permanent traits, status, verbosity, and summary\n * refresh seconds.\n */\nexport type GenerativeAgentConfig = {\n  name: string;\n  age?: number;\n  traits: string;\n  status: string;\n  verbose?: boolean;\n  summaryRefreshSeconds?: number;\n  // dailySummaries?: string[];\n};\n\n/**\n * Implementation of a generative agent that can learn and form new memories over\n * time. It extends the BaseChain class, which is a generic\n * sequence of calls to components, including other chains.\n * @example\n * ```typescript\n * const tommie: GenerativeAgent = new GenerativeAgent(\n *   new OpenAI({ temperature: 0.9, maxTokens: 1500 }),\n *   new GenerativeAgentMemory(\n *     new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n *     new TimeWeightedVectorStoreRetriever({\n *       vectorStore: new MemoryVectorStore(new OpenAIEmbeddings()),\n *       otherScoreKeys: [\"importance\"],\n *       k: 15,\n *     }),\n *     { reflectionThreshold: 8 },\n *   ),\n *   {\n *     name: \"Tommie\",\n *     age: 25,\n *     traits: \"anxious, likes design, talkative\",\n *     status: \"looking for a job\",\n *   },\n * );\n *\n * await tommie.addMemory(\n *   \"Tommie remembers his dog, Bruno, from when he was a kid\",\n *   new Date(),\n * );\n * const summary = await tommie.getSummary({ forceRefresh: true });\n * const response = await tommie.generateDialogueResponse(\n *   \"USER says Hello Tommie, how are you today?\",\n * );\n * ```\n */\nexport class GenerativeAgent extends BaseChain {\n  static lc_name() {\n    return \"GenerativeAgent\";\n  }\n\n  // a character with memory and innate characterisitics\n  name: string; // the character's name\n\n  age?: number; // the optional age of the character\n\n  traits: string; // permanent traits to ascribe to the character\n\n  status: string; // the traits of the character you wish not to change\n\n  longTermMemory: GenerativeAgentMemory;\n\n  llm: BaseLanguageModelInterface; // the underlying language model\n\n  verbose: boolean; // false\n\n  private summary: string; // stateful self-summary generated via reflection on the character's memory.\n\n  private summaryRefreshSeconds = 3600;\n\n  private lastRefreshed: Date; // the last time the character's summary was regenerated\n\n  // TODO: Add support for daily summaries\n  // private dailySummaries: string[] = []; // summary of the events in the plan that the agent took.\n\n  _chainType(): string {\n    return \"generative_agent_executor\";\n  }\n\n  get inputKeys(): string[] {\n    return [\"observation\", \"suffix\", \"now\"];\n  }\n\n  get outputKeys(): string[] {\n    return [\"output\", \"continue_dialogue\"];\n  }\n\n  constructor(\n    llm: BaseLanguageModelInterface,\n    longTermMemory: GenerativeAgentMemory,\n    config: GenerativeAgentConfig\n  ) {\n    super();\n    this.llm = llm;\n    this.longTermMemory = longTermMemory;\n    this.name = config.name;\n    this.age = config.age;\n    this.traits = config.traits;\n    this.status = config.status;\n    this.verbose = config.verbose ?? this.verbose;\n    this.summary = \"\";\n    this.summaryRefreshSeconds =\n      config.summaryRefreshSeconds ?? this.summaryRefreshSeconds;\n    this.lastRefreshed = new Date();\n    // this.dailySummaries = config.dailySummaries ?? this.dailySummaries;\n  }\n\n  // LLM methods\n  /**\n   * Parses a newline-separated string into a list of strings.\n   * @param text The string to parse.\n   * @returns An array of strings parsed from the input text.\n   */\n  parseList(text: string): string[] {\n    // parse a newline-seperated string into a list of strings\n    const lines: string[] = text.trim().split(\"\\n\");\n    const result: string[] = lines.map((line: string) =>\n      line.replace(/^\\s*\\d+\\.\\s*/, \"\").trim()\n    );\n    return result;\n  }\n\n  /**\n   * Creates a new LLMChain with the given prompt and the agent's language\n   * model, verbosity, output key, and memory.\n   * @param prompt The prompt to use for the LLMChain.\n   * @returns A new LLMChain instance.\n   */\n  chain(prompt: PromptTemplate): LLMChain {\n    const chain = new LLMChain({\n      llm: this.llm,\n      prompt,\n      verbose: this.verbose,\n      outputKey: \"output\", // new\n      memory: this.longTermMemory,\n    });\n    return chain;\n  }\n\n  /**\n   * Extracts the observed entity from the given observation.\n   * @param observation The observation to extract the entity from.\n   * @param runManager Optional CallbackManagerForChainRun instance.\n   * @returns The extracted entity as a string.\n   */\n  async getEntityFromObservations(\n    observation: string,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string> {\n    const prompt = PromptTemplate.fromTemplate(\n      \"What is the observed entity in the following observation? {observation}\" +\n        \"\\nEntity=\"\n    );\n\n    const result = await this.chain(prompt).call(\n      {\n        observation,\n      },\n      runManager?.getChild(\"entity_extractor\")\n    );\n\n    return result.output;\n  }\n\n  /**\n   * Extracts the action of the given entity from the given observation.\n   * @param observation The observation to extract the action from.\n   * @param entityName The name of the entity to extract the action for.\n   * @param runManager Optional CallbackManagerForChainRun instance.\n   * @returns The extracted action as a string.\n   */\n  async getEntityAction(\n    observation: string,\n    entityName: string,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string> {\n    const prompt = PromptTemplate.fromTemplate(\n      \"What is the {entity} doing in the following observation? {observation}\" +\n        \"\\nThe {entity} is\"\n    );\n\n    const result = await this.chain(prompt).call(\n      {\n        entity: entityName,\n        observation,\n      },\n      runManager?.getChild(\"entity_action_extractor\")\n    );\n    const trimmedResult = result.output.trim();\n    return trimmedResult;\n  }\n\n  /**\n   * Summarizes memories that are most relevant to an observation.\n   * @param observation The observation to summarize related memories for.\n   * @param runManager Optional CallbackManagerForChainRun instance.\n   * @returns The summarized memories as a string.\n   */\n  async summarizeRelatedMemories(\n    observation: string,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string> {\n    // summarize memories that are most relevant to an observation\n    const prompt = PromptTemplate.fromTemplate(\n      `\n{q1}?\nContext from memory:\n{relevant_memories}\nRelevant context:`\n    );\n    const entityName = await this.getEntityFromObservations(\n      observation,\n      runManager\n    );\n    const entityAction = await this.getEntityAction(\n      observation,\n      entityName,\n      runManager\n    );\n    const q1 = `What is the relationship between ${this.name} and ${entityName}`;\n    const q2 = `${entityName} is ${entityAction}`;\n    const response = await this.chain(prompt).call(\n      {\n        q1,\n        queries: [q1, q2],\n      },\n      runManager?.getChild(\"entity_relationships\")\n    );\n\n    return response.output.trim(); // added output\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const { observation, suffix, now } = values;\n    // react to a given observation or dialogue act\n    const prompt = PromptTemplate.fromTemplate(\n      `{agent_summary_description}` +\n        `\\nIt is {current_time}.` +\n        `\\n{agent_name}'s status: {agent_status}` +\n        `\\nSummary of relevant context from {agent_name}'s memory:` +\n        \"\\n{relevant_memories}\" +\n        `\\nMost recent observations: {most_recent_memories}` +\n        `\\nObservation: {observation}` +\n        `\\n\\n${suffix}`\n    );\n\n    const agentSummaryDescription = await this.getSummary({}, runManager); // now = now in param\n    const relevantMemoriesStr = await this.summarizeRelatedMemories(\n      observation,\n      runManager\n    );\n    const currentTime = (now || new Date()).toLocaleString(\"en-US\", {\n      month: \"long\",\n      day: \"numeric\",\n      year: \"numeric\",\n      hour: \"numeric\",\n      minute: \"numeric\",\n      hour12: true,\n    });\n    const chainInputs: ChainValues = {\n      agent_summary_description: agentSummaryDescription,\n      current_time: currentTime,\n      agent_name: this.name,\n      observation,\n      agent_status: this.status,\n      most_recent_memories: \"\",\n    };\n\n    chainInputs[this.longTermMemory.getRelevantMemoriesKey()] =\n      relevantMemoriesStr;\n\n    const consumedTokens = await this.llm.getNumTokens(\n      await prompt.format({ ...chainInputs })\n    );\n\n    chainInputs[this.longTermMemory.getMostRecentMemoriesTokenKey()] =\n      consumedTokens;\n    const response = await this.chain(prompt).call(\n      chainInputs,\n      runManager?.getChild(\"reaction_from_summary\")\n    );\n\n    const rawOutput = response.output;\n    let output = rawOutput;\n    let continue_dialogue = false;\n\n    if (rawOutput.includes(\"REACT:\")) {\n      const reaction = this._cleanResponse(rawOutput.split(\"REACT:\").pop());\n      await this.addMemory(\n        `${this.name} observed ${observation} and reacted by ${reaction}`,\n        now,\n        {},\n        runManager?.getChild(\"memory\")\n      );\n      output = `${reaction}`;\n      continue_dialogue = false;\n    } else if (rawOutput.includes(\"SAY:\")) {\n      const saidValue = this._cleanResponse(rawOutput.split(\"SAY:\").pop());\n      await this.addMemory(\n        `${this.name} observed ${observation} and said ${saidValue}`,\n        now,\n        {},\n        runManager?.getChild(\"memory\")\n      );\n      output = `${this.name} said ${saidValue}`;\n      continue_dialogue = true;\n    } else if (rawOutput.includes(\"GOODBYE:\")) {\n      const farewell = this._cleanResponse(\n        rawOutput.split(\"GOODBYE:\").pop() ?? \"\"\n      );\n      await this.addMemory(\n        `${this.name} observed ${observation} and said ${farewell}`,\n        now,\n        {},\n        runManager?.getChild(\"memory\")\n      );\n      output = `${this.name} said ${farewell}`;\n      continue_dialogue = false;\n    }\n\n    return { output, continue_dialogue };\n  }\n\n  private _cleanResponse(text: string | undefined): string {\n    if (text === undefined) {\n      return \"\";\n    }\n    const regex = new RegExp(`^${this.name} `);\n    return text.replace(regex, \"\").trim();\n  }\n\n  /**\n   * Generates a reaction to the given observation.\n   * @param observation The observation to generate a reaction for.\n   * @param now Optional current date.\n   * @returns A boolean indicating whether to continue the dialogue and the output string.\n   */\n  async generateReaction(\n    observation: string,\n    now?: Date\n  ): Promise<[boolean, string]> {\n    const callToActionTemplate: string =\n      `Should {agent_name} react to the observation, and if so,` +\n      ` what would be an appropriate reaction? Respond in one line.` +\n      ` If the action is to engage in dialogue, write:\\nSAY: \"what to say\"` +\n      ` \\notherwise, write:\\nREACT: {agent_name}'s reaction (if anything).` +\n      ` \\nEither do nothing, react, or say something but not both.\\n\\n`;\n\n    const { output, continue_dialogue } = await this.call({\n      observation,\n      suffix: callToActionTemplate,\n      now,\n    });\n    return [continue_dialogue, output];\n  }\n\n  /**\n   * Generates a dialogue response to the given observation.\n   * @param observation The observation to generate a dialogue response for.\n   * @param now Optional current date.\n   * @returns A boolean indicating whether to continue the dialogue and the output string.\n   */\n  async generateDialogueResponse(\n    observation: string,\n    now?: Date\n  ): Promise<[boolean, string]> {\n    const callToActionTemplate = `What would ${this.name} say? To end the conversation, write: GOODBYE: \"what to say\". Otherwise to continue the conversation, write: SAY: \"what to say next\"\\n\\n`;\n    const { output, continue_dialogue } = await this.call({\n      observation,\n      suffix: callToActionTemplate,\n      now,\n    });\n    return [continue_dialogue, output];\n  }\n\n  // Agent stateful' summary methods\n  // Each dialog or response prompt includes a header\n  // summarizing the agent's self-description. This is\n  // updated periodically through probing it's memories\n  /**\n   * Gets the agent's summary, which includes the agent's name, age, traits,\n   * and a summary of the agent's core characteristics. The summary is\n   * updated periodically through probing the agent's memories.\n   * @param config Optional configuration object with current date and a boolean to force refresh.\n   * @param runManager Optional CallbackManagerForChainRun instance.\n   * @returns The agent's summary as a string.\n   */\n  async getSummary(\n    config?: {\n      now?: Date;\n      forceRefresh?: boolean;\n    },\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string> {\n    const { now = new Date(), forceRefresh = false } = config ?? {};\n\n    const sinceRefresh = Math.floor(\n      (now.getTime() - this.lastRefreshed.getTime()) / 1000\n    );\n\n    if (\n      !this.summary ||\n      sinceRefresh >= this.summaryRefreshSeconds ||\n      forceRefresh\n    ) {\n      this.summary = await this.computeAgentSummary(runManager);\n      this.lastRefreshed = now;\n    }\n\n    let age;\n    if (this.age) {\n      age = this.age;\n    } else {\n      age = \"N/A\";\n    }\n\n    return `Name: ${this.name} (age: ${age})\nInnate traits: ${this.traits}\n${this.summary}`;\n  }\n\n  /**\n   * Computes the agent's summary by summarizing the agent's core\n   * characteristics given the agent's relevant memories.\n   * @param runManager Optional CallbackManagerForChainRun instance.\n   * @returns The computed summary as a string.\n   */\n  async computeAgentSummary(\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string> {\n    const prompt = PromptTemplate.fromTemplate(\n      \"How would you summarize {name}'s core characteristics given the following statements:\\n\" +\n        \"----------\" +\n        \"{relevant_memories}\" +\n        \"----------\" +\n        \"Do not embellish.\" +\n        \"\\n\\nSummary: \"\n    );\n    // the agent seeks to think about their core characterisitics\n    const result = await this.chain(prompt).call(\n      {\n        name: this.name,\n        queries: [`${this.name}'s core characteristics`],\n      },\n      runManager?.getChild(\"compute_agent_summary\")\n    );\n    return result.output.trim();\n  }\n\n  /**\n   * Returns a full header of the agent's status, summary, and current time.\n   * @param config Optional configuration object with current date and a boolean to force refresh.\n   * @returns The full header as a string.\n   */\n  getFullHeader(\n    config: {\n      now?: Date;\n      forceRefresh?: boolean;\n    } = {}\n  ): string {\n    const { now = new Date(), forceRefresh = false } = config;\n    // return a full header of the agent's status, summary, and current time.\n    const summary = this.getSummary({ now, forceRefresh });\n    const currentTimeString = now.toLocaleString(\"en-US\", {\n      month: \"long\",\n      day: \"numeric\",\n      year: \"numeric\",\n      hour: \"numeric\",\n      minute: \"numeric\",\n      hour12: true,\n    });\n    return `${summary}\\nIt is ${currentTimeString}.\\n${this.name}'s status: ${this.status}`;\n  }\n\n  /**\n   * Adds a memory to the agent's long-term memory.\n   * @param memoryContent The content of the memory to add.\n   * @param now Optional current date.\n   * @param metadata Optional metadata for the memory.\n   * @param callbacks Optional Callbacks instance.\n   * @returns The result of adding the memory to the agent's long-term memory.\n   */\n  async addMemory(\n    memoryContent: string,\n    now?: Date,\n    metadata?: Record<string, unknown>,\n    callbacks?: Callbacks\n  ) {\n    return this.longTermMemory.addMemory(\n      memoryContent,\n      now,\n      metadata,\n      callbacks\n    );\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA6DA,IAAa,kBAAb,cAAqC,UAAU;CAC7C,OAAO,UAAU;AACf,SAAO;CACR;CAGD;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA,AAAQ;CAER,AAAQ,wBAAwB;CAEhC,AAAQ;CAKR,aAAqB;AACnB,SAAO;CACR;CAED,IAAI,YAAsB;AACxB,SAAO;GAAC;GAAe;GAAU;EAAM;CACxC;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,UAAU,mBAAoB;CACvC;CAED,YACEA,KACAC,gBACAC,QACA;EACA,OAAO;EACP,KAAK,MAAM;EACX,KAAK,iBAAiB;EACtB,KAAK,OAAO,OAAO;EACnB,KAAK,MAAM,OAAO;EAClB,KAAK,SAAS,OAAO;EACrB,KAAK,SAAS,OAAO;EACrB,KAAK,UAAU,OAAO,WAAW,KAAK;EACtC,KAAK,UAAU;EACf,KAAK,wBACH,OAAO,yBAAyB,KAAK;EACvC,KAAK,gCAAgB,IAAI;CAE1B;;;;;;CAQD,UAAUC,MAAwB;EAEhC,MAAMC,QAAkB,KAAK,MAAM,CAAC,MAAM,KAAK;EAC/C,MAAMC,SAAmB,MAAM,IAAI,CAACC,SAClC,KAAK,QAAQ,gBAAgB,GAAG,CAAC,MAAM,CACxC;AACD,SAAO;CACR;;;;;;;CAQD,MAAMC,QAAkC;EACtC,MAAM,QAAQ,IAAI,SAAS;GACzB,KAAK,KAAK;GACV;GACA,SAAS,KAAK;GACd,WAAW;GACX,QAAQ,KAAK;EACd;AACD,SAAO;CACR;;;;;;;CAQD,MAAM,0BACJC,aACAC,YACiB;EACjB,MAAM,SAAS,eAAe,aAC5B,mFAED;EAED,MAAM,SAAS,MAAM,KAAK,MAAM,OAAO,CAAC,KACtC,EACE,YACD,GACD,YAAY,SAAS,mBAAmB,CACzC;AAED,SAAO,OAAO;CACf;;;;;;;;CASD,MAAM,gBACJD,aACAE,YACAD,YACiB;EACjB,MAAM,SAAS,eAAe,aAC5B,0FAED;EAED,MAAM,SAAS,MAAM,KAAK,MAAM,OAAO,CAAC,KACtC;GACE,QAAQ;GACR;EACD,GACD,YAAY,SAAS,0BAA0B,CAChD;EACD,MAAM,gBAAgB,OAAO,OAAO,MAAM;AAC1C,SAAO;CACR;;;;;;;CAQD,MAAM,yBACJD,aACAC,YACiB;EAEjB,MAAM,SAAS,eAAe,aAC5B,CAAC;;;;iBAIU,CAAC,CACb;EACD,MAAM,aAAa,MAAM,KAAK,0BAC5B,aACA,WACD;EACD,MAAM,eAAe,MAAM,KAAK,gBAC9B,aACA,YACA,WACD;EACD,MAAM,KAAK,CAAC,iCAAiC,EAAE,KAAK,KAAK,KAAK,EAAE,YAAY;EAC5E,MAAM,KAAK,GAAG,WAAW,IAAI,EAAE,cAAc;EAC7C,MAAM,WAAW,MAAM,KAAK,MAAM,OAAO,CAAC,KACxC;GACE;GACA,SAAS,CAAC,IAAI,EAAG;EAClB,GACD,YAAY,SAAS,uBAAuB,CAC7C;AAED,SAAO,SAAS,OAAO,MAAM;CAC9B;CAED,MAAM,MACJE,QACAF,YACsB;EACtB,MAAM,EAAE,aAAa,QAAQ,KAAK,GAAG;EAErC,MAAM,SAAS,eAAe,aAC5B,CAOG;;;;;;8BAAI,EAAE,QAAQ,CAClB;EAED,MAAM,0BAA0B,MAAM,KAAK,WAAW,CAAE,GAAE,WAAW;EACrE,MAAM,sBAAsB,MAAM,KAAK,yBACrC,aACA,WACD;EACD,MAAM,eAAe,uBAAO,IAAI,QAAQ,eAAe,SAAS;GAC9D,OAAO;GACP,KAAK;GACL,MAAM;GACN,MAAM;GACN,QAAQ;GACR,QAAQ;EACT,EAAC;EACF,MAAMG,cAA2B;GAC/B,2BAA2B;GAC3B,cAAc;GACd,YAAY,KAAK;GACjB;GACA,cAAc,KAAK;GACnB,sBAAsB;EACvB;EAED,YAAY,KAAK,eAAe,wBAAwB,IACtD;EAEF,MAAM,iBAAiB,MAAM,KAAK,IAAI,aACpC,MAAM,OAAO,OAAO,EAAE,GAAG,YAAa,EAAC,CACxC;EAED,YAAY,KAAK,eAAe,+BAA+B,IAC7D;EACF,MAAM,WAAW,MAAM,KAAK,MAAM,OAAO,CAAC,KACxC,aACA,YAAY,SAAS,wBAAwB,CAC9C;EAED,MAAM,YAAY,SAAS;EAC3B,IAAI,SAAS;EACb,IAAI,oBAAoB;AAExB,MAAI,UAAU,SAAS,SAAS,EAAE;GAChC,MAAM,WAAW,KAAK,eAAe,UAAU,MAAM,SAAS,CAAC,KAAK,CAAC;GACrE,MAAM,KAAK,UACT,GAAG,KAAK,KAAK,UAAU,EAAE,YAAY,gBAAgB,EAAE,UAAU,EACjE,KACA,CAAE,GACF,YAAY,SAAS,SAAS,CAC/B;GACD,SAAS,GAAG,UAAU;GACtB,oBAAoB;EACrB,WAAU,UAAU,SAAS,OAAO,EAAE;GACrC,MAAM,YAAY,KAAK,eAAe,UAAU,MAAM,OAAO,CAAC,KAAK,CAAC;GACpE,MAAM,KAAK,UACT,GAAG,KAAK,KAAK,UAAU,EAAE,YAAY,UAAU,EAAE,WAAW,EAC5D,KACA,CAAE,GACF,YAAY,SAAS,SAAS,CAC/B;GACD,SAAS,GAAG,KAAK,KAAK,MAAM,EAAE,WAAW;GACzC,oBAAoB;EACrB,WAAU,UAAU,SAAS,WAAW,EAAE;GACzC,MAAM,WAAW,KAAK,eACpB,UAAU,MAAM,WAAW,CAAC,KAAK,IAAI,GACtC;GACD,MAAM,KAAK,UACT,GAAG,KAAK,KAAK,UAAU,EAAE,YAAY,UAAU,EAAE,UAAU,EAC3D,KACA,CAAE,GACF,YAAY,SAAS,SAAS,CAC/B;GACD,SAAS,GAAG,KAAK,KAAK,MAAM,EAAE,UAAU;GACxC,oBAAoB;EACrB;AAED,SAAO;GAAE;GAAQ;EAAmB;CACrC;CAED,AAAQ,eAAeC,MAAkC;AACvD,MAAI,SAAS,OACX,QAAO;EAET,MAAM,wBAAQ,IAAI,OAAO,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,CAAC;AACzC,SAAO,KAAK,QAAQ,OAAO,GAAG,CAAC,MAAM;CACtC;;;;;;;CAQD,MAAM,iBACJL,aACAM,KAC4B;EAC5B,MAAMC,uBACJ;EAMF,MAAM,EAAE,QAAQ,mBAAmB,GAAG,MAAM,KAAK,KAAK;GACpD;GACA,QAAQ;GACR;EACD,EAAC;AACF,SAAO,CAAC,mBAAmB,MAAO;CACnC;;;;;;;CAQD,MAAM,yBACJP,aACAM,KAC4B;EAC5B,MAAM,uBAAuB,CAAC,WAAW,EAAE,KAAK,KAAK,wIAAwI,CAAC;EAC9L,MAAM,EAAE,QAAQ,mBAAmB,GAAG,MAAM,KAAK,KAAK;GACpD;GACA,QAAQ;GACR;EACD,EAAC;AACF,SAAO,CAAC,mBAAmB,MAAO;CACnC;;;;;;;;;CAcD,MAAM,WACJE,QAIAP,YACiB;EACjB,MAAM,EAAE,sBAAM,IAAI,QAAQ,eAAe,OAAO,GAAG,UAAU,CAAE;EAE/D,MAAM,eAAe,KAAK,OACvB,IAAI,SAAS,GAAG,KAAK,cAAc,SAAS,IAAI,IAClD;AAED,MACE,CAAC,KAAK,WACN,gBAAgB,KAAK,yBACrB,cACA;GACA,KAAK,UAAU,MAAM,KAAK,oBAAoB,WAAW;GACzD,KAAK,gBAAgB;EACtB;EAED,IAAI;AACJ,MAAI,KAAK,KACP,MAAM,KAAK;OAEX,MAAM;AAGR,SAAO,CAAC,MAAM,EAAE,KAAK,KAAK,OAAO,EAAE,IAAI;eAC5B,EAAE,KAAK,OAAO;AAC7B,EAAE,KAAK,SAAS;CACb;;;;;;;CAQD,MAAM,oBACJA,YACiB;EACjB,MAAM,SAAS,eAAe,aAC5B,+JAMD;EAED,MAAM,SAAS,MAAM,KAAK,MAAM,OAAO,CAAC,KACtC;GACE,MAAM,KAAK;GACX,SAAS,CAAC,GAAG,KAAK,KAAK,uBAAuB,CAAC,AAAC;EACjD,GACD,YAAY,SAAS,wBAAwB,CAC9C;AACD,SAAO,OAAO,OAAO,MAAM;CAC5B;;;;;;CAOD,cACEQ,SAGI,CAAE,GACE;EACR,MAAM,EAAE,sBAAM,IAAI,QAAQ,eAAe,OAAO,GAAG;EAEnD,MAAM,UAAU,KAAK,WAAW;GAAE;GAAK;EAAc,EAAC;EACtD,MAAM,oBAAoB,IAAI,eAAe,SAAS;GACpD,OAAO;GACP,KAAK;GACL,MAAM;GACN,MAAM;GACN,QAAQ;GACR,QAAQ;EACT,EAAC;AACF,SAAO,GAAG,QAAQ,QAAQ,EAAE,kBAAkB,GAAG,EAAE,KAAK,KAAK,WAAW,EAAE,KAAK,QAAQ;CACxF;;;;;;;;;CAUD,MAAM,UACJC,eACAJ,KACAK,UACAC,WACA;AACA,SAAO,KAAK,eAAe,UACzB,eACA,KACA,UACA,UACD;CACF;AACF"}
+{"version":3,"file":"generative_agent.js","names":[],"sources":["../../../src/experimental/generative_agents/generative_agent.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport {\n  CallbackManagerForChainRun,\n  Callbacks,\n} from \"@langchain/core/callbacks/manager\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nimport { GenerativeAgentMemory } from \"./generative_agent_memory.js\";\nimport { BaseChain } from \"../../chains/base.js\";\n\n/**\n * Configuration for the GenerativeAgent class. Defines the character's\n * name, optional age, permanent traits, status, verbosity, and summary\n * refresh seconds.\n */\nexport type GenerativeAgentConfig = {\n  name: string;\n  age?: number;\n  traits: string;\n  status: string;\n  verbose?: boolean;\n  summaryRefreshSeconds?: number;\n  // dailySummaries?: string[];\n};\n\n/**\n * Implementation of a generative agent that can learn and form new memories over\n * time. It extends the BaseChain class, which is a generic\n * sequence of calls to components, including other chains.\n * @example\n * ```typescript\n * const tommie: GenerativeAgent = new GenerativeAgent(\n *   new OpenAI({ temperature: 0.9, maxTokens: 1500 }),\n *   new GenerativeAgentMemory(\n *     new ChatOpenAI({ model: \"gpt-4o-mini\" }),\n *     new TimeWeightedVectorStoreRetriever({\n *       vectorStore: new MemoryVectorStore(new OpenAIEmbeddings()),\n *       otherScoreKeys: [\"importance\"],\n *       k: 15,\n *     }),\n *     { reflectionThreshold: 8 },\n *   ),\n *   {\n *     name: \"Tommie\",\n *     age: 25,\n *     traits: \"anxious, likes design, talkative\",\n *     status: \"looking for a job\",\n *   },\n * );\n *\n * await tommie.addMemory(\n *   \"Tommie remembers his dog, Bruno, from when he was a kid\",\n *   new Date(),\n * );\n * const summary = await tommie.getSummary({ forceRefresh: true });\n * const response = await tommie.generateDialogueResponse(\n *   \"USER says Hello Tommie, how are you today?\",\n * );\n * ```\n */\nexport class GenerativeAgent extends BaseChain {\n  static lc_name() {\n    return \"GenerativeAgent\";\n  }\n\n  // a character with memory and innate characterisitics\n  name: string; // the character's name\n\n  age?: number; // the optional age of the character\n\n  traits: string; // permanent traits to ascribe to the character\n\n  status: string; // the traits of the character you wish not to change\n\n  longTermMemory: GenerativeAgentMemory;\n\n  llm: BaseLanguageModelInterface; // the underlying language model\n\n  verbose: boolean; // false\n\n  private summary: string; // stateful self-summary generated via reflection on the character's memory.\n\n  private summaryRefreshSeconds = 3600;\n\n  private lastRefreshed: Date; // the last time the character's summary was regenerated\n\n  // TODO: Add support for daily summaries\n  // private dailySummaries: string[] = []; // summary of the events in the plan that the agent took.\n\n  _chainType(): string {\n    return \"generative_agent_executor\";\n  }\n\n  get inputKeys(): string[] {\n    return [\"observation\", \"suffix\", \"now\"];\n  }\n\n  get outputKeys(): string[] {\n    return [\"output\", \"continue_dialogue\"];\n  }\n\n  constructor(\n    llm: BaseLanguageModelInterface,\n    longTermMemory: GenerativeAgentMemory,\n    config: GenerativeAgentConfig\n  ) {\n    super();\n    this.llm = llm;\n    this.longTermMemory = longTermMemory;\n    this.name = config.name;\n    this.age = config.age;\n    this.traits = config.traits;\n    this.status = config.status;\n    this.verbose = config.verbose ?? this.verbose;\n    this.summary = \"\";\n    this.summaryRefreshSeconds =\n      config.summaryRefreshSeconds ?? this.summaryRefreshSeconds;\n    this.lastRefreshed = new Date();\n    // this.dailySummaries = config.dailySummaries ?? this.dailySummaries;\n  }\n\n  // LLM methods\n  /**\n   * Parses a newline-separated string into a list of strings.\n   * @param text The string to parse.\n   * @returns An array of strings parsed from the input text.\n   */\n  parseList(text: string): string[] {\n    // parse a newline-seperated string into a list of strings\n    const lines: string[] = text.trim().split(\"\\n\");\n    const result: string[] = lines.map((line: string) =>\n      line.replace(/^\\s*\\d+\\.\\s*/, \"\").trim()\n    );\n    return result;\n  }\n\n  /**\n   * Creates a new LLMChain with the given prompt and the agent's language\n   * model, verbosity, output key, and memory.\n   * @param prompt The prompt to use for the LLMChain.\n   * @returns A new LLMChain instance.\n   */\n  chain(prompt: PromptTemplate): LLMChain {\n    const chain = new LLMChain({\n      llm: this.llm,\n      prompt,\n      verbose: this.verbose,\n      outputKey: \"output\", // new\n      memory: this.longTermMemory,\n    });\n    return chain;\n  }\n\n  /**\n   * Extracts the observed entity from the given observation.\n   * @param observation The observation to extract the entity from.\n   * @param runManager Optional CallbackManagerForChainRun instance.\n   * @returns The extracted entity as a string.\n   */\n  async getEntityFromObservations(\n    observation: string,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string> {\n    const prompt = PromptTemplate.fromTemplate(\n      \"What is the observed entity in the following observation? {observation}\" +\n        \"\\nEntity=\"\n    );\n\n    const result = await this.chain(prompt).call(\n      {\n        observation,\n      },\n      runManager?.getChild(\"entity_extractor\")\n    );\n\n    return result.output;\n  }\n\n  /**\n   * Extracts the action of the given entity from the given observation.\n   * @param observation The observation to extract the action from.\n   * @param entityName The name of the entity to extract the action for.\n   * @param runManager Optional CallbackManagerForChainRun instance.\n   * @returns The extracted action as a string.\n   */\n  async getEntityAction(\n    observation: string,\n    entityName: string,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string> {\n    const prompt = PromptTemplate.fromTemplate(\n      \"What is the {entity} doing in the following observation? {observation}\" +\n        \"\\nThe {entity} is\"\n    );\n\n    const result = await this.chain(prompt).call(\n      {\n        entity: entityName,\n        observation,\n      },\n      runManager?.getChild(\"entity_action_extractor\")\n    );\n    const trimmedResult = result.output.trim();\n    return trimmedResult;\n  }\n\n  /**\n   * Summarizes memories that are most relevant to an observation.\n   * @param observation The observation to summarize related memories for.\n   * @param runManager Optional CallbackManagerForChainRun instance.\n   * @returns The summarized memories as a string.\n   */\n  async summarizeRelatedMemories(\n    observation: string,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string> {\n    // summarize memories that are most relevant to an observation\n    const prompt = PromptTemplate.fromTemplate(\n      `\n{q1}?\nContext from memory:\n{relevant_memories}\nRelevant context:`\n    );\n    const entityName = await this.getEntityFromObservations(\n      observation,\n      runManager\n    );\n    const entityAction = await this.getEntityAction(\n      observation,\n      entityName,\n      runManager\n    );\n    const q1 = `What is the relationship between ${this.name} and ${entityName}`;\n    const q2 = `${entityName} is ${entityAction}`;\n    const response = await this.chain(prompt).call(\n      {\n        q1,\n        queries: [q1, q2],\n      },\n      runManager?.getChild(\"entity_relationships\")\n    );\n\n    return response.output.trim(); // added output\n  }\n\n  async _call(\n    values: ChainValues,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<ChainValues> {\n    const { observation, suffix, now } = values;\n    // react to a given observation or dialogue act\n    const prompt = PromptTemplate.fromTemplate(\n      `{agent_summary_description}` +\n        `\\nIt is {current_time}.` +\n        `\\n{agent_name}'s status: {agent_status}` +\n        `\\nSummary of relevant context from {agent_name}'s memory:` +\n        \"\\n{relevant_memories}\" +\n        `\\nMost recent observations: {most_recent_memories}` +\n        `\\nObservation: {observation}` +\n        `\\n\\n${suffix}`\n    );\n\n    const agentSummaryDescription = await this.getSummary({}, runManager); // now = now in param\n    const relevantMemoriesStr = await this.summarizeRelatedMemories(\n      observation,\n      runManager\n    );\n    const currentTime = (now || new Date()).toLocaleString(\"en-US\", {\n      month: \"long\",\n      day: \"numeric\",\n      year: \"numeric\",\n      hour: \"numeric\",\n      minute: \"numeric\",\n      hour12: true,\n    });\n    const chainInputs: ChainValues = {\n      agent_summary_description: agentSummaryDescription,\n      current_time: currentTime,\n      agent_name: this.name,\n      observation,\n      agent_status: this.status,\n      most_recent_memories: \"\",\n    };\n\n    chainInputs[this.longTermMemory.getRelevantMemoriesKey()] =\n      relevantMemoriesStr;\n\n    const consumedTokens = await this.llm.getNumTokens(\n      await prompt.format({ ...chainInputs })\n    );\n\n    chainInputs[this.longTermMemory.getMostRecentMemoriesTokenKey()] =\n      consumedTokens;\n    const response = await this.chain(prompt).call(\n      chainInputs,\n      runManager?.getChild(\"reaction_from_summary\")\n    );\n\n    const rawOutput = response.output;\n    let output = rawOutput;\n    let continue_dialogue = false;\n\n    if (rawOutput.includes(\"REACT:\")) {\n      const reaction = this._cleanResponse(rawOutput.split(\"REACT:\").pop());\n      await this.addMemory(\n        `${this.name} observed ${observation} and reacted by ${reaction}`,\n        now,\n        {},\n        runManager?.getChild(\"memory\")\n      );\n      output = `${reaction}`;\n      continue_dialogue = false;\n    } else if (rawOutput.includes(\"SAY:\")) {\n      const saidValue = this._cleanResponse(rawOutput.split(\"SAY:\").pop());\n      await this.addMemory(\n        `${this.name} observed ${observation} and said ${saidValue}`,\n        now,\n        {},\n        runManager?.getChild(\"memory\")\n      );\n      output = `${this.name} said ${saidValue}`;\n      continue_dialogue = true;\n    } else if (rawOutput.includes(\"GOODBYE:\")) {\n      const farewell = this._cleanResponse(\n        rawOutput.split(\"GOODBYE:\").pop() ?? \"\"\n      );\n      await this.addMemory(\n        `${this.name} observed ${observation} and said ${farewell}`,\n        now,\n        {},\n        runManager?.getChild(\"memory\")\n      );\n      output = `${this.name} said ${farewell}`;\n      continue_dialogue = false;\n    }\n\n    return { output, continue_dialogue };\n  }\n\n  private _cleanResponse(text: string | undefined): string {\n    if (text === undefined) {\n      return \"\";\n    }\n    const regex = new RegExp(`^${this.name} `);\n    return text.replace(regex, \"\").trim();\n  }\n\n  /**\n   * Generates a reaction to the given observation.\n   * @param observation The observation to generate a reaction for.\n   * @param now Optional current date.\n   * @returns A boolean indicating whether to continue the dialogue and the output string.\n   */\n  async generateReaction(\n    observation: string,\n    now?: Date\n  ): Promise<[boolean, string]> {\n    const callToActionTemplate: string =\n      `Should {agent_name} react to the observation, and if so,` +\n      ` what would be an appropriate reaction? Respond in one line.` +\n      ` If the action is to engage in dialogue, write:\\nSAY: \"what to say\"` +\n      ` \\notherwise, write:\\nREACT: {agent_name}'s reaction (if anything).` +\n      ` \\nEither do nothing, react, or say something but not both.\\n\\n`;\n\n    const { output, continue_dialogue } = await this.call({\n      observation,\n      suffix: callToActionTemplate,\n      now,\n    });\n    return [continue_dialogue, output];\n  }\n\n  /**\n   * Generates a dialogue response to the given observation.\n   * @param observation The observation to generate a dialogue response for.\n   * @param now Optional current date.\n   * @returns A boolean indicating whether to continue the dialogue and the output string.\n   */\n  async generateDialogueResponse(\n    observation: string,\n    now?: Date\n  ): Promise<[boolean, string]> {\n    const callToActionTemplate = `What would ${this.name} say? To end the conversation, write: GOODBYE: \"what to say\". Otherwise to continue the conversation, write: SAY: \"what to say next\"\\n\\n`;\n    const { output, continue_dialogue } = await this.call({\n      observation,\n      suffix: callToActionTemplate,\n      now,\n    });\n    return [continue_dialogue, output];\n  }\n\n  // Agent stateful' summary methods\n  // Each dialog or response prompt includes a header\n  // summarizing the agent's self-description. This is\n  // updated periodically through probing it's memories\n  /**\n   * Gets the agent's summary, which includes the agent's name, age, traits,\n   * and a summary of the agent's core characteristics. The summary is\n   * updated periodically through probing the agent's memories.\n   * @param config Optional configuration object with current date and a boolean to force refresh.\n   * @param runManager Optional CallbackManagerForChainRun instance.\n   * @returns The agent's summary as a string.\n   */\n  async getSummary(\n    config?: {\n      now?: Date;\n      forceRefresh?: boolean;\n    },\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string> {\n    const { now = new Date(), forceRefresh = false } = config ?? {};\n\n    const sinceRefresh = Math.floor(\n      (now.getTime() - this.lastRefreshed.getTime()) / 1000\n    );\n\n    if (\n      !this.summary ||\n      sinceRefresh >= this.summaryRefreshSeconds ||\n      forceRefresh\n    ) {\n      this.summary = await this.computeAgentSummary(runManager);\n      this.lastRefreshed = now;\n    }\n\n    let age;\n    if (this.age) {\n      age = this.age;\n    } else {\n      age = \"N/A\";\n    }\n\n    return `Name: ${this.name} (age: ${age})\nInnate traits: ${this.traits}\n${this.summary}`;\n  }\n\n  /**\n   * Computes the agent's summary by summarizing the agent's core\n   * characteristics given the agent's relevant memories.\n   * @param runManager Optional CallbackManagerForChainRun instance.\n   * @returns The computed summary as a string.\n   */\n  async computeAgentSummary(\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string> {\n    const prompt = PromptTemplate.fromTemplate(\n      \"How would you summarize {name}'s core characteristics given the following statements:\\n\" +\n        \"----------\" +\n        \"{relevant_memories}\" +\n        \"----------\" +\n        \"Do not embellish.\" +\n        \"\\n\\nSummary: \"\n    );\n    // the agent seeks to think about their core characterisitics\n    const result = await this.chain(prompt).call(\n      {\n        name: this.name,\n        queries: [`${this.name}'s core characteristics`],\n      },\n      runManager?.getChild(\"compute_agent_summary\")\n    );\n    return result.output.trim();\n  }\n\n  /**\n   * Returns a full header of the agent's status, summary, and current time.\n   * @param config Optional configuration object with current date and a boolean to force refresh.\n   * @returns The full header as a string.\n   */\n  getFullHeader(\n    config: {\n      now?: Date;\n      forceRefresh?: boolean;\n    } = {}\n  ): string {\n    const { now = new Date(), forceRefresh = false } = config;\n    // return a full header of the agent's status, summary, and current time.\n    const summary = this.getSummary({ now, forceRefresh });\n    const currentTimeString = now.toLocaleString(\"en-US\", {\n      month: \"long\",\n      day: \"numeric\",\n      year: \"numeric\",\n      hour: \"numeric\",\n      minute: \"numeric\",\n      hour12: true,\n    });\n    return `${summary}\\nIt is ${currentTimeString}.\\n${this.name}'s status: ${this.status}`;\n  }\n\n  /**\n   * Adds a memory to the agent's long-term memory.\n   * @param memoryContent The content of the memory to add.\n   * @param now Optional current date.\n   * @param metadata Optional metadata for the memory.\n   * @param callbacks Optional Callbacks instance.\n   * @returns The result of adding the memory to the agent's long-term memory.\n   */\n  async addMemory(\n    memoryContent: string,\n    now?: Date,\n    metadata?: Record<string, unknown>,\n    callbacks?: Callbacks\n  ) {\n    return this.longTermMemory.addMemory(\n      memoryContent,\n      now,\n      metadata,\n      callbacks\n    );\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA6DA,IAAa,kBAAb,cAAqC,UAAU;CAC7C,OAAO,UAAU;AACf,SAAO;;CAIT;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA,AAAQ;CAER,AAAQ,wBAAwB;CAEhC,AAAQ;CAKR,aAAqB;AACnB,SAAO;;CAGT,IAAI,YAAsB;AACxB,SAAO;GAAC;GAAe;GAAU;GAAM;;CAGzC,IAAI,aAAuB;AACzB,SAAO,CAAC,UAAU,oBAAoB;;CAGxC,YACE,KACA,gBACA,QACA;AACA,SAAO;AACP,OAAK,MAAM;AACX,OAAK,iBAAiB;AACtB,OAAK,OAAO,OAAO;AACnB,OAAK,MAAM,OAAO;AAClB,OAAK,SAAS,OAAO;AACrB,OAAK,SAAS,OAAO;AACrB,OAAK,UAAU,OAAO,WAAW,KAAK;AACtC,OAAK,UAAU;AACf,OAAK,wBACH,OAAO,yBAAyB,KAAK;AACvC,OAAK,gCAAgB,IAAI,MAAM;;;;;;;CAUjC,UAAU,MAAwB;AAMhC,SAJwB,KAAK,MAAM,CAAC,MAAM,KAAK,CAChB,KAAK,SAClC,KAAK,QAAQ,gBAAgB,GAAG,CAAC,MAAM,CACxC;;;;;;;;CAUH,MAAM,QAAkC;AAQtC,SAPc,IAAI,SAAS;GACzB,KAAK,KAAK;GACV;GACA,SAAS,KAAK;GACd,WAAW;GACX,QAAQ,KAAK;GACd,CAAC;;;;;;;;CAUJ,MAAM,0BACJ,aACA,YACiB;EACjB,MAAM,SAAS,eAAe,aAC5B,mFAED;AASD,UAPe,MAAM,KAAK,MAAM,OAAO,CAAC,KACtC,EACE,aACD,EACD,YAAY,SAAS,mBAAmB,CACzC,EAEa;;;;;;;;;CAUhB,MAAM,gBACJ,aACA,YACA,YACiB;EACjB,MAAM,SAAS,eAAe,aAC5B,0FAED;AAUD,UARe,MAAM,KAAK,MAAM,OAAO,CAAC,KACtC;GACE,QAAQ;GACR;GACD,EACD,YAAY,SAAS,0BAA0B,CAChD,EAC4B,OAAO,MAAM;;;;;;;;CAU5C,MAAM,yBACJ,aACA,YACiB;EAEjB,MAAM,SAAS,eAAe,aAC5B;;;;mBAKD;EACD,MAAM,aAAa,MAAM,KAAK,0BAC5B,aACA,WACD;EACD,MAAM,eAAe,MAAM,KAAK,gBAC9B,aACA,YACA,WACD;EACD,MAAM,KAAK,oCAAoC,KAAK,KAAK,OAAO;EAChE,MAAM,KAAK,GAAG,WAAW,MAAM;AAS/B,UARiB,MAAM,KAAK,MAAM,OAAO,CAAC,KACxC;GACE;GACA,SAAS,CAAC,IAAI,GAAG;GAClB,EACD,YAAY,SAAS,uBAAuB,CAC7C,EAEe,OAAO,MAAM;;CAG/B,MAAM,MACJ,QACA,YACsB;EACtB,MAAM,EAAE,aAAa,QAAQ,QAAQ;EAErC,MAAM,SAAS,eAAe,aAC5B;;;;;;gCAOS,SACV;EAED,MAAM,0BAA0B,MAAM,KAAK,WAAW,EAAE,EAAE,WAAW;EACrE,MAAM,sBAAsB,MAAM,KAAK,yBACrC,aACA,WACD;EASD,MAAM,cAA2B;GAC/B,2BAA2B;GAC3B,eAVmB,uBAAO,IAAI,MAAM,EAAE,eAAe,SAAS;IAC9D,OAAO;IACP,KAAK;IACL,MAAM;IACN,MAAM;IACN,QAAQ;IACR,QAAQ;IACT,CAAC;GAIA,YAAY,KAAK;GACjB;GACA,cAAc,KAAK;GACnB,sBAAsB;GACvB;AAED,cAAY,KAAK,eAAe,wBAAwB,IACtD;EAEF,MAAM,iBAAiB,MAAM,KAAK,IAAI,aACpC,MAAM,OAAO,OAAO,EAAE,GAAG,aAAa,CAAC,CACxC;AAED,cAAY,KAAK,eAAe,+BAA+B,IAC7D;EAMF,MAAM,aALW,MAAM,KAAK,MAAM,OAAO,CAAC,KACxC,aACA,YAAY,SAAS,wBAAwB,CAC9C,EAE0B;EAC3B,IAAI,SAAS;EACb,IAAI,oBAAoB;AAExB,MAAI,UAAU,SAAS,SAAS,EAAE;GAChC,MAAM,WAAW,KAAK,eAAe,UAAU,MAAM,SAAS,CAAC,KAAK,CAAC;AACrE,SAAM,KAAK,UACT,GAAG,KAAK,KAAK,YAAY,YAAY,kBAAkB,YACvD,KACA,EAAE,EACF,YAAY,SAAS,SAAS,CAC/B;AACD,YAAS,GAAG;AACZ,uBAAoB;aACX,UAAU,SAAS,OAAO,EAAE;GACrC,MAAM,YAAY,KAAK,eAAe,UAAU,MAAM,OAAO,CAAC,KAAK,CAAC;AACpE,SAAM,KAAK,UACT,GAAG,KAAK,KAAK,YAAY,YAAY,YAAY,aACjD,KACA,EAAE,EACF,YAAY,SAAS,SAAS,CAC/B;AACD,YAAS,GAAG,KAAK,KAAK,QAAQ;AAC9B,uBAAoB;aACX,UAAU,SAAS,WAAW,EAAE;GACzC,MAAM,WAAW,KAAK,eACpB,UAAU,MAAM,WAAW,CAAC,KAAK,IAAI,GACtC;AACD,SAAM,KAAK,UACT,GAAG,KAAK,KAAK,YAAY,YAAY,YAAY,YACjD,KACA,EAAE,EACF,YAAY,SAAS,SAAS,CAC/B;AACD,YAAS,GAAG,KAAK,KAAK,QAAQ;AAC9B,uBAAoB;;AAGtB,SAAO;GAAE;GAAQ;GAAmB;;CAGtC,AAAQ,eAAe,MAAkC;AACvD,MAAI,SAAS,OACX,QAAO;EAET,MAAM,QAAQ,IAAI,OAAO,IAAI,KAAK,KAAK,GAAG;AAC1C,SAAO,KAAK,QAAQ,OAAO,GAAG,CAAC,MAAM;;;;;;;;CASvC,MAAM,iBACJ,aACA,KAC4B;EAQ5B,MAAM,EAAE,QAAQ,sBAAsB,MAAM,KAAK,KAAK;GACpD;GACA,QARA;GASA;GACD,CAAC;AACF,SAAO,CAAC,mBAAmB,OAAO;;;;;;;;CASpC,MAAM,yBACJ,aACA,KAC4B;EAC5B,MAAM,uBAAuB,cAAc,KAAK,KAAK;EACrD,MAAM,EAAE,QAAQ,sBAAsB,MAAM,KAAK,KAAK;GACpD;GACA,QAAQ;GACR;GACD,CAAC;AACF,SAAO,CAAC,mBAAmB,OAAO;;;;;;;;;;CAepC,MAAM,WACJ,QAIA,YACiB;EACjB,MAAM,EAAE,sBAAM,IAAI,MAAM,EAAE,eAAe,UAAU,UAAU,EAAE;EAE/D,MAAM,eAAe,KAAK,OACvB,IAAI,SAAS,GAAG,KAAK,cAAc,SAAS,IAAI,IAClD;AAED,MACE,CAAC,KAAK,WACN,gBAAgB,KAAK,yBACrB,cACA;AACA,QAAK,UAAU,MAAM,KAAK,oBAAoB,WAAW;AACzD,QAAK,gBAAgB;;EAGvB,IAAI;AACJ,MAAI,KAAK,IACP,OAAM,KAAK;MAEX,OAAM;AAGR,SAAO,SAAS,KAAK,KAAK,SAAS,IAAI;iBAC1B,KAAK,OAAO;EAC3B,KAAK;;;;;;;;CASL,MAAM,oBACJ,YACiB;EACjB,MAAM,SAAS,eAAe,aAC5B,+JAMD;AASD,UAPe,MAAM,KAAK,MAAM,OAAO,CAAC,KACtC;GACE,MAAM,KAAK;GACX,SAAS,CAAC,GAAG,KAAK,KAAK,yBAAyB;GACjD,EACD,YAAY,SAAS,wBAAwB,CAC9C,EACa,OAAO,MAAM;;;;;;;CAQ7B,cACE,SAGI,EAAE,EACE;EACR,MAAM,EAAE,sBAAM,IAAI,MAAM,EAAE,eAAe,UAAU;AAWnD,SAAO,GATS,KAAK,WAAW;GAAE;GAAK;GAAc,CAAC,CASpC,UARQ,IAAI,eAAe,SAAS;GACpD,OAAO;GACP,KAAK;GACL,MAAM;GACN,MAAM;GACN,QAAQ;GACR,QAAQ;GACT,CAAC,CAC4C,KAAK,KAAK,KAAK,aAAa,KAAK;;;;;;;;;;CAWjF,MAAM,UACJ,eACA,KACA,UACA,WACA;AACA,SAAO,KAAK,eAAe,UACzB,eACA,KACA,UACA,UACD"}

package/dist/experimental/generative_agents/generative_agent_memory.cjs

@@ -1,9 +1,9 @@
-const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
 const require_base = require('../../chains/base.cjs');
 const require_llm_chain = require('../../chains/llm_chain.cjs');
-const __langchain_core_documents = require_rolldown_runtime.__toESM(require("@langchain/core/documents"));
-const __langchain_core_prompts = require_rolldown_runtime.__toESM(require("@langchain/core/prompts"));
-const __langchain_core_memory = require_rolldown_runtime.__toESM(require("@langchain/core/memory"));
+let _langchain_core_prompts = require("@langchain/core/prompts");
+let _langchain_core_memory = require("@langchain/core/memory");
+let _langchain_core_documents = require("@langchain/core/documents");
 
 //#region src/experimental/generative_agents/generative_agent_memory.ts
 /**
@@ -51,19 +51,18 @@ var GenerativeAgentMemoryChain = class GenerativeAgentMemoryChain extends requir
 	* @returns A new LLMChain instance.
 	*/
 	chain(prompt) {
-		const chain = new require_llm_chain.LLMChain({
+		return new require_llm_chain.LLMChain({
 			llm: this.llm,
 			prompt,
 			verbose: this.verbose,
 			outputKey: "output"
 		});
-		return chain;
 	}
 	async _call(values, runManager) {
 		const { memory_content: memoryContent, now } = values;
 		const importanceScore = await this.scoreMemoryImportance(memoryContent, runManager);
 		this.aggregateImportance += importanceScore;
-		const document = new __langchain_core_documents.Document({
+		const document = new _langchain_core_documents.Document({
 			pageContent: memoryContent,
 			metadata: {
 				importance: importanceScore,
@@ -109,16 +108,12 @@ var GenerativeAgentMemoryChain = class GenerativeAgentMemoryChain extends requir
 	* @returns The importance score of the memory as a number.
 	*/
 	async scoreMemoryImportance(memoryContent, runManager) {
-		const prompt = __langchain_core_prompts.PromptTemplate.fromTemplate("On the scale of 1 to 10, where 1 is purely mundane (e.g., brushing teeth, making bed) and 10 is extremely poignant (e.g., a break up, college acceptance), rate the likely poignancy of the following piece of memory. Respond with a single integer.\nMemory: {memory_content}\nRating: ");
-		const score = await this.chain(prompt).run(memoryContent, runManager?.getChild("determine_importance"));
-		const strippedScore = score.trim();
+		const prompt = _langchain_core_prompts.PromptTemplate.fromTemplate("On the scale of 1 to 10, where 1 is purely mundane (e.g., brushing teeth, making bed) and 10 is extremely poignant (e.g., a break up, college acceptance), rate the likely poignancy of the following piece of memory. Respond with a single integer.\nMemory: {memory_content}\nRating: ");
+		const strippedScore = (await this.chain(prompt).run(memoryContent, runManager?.getChild("determine_importance"))).trim();
 		if (this.verbose) console.log("Importance score:", strippedScore);
 		const match = strippedScore.match(/^\D*(\d+)/);
-		if (match) {
-			const capturedNumber = parseFloat(match[1]);
-			const result = capturedNumber / 10 * this.importanceWeight;
-			return result;
-		} else return 0;
+		if (match) return parseFloat(match[1]) / 10 * this.importanceWeight;
+		else return 0;
 	}
 	/**
 	* Method that retrieves the topics of reflection based on the last K
@@ -128,9 +123,8 @@ var GenerativeAgentMemoryChain = class GenerativeAgentMemoryChain extends requir
 	* @returns An array of topics of reflection as strings.
 	*/
 	async getTopicsOfReflection(lastK, runManager) {
-		const prompt = __langchain_core_prompts.PromptTemplate.fromTemplate("{observations}\n\nGiven only the information above, what are the 3 most salient high-level questions we can answer about the subjects in the statements? Provide each question on a new line.\n\n");
-		const observations = this.memoryRetriever.getMemoryStream().slice(-lastK);
-		const observationStr = observations.map((o) => o.pageContent).join("\n");
+		const prompt = _langchain_core_prompts.PromptTemplate.fromTemplate("{observations}\n\nGiven only the information above, what are the 3 most salient high-level questions we can answer about the subjects in the statements? Provide each question on a new line.\n\n");
+		const observationStr = this.memoryRetriever.getMemoryStream().slice(-lastK).map((o) => o.pageContent).join("\n");
 		const result = await this.chain(prompt).run(observationStr, runManager?.getChild("reflection_topics"));
 		return GenerativeAgentMemoryChain.parseList(result);
 	}
@@ -143,9 +137,8 @@ var GenerativeAgentMemoryChain = class GenerativeAgentMemoryChain extends requir
 	* @returns An array of insights as strings.
 	*/
 	async getInsightsOnTopic(topic, now, runManager) {
-		const prompt = __langchain_core_prompts.PromptTemplate.fromTemplate("Statements about {topic}\n{related_statements}\n\nWhat 5 high-level insights can you infer from the above statements? (example format: insight (because of 1, 5, 3))");
-		const relatedMemories = await this.fetchMemories(topic, now, runManager);
-		const relatedStatements = relatedMemories.map((memory, index) => `${index + 1}. ${memory.pageContent}`).join("\n");
+		const prompt = _langchain_core_prompts.PromptTemplate.fromTemplate("Statements about {topic}\n{related_statements}\n\nWhat 5 high-level insights can you infer from the above statements? (example format: insight (because of 1, 5, 3))");
+		const relatedStatements = (await this.fetchMemories(topic, now, runManager)).map((memory, index) => `${index + 1}. ${memory.pageContent}`).join("\n");
 		const result = await this.chain(prompt).call({
 			topic,
 			related_statements: relatedStatements
@@ -196,7 +189,7 @@ var GenerativeAgentMemoryChain = class GenerativeAgentMemoryChain extends requir
 * const summary = await tommiesMemory.getSummary();
 * ```
 */
-var GenerativeAgentMemory = class extends __langchain_core_memory.BaseMemory {
+var GenerativeAgentMemory = class extends _langchain_core_memory.BaseMemory {
 	llm;
 	memoryRetriever;
 	verbose;
@@ -290,8 +283,7 @@ var GenerativeAgentMemory = class extends __langchain_core_memory.BaseMemory {
 			});
 			content.push(`${createdTime}: ${memory.pageContent.trim()}`);
 		}
-		const joinedContent = content.map((mem) => `${mem}`).join("\n");
-		return joinedContent;
+		return content.map((mem) => `${mem}`).join("\n");
 	}
 	/**
 	* Method that formats the given relevant memories in a simple manner.
@@ -299,8 +291,7 @@ var GenerativeAgentMemory = class extends __langchain_core_memory.BaseMemory {
 	* @returns The formatted memories as a string.
 	*/
 	formatMemoriesSimple(relevantMemories) {
-		const joinedContent = relevantMemories.map((mem) => `${mem.pageContent}`).join("; ");
-		return joinedContent;
+		return relevantMemories.map((mem) => `${mem.pageContent}`).join("; ");
 	}
 	/**
 	* Method that retrieves memories until a token limit is reached.

package/dist/experimental/generative_agents/generative_agent_memory.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"generative_agent_memory.cjs","names":["BaseChain","llm: BaseLanguageModelInterface","memoryRetriever: TimeWeightedVectorStoreRetriever","config: Omit<GenerativeAgentMemoryConfig, \"maxTokensLimit\">","prompt: PromptTemplate","LLMChain","values: ChainValues","runManager?: CallbackManagerForChainRun","Document","now?: Date","newInsights: string[]","memoryContent: string","PromptTemplate","lastK: number","o: { pageContent: string }","topic: string","relatedStatements: string","text: string","observation: string","_now?: Date","BaseMemory","config?: GenerativeAgentMemoryConfig","metadata?: Record<string, unknown>","callbacks?: Callbacks","relevantMemories: Document[]","consumedTokens: number","inputs: InputValues","query: string","_inputs: InputValues","outputs: OutputValues"],"sources":["../../../src/experimental/generative_agents/generative_agent_memory.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { Document } from \"@langchain/core/documents\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BaseMemory, InputValues, OutputValues } from \"@langchain/core/memory\";\nimport {\n  CallbackManagerForChainRun,\n  Callbacks,\n} from \"@langchain/core/callbacks/manager\";\nimport { TimeWeightedVectorStoreRetriever } from \"../../retrievers/time_weighted.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\n\nexport type GenerativeAgentMemoryConfig = {\n  reflectionThreshold?: number;\n  importanceWeight?: number;\n  verbose?: boolean;\n  maxTokensLimit?: number;\n};\n\n/**\n * Class that manages the memory of a generative agent in LangChain. It\n * extends the `BaseChain` class and has methods for adding observations\n * or memories to the agent's memory, scoring the importance of a memory,\n * reflecting on recent events to add synthesized memories, and generating\n * insights on a topic of reflection based on pertinent memories.\n */\nclass GenerativeAgentMemoryChain extends BaseChain {\n  static lc_name() {\n    return \"GenerativeAgentMemoryChain\";\n  }\n\n  reflecting = false;\n\n  reflectionThreshold?: number;\n\n  importanceWeight = 0.15;\n\n  memoryRetriever: TimeWeightedVectorStoreRetriever;\n\n  llm: BaseLanguageModelInterface;\n\n  verbose = false;\n\n  private aggregateImportance = 0.0;\n\n  constructor(\n    llm: BaseLanguageModelInterface,\n    memoryRetriever: TimeWeightedVectorStoreRetriever,\n    config: Omit<GenerativeAgentMemoryConfig, \"maxTokensLimit\">\n  ) {\n    super();\n    this.llm = llm;\n    this.memoryRetriever = memoryRetriever;\n    this.reflectionThreshold = config.reflectionThreshold;\n    this.importanceWeight = config.importanceWeight ?? this.importanceWeight;\n    this.verbose = config.verbose ?? this.verbose;\n  }\n\n  _chainType(): string {\n    return \"generative_agent_memory\";\n  }\n\n  get inputKeys(): string[] {\n    return [\"memory_content\", \"now\", \"memory_metadata\"];\n  }\n\n  get outputKeys(): string[] {\n    return [\"output\"];\n  }\n\n  /**\n   * Method that creates a new LLMChain with the given prompt.\n   * @param prompt The PromptTemplate to use for the new LLMChain.\n   * @returns A new LLMChain instance.\n   */\n  chain(prompt: PromptTemplate): LLMChain {\n    const chain = new LLMChain({\n      llm: this.llm,\n      prompt,\n      verbose: this.verbose,\n      outputKey: \"output\",\n    });\n    return chain;\n  }\n\n  async _call(values: ChainValues, runManager?: CallbackManagerForChainRun) {\n    const { memory_content: memoryContent, now } = values;\n    // add an observation or memory to the agent's memory\n    const importanceScore = await this.scoreMemoryImportance(\n      memoryContent,\n      runManager\n    );\n    this.aggregateImportance += importanceScore;\n    const document = new Document({\n      pageContent: memoryContent,\n      metadata: {\n        importance: importanceScore,\n        ...values.memory_metadata,\n      },\n    });\n    await this.memoryRetriever.addDocuments([document]);\n    // after an agent has processed a certain amount of memories (as measured by aggregate importance),\n    // it is time to pause and reflect on recent events to add more synthesized memories to the agent's\n    // memory stream.\n    if (\n      this.reflectionThreshold !== undefined &&\n      this.aggregateImportance > this.reflectionThreshold &&\n      !this.reflecting\n    ) {\n      console.log(\"Reflecting on current memories...\");\n      this.reflecting = true;\n      await this.pauseToReflect(now, runManager);\n      this.aggregateImportance = 0.0;\n      this.reflecting = false;\n    }\n    return { output: importanceScore };\n  }\n\n  /**\n   * Method that pauses the agent to reflect on recent events and generate\n   * new insights.\n   * @param now The current date.\n   * @param runManager The CallbackManagerForChainRun to use for the reflection.\n   * @returns An array of new insights as strings.\n   */\n  async pauseToReflect(\n    now?: Date,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string[]> {\n    if (this.verbose) {\n      console.log(\"Pausing to reflect...\");\n    }\n    const newInsights: string[] = [];\n    const topics = await this.getTopicsOfReflection(50, runManager);\n    for (const topic of topics) {\n      const insights = await this.getInsightsOnTopic(topic, now, runManager);\n      for (const insight of insights) {\n        // add memory\n        await this.call(\n          {\n            memory_content: insight,\n            now,\n            memory_metadata: {\n              source: \"reflection_insight\",\n            },\n          },\n          runManager?.getChild(\"reflection_insight_memory\")\n        );\n      }\n      newInsights.push(...insights);\n    }\n    return newInsights;\n  }\n\n  /**\n   * Method that scores the importance of a given memory.\n   * @param memoryContent The content of the memory to score.\n   * @param runManager The CallbackManagerForChainRun to use for scoring.\n   * @returns The importance score of the memory as a number.\n   */\n  async scoreMemoryImportance(\n    memoryContent: string,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<number> {\n    // score the absolute importance of a given memory\n    const prompt = PromptTemplate.fromTemplate(\n      \"On the scale of 1 to 10, where 1 is purely mundane\" +\n        \" (e.g., brushing teeth, making bed) and 10 is\" +\n        \" extremely poignant (e.g., a break up, college\" +\n        \" acceptance), rate the likely poignancy of the\" +\n        \" following piece of memory. Respond with a single integer.\" +\n        \"\\nMemory: {memory_content}\" +\n        \"\\nRating: \"\n    );\n    const score = await this.chain(prompt).run(\n      memoryContent,\n      runManager?.getChild(\"determine_importance\")\n    );\n\n    const strippedScore = score.trim();\n\n    if (this.verbose) {\n      console.log(\"Importance score:\", strippedScore);\n    }\n    const match = strippedScore.match(/^\\D*(\\d+)/);\n    if (match) {\n      const capturedNumber = parseFloat(match[1]);\n      const result = (capturedNumber / 10) * this.importanceWeight;\n      return result;\n    } else {\n      return 0.0;\n    }\n  }\n\n  /**\n   * Method that retrieves the topics of reflection based on the last K\n   * memories.\n   * @param lastK The number of most recent memories to consider for generating topics.\n   * @param runManager The CallbackManagerForChainRun to use for retrieving topics.\n   * @returns An array of topics of reflection as strings.\n   */\n  async getTopicsOfReflection(\n    lastK: number,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string[]> {\n    const prompt = PromptTemplate.fromTemplate(\n      \"{observations}\\n\\n\" +\n        \"Given only the information above, what are the 3 most salient\" +\n        \" high-level questions we can answer about the subjects in\" +\n        \" the statements? Provide each question on a new line.\\n\\n\"\n    );\n\n    const observations = this.memoryRetriever.getMemoryStream().slice(-lastK);\n    const observationStr = observations\n      .map((o: { pageContent: string }) => o.pageContent)\n      .join(\"\\n\");\n    const result = await this.chain(prompt).run(\n      observationStr,\n      runManager?.getChild(\"reflection_topics\")\n    );\n    return GenerativeAgentMemoryChain.parseList(result);\n  }\n\n  /**\n   * Method that generates insights on a given topic of reflection based on\n   * pertinent memories.\n   * @param topic The topic of reflection.\n   * @param now The current date.\n   * @param runManager The CallbackManagerForChainRun to use for generating insights.\n   * @returns An array of insights as strings.\n   */\n  async getInsightsOnTopic(\n    topic: string,\n    now?: Date,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string[]> {\n    // generate insights on a topic of reflection, based on pertinent memories\n    const prompt = PromptTemplate.fromTemplate(\n      \"Statements about {topic}\\n\" +\n        \"{related_statements}\\n\\n\" +\n        \"What 5 high-level insights can you infer from the above statements?\" +\n        \" (example format: insight (because of 1, 5, 3))\"\n    );\n\n    const relatedMemories = await this.fetchMemories(topic, now, runManager);\n    const relatedStatements: string = relatedMemories\n      .map((memory, index) => `${index + 1}. ${memory.pageContent}`)\n      .join(\"\\n\");\n    const result = await this.chain(prompt).call(\n      {\n        topic,\n        related_statements: relatedStatements,\n      },\n      runManager?.getChild(\"reflection_insights\")\n    );\n    return GenerativeAgentMemoryChain.parseList(result.output); // added output\n  }\n\n  /**\n   * Method that parses a newline-separated string into a list of strings.\n   * @param text The newline-separated string to parse.\n   * @returns An array of strings.\n   */\n  static parseList(text: string): string[] {\n    // parse a newine seperates string into a list of strings\n    return text.split(\"\\n\").map((s) => s.trim());\n  }\n\n  // TODO: Mock \"now\" to simulate different times\n  /**\n   * Method that fetches memories related to a given observation.\n   * @param observation The observation to fetch memories for.\n   * @param _now The current date.\n   * @param runManager The CallbackManagerForChainRun to use for fetching memories.\n   * @returns An array of Document instances representing the fetched memories.\n   */\n  async fetchMemories(\n    observation: string,\n    _now?: Date,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<Document[]> {\n    return this.memoryRetriever.invoke(\n      observation,\n      runManager?.getChild(\"memory_retriever\")\n    );\n  }\n}\n\n/**\n * Class that manages the memory of a generative agent in LangChain. It\n * extends the `BaseMemory` class and has methods for adding a memory,\n * formatting memories, getting memories until a token limit is reached,\n * loading memory variables, saving the context of a model run to memory,\n * and clearing memory contents.\n * @example\n * ```typescript\n * const createNewMemoryRetriever = async () => {\n *   const vectorStore = new MemoryVectorStore(new OpenAIEmbeddings());\n *   const retriever = new TimeWeightedVectorStoreRetriever({\n *     vectorStore,\n *     otherScoreKeys: [\"importance\"],\n *     k: 15,\n *   });\n *   return retriever;\n * };\n * const tommiesMemory = new GenerativeAgentMemory(\n *   llm,\n *   await createNewMemoryRetriever(),\n *   { reflectionThreshold: 8 },\n * );\n * const summary = await tommiesMemory.getSummary();\n * ```\n */\nexport class GenerativeAgentMemory extends BaseMemory {\n  llm: BaseLanguageModelInterface;\n\n  memoryRetriever: TimeWeightedVectorStoreRetriever;\n\n  verbose: boolean;\n\n  reflectionThreshold?: number;\n\n  private maxTokensLimit = 1200;\n\n  queriesKey = \"queries\";\n\n  mostRecentMemoriesTokenKey = \"recent_memories_token\";\n\n  addMemoryKey = \"addMemory\";\n\n  relevantMemoriesKey = \"relevant_memories\";\n\n  relevantMemoriesSimpleKey = \"relevant_memories_simple\";\n\n  mostRecentMemoriesKey = \"most_recent_memories\";\n\n  nowKey = \"now\";\n\n  memoryChain: GenerativeAgentMemoryChain;\n\n  constructor(\n    llm: BaseLanguageModelInterface,\n    memoryRetriever: TimeWeightedVectorStoreRetriever,\n    config?: GenerativeAgentMemoryConfig\n  ) {\n    super();\n    this.llm = llm;\n    this.memoryRetriever = memoryRetriever;\n    this.verbose = config?.verbose ?? this.verbose;\n    this.reflectionThreshold =\n      config?.reflectionThreshold ?? this.reflectionThreshold;\n    this.maxTokensLimit = config?.maxTokensLimit ?? this.maxTokensLimit;\n    this.memoryChain = new GenerativeAgentMemoryChain(llm, memoryRetriever, {\n      reflectionThreshold: config?.reflectionThreshold,\n      importanceWeight: config?.importanceWeight,\n    });\n  }\n\n  /**\n   * Method that returns the key for relevant memories.\n   * @returns The key for relevant memories as a string.\n   */\n  getRelevantMemoriesKey(): string {\n    return this.relevantMemoriesKey;\n  }\n\n  /**\n   * Method that returns the key for the most recent memories token.\n   * @returns The key for the most recent memories token as a string.\n   */\n  getMostRecentMemoriesTokenKey(): string {\n    return this.mostRecentMemoriesTokenKey;\n  }\n\n  /**\n   * Method that returns the key for adding a memory.\n   * @returns The key for adding a memory as a string.\n   */\n  getAddMemoryKey(): string {\n    return this.addMemoryKey;\n  }\n\n  /**\n   * Method that returns the key for the current time.\n   * @returns The key for the current time as a string.\n   */\n  getCurrentTimeKey(): string {\n    return this.nowKey;\n  }\n\n  get memoryKeys(): string[] {\n    // Return an array of memory keys\n    return [this.relevantMemoriesKey, this.mostRecentMemoriesKey];\n  }\n\n  /**\n   * Method that adds a memory to the agent's memory.\n   * @param memoryContent The content of the memory to add.\n   * @param now The current date.\n   * @param metadata The metadata for the memory.\n   * @param callbacks The Callbacks to use for adding the memory.\n   * @returns The result of the memory addition.\n   */\n  async addMemory(\n    memoryContent: string,\n    now?: Date,\n    metadata?: Record<string, unknown>,\n    callbacks?: Callbacks\n  ) {\n    return this.memoryChain.call(\n      { memory_content: memoryContent, now, memory_metadata: metadata },\n      callbacks\n    );\n  }\n\n  /**\n   * Method that formats the given relevant memories in detail.\n   * @param relevantMemories The relevant memories to format.\n   * @returns The formatted memories as a string.\n   */\n  formatMemoriesDetail(relevantMemories: Document[]): string {\n    if (!relevantMemories.length) {\n      return \"No relevant information.\";\n    }\n    const contentStrings = new Set();\n    const content = [];\n    for (const memory of relevantMemories) {\n      if (memory.pageContent in contentStrings) {\n        continue;\n      }\n      contentStrings.add(memory.pageContent);\n      const createdTime = memory.metadata.created_at.toLocaleString(\"en-US\", {\n        month: \"long\",\n        day: \"numeric\",\n        year: \"numeric\",\n        hour: \"numeric\",\n        minute: \"numeric\",\n        hour12: true,\n      });\n      content.push(`${createdTime}: ${memory.pageContent.trim()}`);\n    }\n    const joinedContent = content.map((mem) => `${mem}`).join(\"\\n\");\n    return joinedContent;\n  }\n\n  /**\n   * Method that formats the given relevant memories in a simple manner.\n   * @param relevantMemories The relevant memories to format.\n   * @returns The formatted memories as a string.\n   */\n  formatMemoriesSimple(relevantMemories: Document[]): string {\n    const joinedContent = relevantMemories\n      .map((mem) => `${mem.pageContent}`)\n      .join(\"; \");\n    return joinedContent;\n  }\n\n  /**\n   * Method that retrieves memories until a token limit is reached.\n   * @param consumedTokens The number of tokens consumed so far.\n   * @returns The memories as a string.\n   */\n  async getMemoriesUntilLimit(consumedTokens: number): Promise<string> {\n    // reduce the number of tokens in the documents\n    const result = [];\n    for (const doc of this.memoryRetriever\n      .getMemoryStream()\n      .slice()\n      .reverse()) {\n      if (consumedTokens >= this.maxTokensLimit) {\n        if (this.verbose) {\n          console.log(\"Exceeding max tokens for LLM, filtering memories\");\n        }\n        break;\n      }\n      // eslint-disable-next-line no-param-reassign\n      consumedTokens += await this.llm.getNumTokens(doc.pageContent);\n      if (consumedTokens < this.maxTokensLimit) {\n        result.push(doc);\n      }\n    }\n    return this.formatMemoriesSimple(result);\n  }\n\n  get memoryVariables(): string[] {\n    // input keys this memory class will load dynamically\n    return [];\n  }\n\n  /**\n   * Method that loads memory variables based on the given inputs.\n   * @param inputs The inputs to use for loading memory variables.\n   * @returns An object containing the loaded memory variables.\n   */\n  async loadMemoryVariables(\n    inputs: InputValues\n  ): Promise<Record<string, string>> {\n    const queries = inputs[this.queriesKey];\n    const now = inputs[this.nowKey];\n    if (queries !== undefined) {\n      const relevantMemories = (\n        await Promise.all(\n          queries.map((query: string) =>\n            this.memoryChain.fetchMemories(query, now)\n          )\n        )\n      ).flat();\n      return {\n        [this.relevantMemoriesKey]: this.formatMemoriesDetail(relevantMemories),\n        [this.relevantMemoriesSimpleKey]:\n          this.formatMemoriesSimple(relevantMemories),\n      };\n    }\n    const mostRecentMemoriesToken = inputs[this.mostRecentMemoriesTokenKey];\n    if (mostRecentMemoriesToken !== undefined) {\n      return {\n        [this.mostRecentMemoriesKey]: await this.getMemoriesUntilLimit(\n          mostRecentMemoriesToken\n        ),\n      };\n    }\n    return {};\n  }\n\n  /**\n   * Method that saves the context of a model run to memory.\n   * @param _inputs The inputs of the model run.\n   * @param outputs The outputs of the model run.\n   * @returns Nothing.\n   */\n  async saveContext(\n    _inputs: InputValues,\n    outputs: OutputValues\n  ): Promise<void> {\n    // save the context of this model run to memory\n    const mem = outputs[this.addMemoryKey];\n    const now = outputs[this.nowKey];\n    if (mem) {\n      await this.addMemory(mem, now, {});\n    }\n  }\n\n  /**\n   * Method that clears the memory contents.\n   * @returns Nothing.\n   */\n  clear(): void {\n    // TODO: clear memory contents\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;AA2BA,IAAM,6BAAN,MAAM,mCAAmCA,uBAAU;CACjD,OAAO,UAAU;AACf,SAAO;CACR;CAED,aAAa;CAEb;CAEA,mBAAmB;CAEnB;CAEA;CAEA,UAAU;CAEV,AAAQ,sBAAsB;CAE9B,YACEC,KACAC,iBACAC,QACA;EACA,OAAO;EACP,KAAK,MAAM;EACX,KAAK,kBAAkB;EACvB,KAAK,sBAAsB,OAAO;EAClC,KAAK,mBAAmB,OAAO,oBAAoB,KAAK;EACxD,KAAK,UAAU,OAAO,WAAW,KAAK;CACvC;CAED,aAAqB;AACnB,SAAO;CACR;CAED,IAAI,YAAsB;AACxB,SAAO;GAAC;GAAkB;GAAO;EAAkB;CACpD;CAED,IAAI,aAAuB;AACzB,SAAO,CAAC,QAAS;CAClB;;;;;;CAOD,MAAMC,QAAkC;EACtC,MAAM,QAAQ,IAAIC,2BAAS;GACzB,KAAK,KAAK;GACV;GACA,SAAS,KAAK;GACd,WAAW;EACZ;AACD,SAAO;CACR;CAED,MAAM,MAAMC,QAAqBC,YAAyC;EACxE,MAAM,EAAE,gBAAgB,eAAe,KAAK,GAAG;EAE/C,MAAM,kBAAkB,MAAM,KAAK,sBACjC,eACA,WACD;EACD,KAAK,uBAAuB;EAC5B,MAAM,WAAW,IAAIC,oCAAS;GAC5B,aAAa;GACb,UAAU;IACR,YAAY;IACZ,GAAG,OAAO;GACX;EACF;EACD,MAAM,KAAK,gBAAgB,aAAa,CAAC,QAAS,EAAC;AAInD,MACE,KAAK,wBAAwB,UAC7B,KAAK,sBAAsB,KAAK,uBAChC,CAAC,KAAK,YACN;GACA,QAAQ,IAAI,oCAAoC;GAChD,KAAK,aAAa;GAClB,MAAM,KAAK,eAAe,KAAK,WAAW;GAC1C,KAAK,sBAAsB;GAC3B,KAAK,aAAa;EACnB;AACD,SAAO,EAAE,QAAQ,gBAAiB;CACnC;;;;;;;;CASD,MAAM,eACJC,KACAF,YACmB;AACnB,MAAI,KAAK,SACP,QAAQ,IAAI,wBAAwB;EAEtC,MAAMG,cAAwB,CAAE;EAChC,MAAM,SAAS,MAAM,KAAK,sBAAsB,IAAI,WAAW;AAC/D,OAAK,MAAM,SAAS,QAAQ;GAC1B,MAAM,WAAW,MAAM,KAAK,mBAAmB,OAAO,KAAK,WAAW;AACtE,QAAK,MAAM,WAAW,UAEpB,MAAM,KAAK,KACT;IACE,gBAAgB;IAChB;IACA,iBAAiB,EACf,QAAQ,qBACT;GACF,GACD,YAAY,SAAS,4BAA4B,CAClD;GAEH,YAAY,KAAK,GAAG,SAAS;EAC9B;AACD,SAAO;CACR;;;;;;;CAQD,MAAM,sBACJC,eACAJ,YACiB;EAEjB,MAAM,SAASK,wCAAe,aAC5B,4RAOD;EACD,MAAM,QAAQ,MAAM,KAAK,MAAM,OAAO,CAAC,IACrC,eACA,YAAY,SAAS,uBAAuB,CAC7C;EAED,MAAM,gBAAgB,MAAM,MAAM;AAElC,MAAI,KAAK,SACP,QAAQ,IAAI,qBAAqB,cAAc;EAEjD,MAAM,QAAQ,cAAc,MAAM,YAAY;AAC9C,MAAI,OAAO;GACT,MAAM,iBAAiB,WAAW,MAAM,GAAG;GAC3C,MAAM,SAAU,iBAAiB,KAAM,KAAK;AAC5C,UAAO;EACR,MACC,QAAO;CAEV;;;;;;;;CASD,MAAM,sBACJC,OACAN,YACmB;EACnB,MAAM,SAASK,wCAAe,aAC5B,oMAID;EAED,MAAM,eAAe,KAAK,gBAAgB,iBAAiB,CAAC,MAAM,CAAC,MAAM;EACzE,MAAM,iBAAiB,aACpB,IAAI,CAACE,MAA+B,EAAE,YAAY,CAClD,KAAK,KAAK;EACb,MAAM,SAAS,MAAM,KAAK,MAAM,OAAO,CAAC,IACtC,gBACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,SAAO,2BAA2B,UAAU,OAAO;CACpD;;;;;;;;;CAUD,MAAM,mBACJC,OACAN,KACAF,YACmB;EAEnB,MAAM,SAASK,wCAAe,aAC5B,uKAID;EAED,MAAM,kBAAkB,MAAM,KAAK,cAAc,OAAO,KAAK,WAAW;EACxE,MAAMI,oBAA4B,gBAC/B,IAAI,CAAC,QAAQ,UAAU,GAAG,QAAQ,EAAE,EAAE,EAAE,OAAO,aAAa,CAAC,CAC7D,KAAK,KAAK;EACb,MAAM,SAAS,MAAM,KAAK,MAAM,OAAO,CAAC,KACtC;GACE;GACA,oBAAoB;EACrB,GACD,YAAY,SAAS,sBAAsB,CAC5C;AACD,SAAO,2BAA2B,UAAU,OAAO,OAAO;CAC3D;;;;;;CAOD,OAAO,UAAUC,MAAwB;AAEvC,SAAO,KAAK,MAAM,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC;CAC7C;;;;;;;;CAUD,MAAM,cACJC,aACAC,MACAZ,YACqB;AACrB,SAAO,KAAK,gBAAgB,OAC1B,aACA,YAAY,SAAS,mBAAmB,CACzC;CACF;AACF;;;;;;;;;;;;;;;;;;;;;;;;;;AA2BD,IAAa,wBAAb,cAA2Ca,mCAAW;CACpD;CAEA;CAEA;CAEA;CAEA,AAAQ,iBAAiB;CAEzB,aAAa;CAEb,6BAA6B;CAE7B,eAAe;CAEf,sBAAsB;CAEtB,4BAA4B;CAE5B,wBAAwB;CAExB,SAAS;CAET;CAEA,YACEnB,KACAC,iBACAmB,QACA;EACA,OAAO;EACP,KAAK,MAAM;EACX,KAAK,kBAAkB;EACvB,KAAK,UAAU,QAAQ,WAAW,KAAK;EACvC,KAAK,sBACH,QAAQ,uBAAuB,KAAK;EACtC,KAAK,iBAAiB,QAAQ,kBAAkB,KAAK;EACrD,KAAK,cAAc,IAAI,2BAA2B,KAAK,iBAAiB;GACtE,qBAAqB,QAAQ;GAC7B,kBAAkB,QAAQ;EAC3B;CACF;;;;;CAMD,yBAAiC;AAC/B,SAAO,KAAK;CACb;;;;;CAMD,gCAAwC;AACtC,SAAO,KAAK;CACb;;;;;CAMD,kBAA0B;AACxB,SAAO,KAAK;CACb;;;;;CAMD,oBAA4B;AAC1B,SAAO,KAAK;CACb;CAED,IAAI,aAAuB;AAEzB,SAAO,CAAC,KAAK,qBAAqB,KAAK,qBAAsB;CAC9D;;;;;;;;;CAUD,MAAM,UACJV,eACAF,KACAa,UACAC,WACA;AACA,SAAO,KAAK,YAAY,KACtB;GAAE,gBAAgB;GAAe;GAAK,iBAAiB;EAAU,GACjE,UACD;CACF;;;;;;CAOD,qBAAqBC,kBAAsC;AACzD,MAAI,CAAC,iBAAiB,OACpB,QAAO;EAET,MAAM,iCAAiB,IAAI;EAC3B,MAAM,UAAU,CAAE;AAClB,OAAK,MAAM,UAAU,kBAAkB;AACrC,OAAI,OAAO,eAAe,eACxB;GAEF,eAAe,IAAI,OAAO,YAAY;GACtC,MAAM,cAAc,OAAO,SAAS,WAAW,eAAe,SAAS;IACrE,OAAO;IACP,KAAK;IACL,MAAM;IACN,MAAM;IACN,QAAQ;IACR,QAAQ;GACT,EAAC;GACF,QAAQ,KAAK,GAAG,YAAY,EAAE,EAAE,OAAO,YAAY,MAAM,EAAE,CAAC;EAC7D;EACD,MAAM,gBAAgB,QAAQ,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,CAAC,KAAK,KAAK;AAC/D,SAAO;CACR;;;;;;CAOD,qBAAqBA,kBAAsC;EACzD,MAAM,gBAAgB,iBACnB,IAAI,CAAC,QAAQ,GAAG,IAAI,aAAa,CAAC,CAClC,KAAK,KAAK;AACb,SAAO;CACR;;;;;;CAOD,MAAM,sBAAsBC,gBAAyC;EAEnE,MAAM,SAAS,CAAE;AACjB,OAAK,MAAM,OAAO,KAAK,gBACpB,iBAAiB,CACjB,OAAO,CACP,SAAS,EAAE;AACZ,OAAI,kBAAkB,KAAK,gBAAgB;AACzC,QAAI,KAAK,SACP,QAAQ,IAAI,mDAAmD;AAEjE;GACD;GAED,kBAAkB,MAAM,KAAK,IAAI,aAAa,IAAI,YAAY;AAC9D,OAAI,iBAAiB,KAAK,gBACxB,OAAO,KAAK,IAAI;EAEnB;AACD,SAAO,KAAK,qBAAqB,OAAO;CACzC;CAED,IAAI,kBAA4B;AAE9B,SAAO,CAAE;CACV;;;;;;CAOD,MAAM,oBACJC,QACiC;EACjC,MAAM,UAAU,OAAO,KAAK;EAC5B,MAAM,MAAM,OAAO,KAAK;AACxB,MAAI,YAAY,QAAW;GACzB,MAAM,oBACJ,MAAM,QAAQ,IACZ,QAAQ,IAAI,CAACC,UACX,KAAK,YAAY,cAAc,OAAO,IAAI,CAC3C,CACF,EACD,MAAM;AACR,UAAO;KACJ,KAAK,sBAAsB,KAAK,qBAAqB,iBAAiB;KACtE,KAAK,4BACJ,KAAK,qBAAqB,iBAAiB;GAC9C;EACF;EACD,MAAM,0BAA0B,OAAO,KAAK;AAC5C,MAAI,4BAA4B,OAC9B,QAAO,GACJ,KAAK,wBAAwB,MAAM,KAAK,sBACvC,wBACD,CACF;AAEH,SAAO,CAAE;CACV;;;;;;;CAQD,MAAM,YACJC,SACAC,SACe;EAEf,MAAM,MAAM,QAAQ,KAAK;EACzB,MAAM,MAAM,QAAQ,KAAK;AACzB,MAAI,KACF,MAAM,KAAK,UAAU,KAAK,KAAK,CAAE,EAAC;CAErC;;;;;CAMD,QAAc,CAEb;AACF"}
+{"version":3,"file":"generative_agent_memory.cjs","names":["BaseChain","LLMChain","Document","PromptTemplate","BaseMemory"],"sources":["../../../src/experimental/generative_agents/generative_agent_memory.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { Document } from \"@langchain/core/documents\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BaseMemory, InputValues, OutputValues } from \"@langchain/core/memory\";\nimport {\n  CallbackManagerForChainRun,\n  Callbacks,\n} from \"@langchain/core/callbacks/manager\";\nimport { TimeWeightedVectorStoreRetriever } from \"../../retrievers/time_weighted.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\n\nexport type GenerativeAgentMemoryConfig = {\n  reflectionThreshold?: number;\n  importanceWeight?: number;\n  verbose?: boolean;\n  maxTokensLimit?: number;\n};\n\n/**\n * Class that manages the memory of a generative agent in LangChain. It\n * extends the `BaseChain` class and has methods for adding observations\n * or memories to the agent's memory, scoring the importance of a memory,\n * reflecting on recent events to add synthesized memories, and generating\n * insights on a topic of reflection based on pertinent memories.\n */\nclass GenerativeAgentMemoryChain extends BaseChain {\n  static lc_name() {\n    return \"GenerativeAgentMemoryChain\";\n  }\n\n  reflecting = false;\n\n  reflectionThreshold?: number;\n\n  importanceWeight = 0.15;\n\n  memoryRetriever: TimeWeightedVectorStoreRetriever;\n\n  llm: BaseLanguageModelInterface;\n\n  verbose = false;\n\n  private aggregateImportance = 0.0;\n\n  constructor(\n    llm: BaseLanguageModelInterface,\n    memoryRetriever: TimeWeightedVectorStoreRetriever,\n    config: Omit<GenerativeAgentMemoryConfig, \"maxTokensLimit\">\n  ) {\n    super();\n    this.llm = llm;\n    this.memoryRetriever = memoryRetriever;\n    this.reflectionThreshold = config.reflectionThreshold;\n    this.importanceWeight = config.importanceWeight ?? this.importanceWeight;\n    this.verbose = config.verbose ?? this.verbose;\n  }\n\n  _chainType(): string {\n    return \"generative_agent_memory\";\n  }\n\n  get inputKeys(): string[] {\n    return [\"memory_content\", \"now\", \"memory_metadata\"];\n  }\n\n  get outputKeys(): string[] {\n    return [\"output\"];\n  }\n\n  /**\n   * Method that creates a new LLMChain with the given prompt.\n   * @param prompt The PromptTemplate to use for the new LLMChain.\n   * @returns A new LLMChain instance.\n   */\n  chain(prompt: PromptTemplate): LLMChain {\n    const chain = new LLMChain({\n      llm: this.llm,\n      prompt,\n      verbose: this.verbose,\n      outputKey: \"output\",\n    });\n    return chain;\n  }\n\n  async _call(values: ChainValues, runManager?: CallbackManagerForChainRun) {\n    const { memory_content: memoryContent, now } = values;\n    // add an observation or memory to the agent's memory\n    const importanceScore = await this.scoreMemoryImportance(\n      memoryContent,\n      runManager\n    );\n    this.aggregateImportance += importanceScore;\n    const document = new Document({\n      pageContent: memoryContent,\n      metadata: {\n        importance: importanceScore,\n        ...values.memory_metadata,\n      },\n    });\n    await this.memoryRetriever.addDocuments([document]);\n    // after an agent has processed a certain amount of memories (as measured by aggregate importance),\n    // it is time to pause and reflect on recent events to add more synthesized memories to the agent's\n    // memory stream.\n    if (\n      this.reflectionThreshold !== undefined &&\n      this.aggregateImportance > this.reflectionThreshold &&\n      !this.reflecting\n    ) {\n      console.log(\"Reflecting on current memories...\");\n      this.reflecting = true;\n      await this.pauseToReflect(now, runManager);\n      this.aggregateImportance = 0.0;\n      this.reflecting = false;\n    }\n    return { output: importanceScore };\n  }\n\n  /**\n   * Method that pauses the agent to reflect on recent events and generate\n   * new insights.\n   * @param now The current date.\n   * @param runManager The CallbackManagerForChainRun to use for the reflection.\n   * @returns An array of new insights as strings.\n   */\n  async pauseToReflect(\n    now?: Date,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string[]> {\n    if (this.verbose) {\n      console.log(\"Pausing to reflect...\");\n    }\n    const newInsights: string[] = [];\n    const topics = await this.getTopicsOfReflection(50, runManager);\n    for (const topic of topics) {\n      const insights = await this.getInsightsOnTopic(topic, now, runManager);\n      for (const insight of insights) {\n        // add memory\n        await this.call(\n          {\n            memory_content: insight,\n            now,\n            memory_metadata: {\n              source: \"reflection_insight\",\n            },\n          },\n          runManager?.getChild(\"reflection_insight_memory\")\n        );\n      }\n      newInsights.push(...insights);\n    }\n    return newInsights;\n  }\n\n  /**\n   * Method that scores the importance of a given memory.\n   * @param memoryContent The content of the memory to score.\n   * @param runManager The CallbackManagerForChainRun to use for scoring.\n   * @returns The importance score of the memory as a number.\n   */\n  async scoreMemoryImportance(\n    memoryContent: string,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<number> {\n    // score the absolute importance of a given memory\n    const prompt = PromptTemplate.fromTemplate(\n      \"On the scale of 1 to 10, where 1 is purely mundane\" +\n        \" (e.g., brushing teeth, making bed) and 10 is\" +\n        \" extremely poignant (e.g., a break up, college\" +\n        \" acceptance), rate the likely poignancy of the\" +\n        \" following piece of memory. Respond with a single integer.\" +\n        \"\\nMemory: {memory_content}\" +\n        \"\\nRating: \"\n    );\n    const score = await this.chain(prompt).run(\n      memoryContent,\n      runManager?.getChild(\"determine_importance\")\n    );\n\n    const strippedScore = score.trim();\n\n    if (this.verbose) {\n      console.log(\"Importance score:\", strippedScore);\n    }\n    const match = strippedScore.match(/^\\D*(\\d+)/);\n    if (match) {\n      const capturedNumber = parseFloat(match[1]);\n      const result = (capturedNumber / 10) * this.importanceWeight;\n      return result;\n    } else {\n      return 0.0;\n    }\n  }\n\n  /**\n   * Method that retrieves the topics of reflection based on the last K\n   * memories.\n   * @param lastK The number of most recent memories to consider for generating topics.\n   * @param runManager The CallbackManagerForChainRun to use for retrieving topics.\n   * @returns An array of topics of reflection as strings.\n   */\n  async getTopicsOfReflection(\n    lastK: number,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string[]> {\n    const prompt = PromptTemplate.fromTemplate(\n      \"{observations}\\n\\n\" +\n        \"Given only the information above, what are the 3 most salient\" +\n        \" high-level questions we can answer about the subjects in\" +\n        \" the statements? Provide each question on a new line.\\n\\n\"\n    );\n\n    const observations = this.memoryRetriever.getMemoryStream().slice(-lastK);\n    const observationStr = observations\n      .map((o: { pageContent: string }) => o.pageContent)\n      .join(\"\\n\");\n    const result = await this.chain(prompt).run(\n      observationStr,\n      runManager?.getChild(\"reflection_topics\")\n    );\n    return GenerativeAgentMemoryChain.parseList(result);\n  }\n\n  /**\n   * Method that generates insights on a given topic of reflection based on\n   * pertinent memories.\n   * @param topic The topic of reflection.\n   * @param now The current date.\n   * @param runManager The CallbackManagerForChainRun to use for generating insights.\n   * @returns An array of insights as strings.\n   */\n  async getInsightsOnTopic(\n    topic: string,\n    now?: Date,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<string[]> {\n    // generate insights on a topic of reflection, based on pertinent memories\n    const prompt = PromptTemplate.fromTemplate(\n      \"Statements about {topic}\\n\" +\n        \"{related_statements}\\n\\n\" +\n        \"What 5 high-level insights can you infer from the above statements?\" +\n        \" (example format: insight (because of 1, 5, 3))\"\n    );\n\n    const relatedMemories = await this.fetchMemories(topic, now, runManager);\n    const relatedStatements: string = relatedMemories\n      .map((memory, index) => `${index + 1}. ${memory.pageContent}`)\n      .join(\"\\n\");\n    const result = await this.chain(prompt).call(\n      {\n        topic,\n        related_statements: relatedStatements,\n      },\n      runManager?.getChild(\"reflection_insights\")\n    );\n    return GenerativeAgentMemoryChain.parseList(result.output); // added output\n  }\n\n  /**\n   * Method that parses a newline-separated string into a list of strings.\n   * @param text The newline-separated string to parse.\n   * @returns An array of strings.\n   */\n  static parseList(text: string): string[] {\n    // parse a newine seperates string into a list of strings\n    return text.split(\"\\n\").map((s) => s.trim());\n  }\n\n  // TODO: Mock \"now\" to simulate different times\n  /**\n   * Method that fetches memories related to a given observation.\n   * @param observation The observation to fetch memories for.\n   * @param _now The current date.\n   * @param runManager The CallbackManagerForChainRun to use for fetching memories.\n   * @returns An array of Document instances representing the fetched memories.\n   */\n  async fetchMemories(\n    observation: string,\n    _now?: Date,\n    runManager?: CallbackManagerForChainRun\n  ): Promise<Document[]> {\n    return this.memoryRetriever.invoke(\n      observation,\n      runManager?.getChild(\"memory_retriever\")\n    );\n  }\n}\n\n/**\n * Class that manages the memory of a generative agent in LangChain. It\n * extends the `BaseMemory` class and has methods for adding a memory,\n * formatting memories, getting memories until a token limit is reached,\n * loading memory variables, saving the context of a model run to memory,\n * and clearing memory contents.\n * @example\n * ```typescript\n * const createNewMemoryRetriever = async () => {\n *   const vectorStore = new MemoryVectorStore(new OpenAIEmbeddings());\n *   const retriever = new TimeWeightedVectorStoreRetriever({\n *     vectorStore,\n *     otherScoreKeys: [\"importance\"],\n *     k: 15,\n *   });\n *   return retriever;\n * };\n * const tommiesMemory = new GenerativeAgentMemory(\n *   llm,\n *   await createNewMemoryRetriever(),\n *   { reflectionThreshold: 8 },\n * );\n * const summary = await tommiesMemory.getSummary();\n * ```\n */\nexport class GenerativeAgentMemory extends BaseMemory {\n  llm: BaseLanguageModelInterface;\n\n  memoryRetriever: TimeWeightedVectorStoreRetriever;\n\n  verbose: boolean;\n\n  reflectionThreshold?: number;\n\n  private maxTokensLimit = 1200;\n\n  queriesKey = \"queries\";\n\n  mostRecentMemoriesTokenKey = \"recent_memories_token\";\n\n  addMemoryKey = \"addMemory\";\n\n  relevantMemoriesKey = \"relevant_memories\";\n\n  relevantMemoriesSimpleKey = \"relevant_memories_simple\";\n\n  mostRecentMemoriesKey = \"most_recent_memories\";\n\n  nowKey = \"now\";\n\n  memoryChain: GenerativeAgentMemoryChain;\n\n  constructor(\n    llm: BaseLanguageModelInterface,\n    memoryRetriever: TimeWeightedVectorStoreRetriever,\n    config?: GenerativeAgentMemoryConfig\n  ) {\n    super();\n    this.llm = llm;\n    this.memoryRetriever = memoryRetriever;\n    this.verbose = config?.verbose ?? this.verbose;\n    this.reflectionThreshold =\n      config?.reflectionThreshold ?? this.reflectionThreshold;\n    this.maxTokensLimit = config?.maxTokensLimit ?? this.maxTokensLimit;\n    this.memoryChain = new GenerativeAgentMemoryChain(llm, memoryRetriever, {\n      reflectionThreshold: config?.reflectionThreshold,\n      importanceWeight: config?.importanceWeight,\n    });\n  }\n\n  /**\n   * Method that returns the key for relevant memories.\n   * @returns The key for relevant memories as a string.\n   */\n  getRelevantMemoriesKey(): string {\n    return this.relevantMemoriesKey;\n  }\n\n  /**\n   * Method that returns the key for the most recent memories token.\n   * @returns The key for the most recent memories token as a string.\n   */\n  getMostRecentMemoriesTokenKey(): string {\n    return this.mostRecentMemoriesTokenKey;\n  }\n\n  /**\n   * Method that returns the key for adding a memory.\n   * @returns The key for adding a memory as a string.\n   */\n  getAddMemoryKey(): string {\n    return this.addMemoryKey;\n  }\n\n  /**\n   * Method that returns the key for the current time.\n   * @returns The key for the current time as a string.\n   */\n  getCurrentTimeKey(): string {\n    return this.nowKey;\n  }\n\n  get memoryKeys(): string[] {\n    // Return an array of memory keys\n    return [this.relevantMemoriesKey, this.mostRecentMemoriesKey];\n  }\n\n  /**\n   * Method that adds a memory to the agent's memory.\n   * @param memoryContent The content of the memory to add.\n   * @param now The current date.\n   * @param metadata The metadata for the memory.\n   * @param callbacks The Callbacks to use for adding the memory.\n   * @returns The result of the memory addition.\n   */\n  async addMemory(\n    memoryContent: string,\n    now?: Date,\n    metadata?: Record<string, unknown>,\n    callbacks?: Callbacks\n  ) {\n    return this.memoryChain.call(\n      { memory_content: memoryContent, now, memory_metadata: metadata },\n      callbacks\n    );\n  }\n\n  /**\n   * Method that formats the given relevant memories in detail.\n   * @param relevantMemories The relevant memories to format.\n   * @returns The formatted memories as a string.\n   */\n  formatMemoriesDetail(relevantMemories: Document[]): string {\n    if (!relevantMemories.length) {\n      return \"No relevant information.\";\n    }\n    const contentStrings = new Set();\n    const content = [];\n    for (const memory of relevantMemories) {\n      if (memory.pageContent in contentStrings) {\n        continue;\n      }\n      contentStrings.add(memory.pageContent);\n      const createdTime = memory.metadata.created_at.toLocaleString(\"en-US\", {\n        month: \"long\",\n        day: \"numeric\",\n        year: \"numeric\",\n        hour: \"numeric\",\n        minute: \"numeric\",\n        hour12: true,\n      });\n      content.push(`${createdTime}: ${memory.pageContent.trim()}`);\n    }\n    const joinedContent = content.map((mem) => `${mem}`).join(\"\\n\");\n    return joinedContent;\n  }\n\n  /**\n   * Method that formats the given relevant memories in a simple manner.\n   * @param relevantMemories The relevant memories to format.\n   * @returns The formatted memories as a string.\n   */\n  formatMemoriesSimple(relevantMemories: Document[]): string {\n    const joinedContent = relevantMemories\n      .map((mem) => `${mem.pageContent}`)\n      .join(\"; \");\n    return joinedContent;\n  }\n\n  /**\n   * Method that retrieves memories until a token limit is reached.\n   * @param consumedTokens The number of tokens consumed so far.\n   * @returns The memories as a string.\n   */\n  async getMemoriesUntilLimit(consumedTokens: number): Promise<string> {\n    // reduce the number of tokens in the documents\n    const result = [];\n    for (const doc of this.memoryRetriever\n      .getMemoryStream()\n      .slice()\n      .reverse()) {\n      if (consumedTokens >= this.maxTokensLimit) {\n        if (this.verbose) {\n          console.log(\"Exceeding max tokens for LLM, filtering memories\");\n        }\n        break;\n      }\n      // eslint-disable-next-line no-param-reassign\n      consumedTokens += await this.llm.getNumTokens(doc.pageContent);\n      if (consumedTokens < this.maxTokensLimit) {\n        result.push(doc);\n      }\n    }\n    return this.formatMemoriesSimple(result);\n  }\n\n  get memoryVariables(): string[] {\n    // input keys this memory class will load dynamically\n    return [];\n  }\n\n  /**\n   * Method that loads memory variables based on the given inputs.\n   * @param inputs The inputs to use for loading memory variables.\n   * @returns An object containing the loaded memory variables.\n   */\n  async loadMemoryVariables(\n    inputs: InputValues\n  ): Promise<Record<string, string>> {\n    const queries = inputs[this.queriesKey];\n    const now = inputs[this.nowKey];\n    if (queries !== undefined) {\n      const relevantMemories = (\n        await Promise.all(\n          queries.map((query: string) =>\n            this.memoryChain.fetchMemories(query, now)\n          )\n        )\n      ).flat();\n      return {\n        [this.relevantMemoriesKey]: this.formatMemoriesDetail(relevantMemories),\n        [this.relevantMemoriesSimpleKey]:\n          this.formatMemoriesSimple(relevantMemories),\n      };\n    }\n    const mostRecentMemoriesToken = inputs[this.mostRecentMemoriesTokenKey];\n    if (mostRecentMemoriesToken !== undefined) {\n      return {\n        [this.mostRecentMemoriesKey]: await this.getMemoriesUntilLimit(\n          mostRecentMemoriesToken\n        ),\n      };\n    }\n    return {};\n  }\n\n  /**\n   * Method that saves the context of a model run to memory.\n   * @param _inputs The inputs of the model run.\n   * @param outputs The outputs of the model run.\n   * @returns Nothing.\n   */\n  async saveContext(\n    _inputs: InputValues,\n    outputs: OutputValues\n  ): Promise<void> {\n    // save the context of this model run to memory\n    const mem = outputs[this.addMemoryKey];\n    const now = outputs[this.nowKey];\n    if (mem) {\n      await this.addMemory(mem, now, {});\n    }\n  }\n\n  /**\n   * Method that clears the memory contents.\n   * @returns Nothing.\n   */\n  clear(): void {\n    // TODO: clear memory contents\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;AA2BA,IAAM,6BAAN,MAAM,mCAAmCA,uBAAU;CACjD,OAAO,UAAU;AACf,SAAO;;CAGT,aAAa;CAEb;CAEA,mBAAmB;CAEnB;CAEA;CAEA,UAAU;CAEV,AAAQ,sBAAsB;CAE9B,YACE,KACA,iBACA,QACA;AACA,SAAO;AACP,OAAK,MAAM;AACX,OAAK,kBAAkB;AACvB,OAAK,sBAAsB,OAAO;AAClC,OAAK,mBAAmB,OAAO,oBAAoB,KAAK;AACxD,OAAK,UAAU,OAAO,WAAW,KAAK;;CAGxC,aAAqB;AACnB,SAAO;;CAGT,IAAI,YAAsB;AACxB,SAAO;GAAC;GAAkB;GAAO;GAAkB;;CAGrD,IAAI,aAAuB;AACzB,SAAO,CAAC,SAAS;;;;;;;CAQnB,MAAM,QAAkC;AAOtC,SANc,IAAIC,2BAAS;GACzB,KAAK,KAAK;GACV;GACA,SAAS,KAAK;GACd,WAAW;GACZ,CAAC;;CAIJ,MAAM,MAAM,QAAqB,YAAyC;EACxE,MAAM,EAAE,gBAAgB,eAAe,QAAQ;EAE/C,MAAM,kBAAkB,MAAM,KAAK,sBACjC,eACA,WACD;AACD,OAAK,uBAAuB;EAC5B,MAAM,WAAW,IAAIC,mCAAS;GAC5B,aAAa;GACb,UAAU;IACR,YAAY;IACZ,GAAG,OAAO;IACX;GACF,CAAC;AACF,QAAM,KAAK,gBAAgB,aAAa,CAAC,SAAS,CAAC;AAInD,MACE,KAAK,wBAAwB,UAC7B,KAAK,sBAAsB,KAAK,uBAChC,CAAC,KAAK,YACN;AACA,WAAQ,IAAI,oCAAoC;AAChD,QAAK,aAAa;AAClB,SAAM,KAAK,eAAe,KAAK,WAAW;AAC1C,QAAK,sBAAsB;AAC3B,QAAK,aAAa;;AAEpB,SAAO,EAAE,QAAQ,iBAAiB;;;;;;;;;CAUpC,MAAM,eACJ,KACA,YACmB;AACnB,MAAI,KAAK,QACP,SAAQ,IAAI,wBAAwB;EAEtC,MAAM,cAAwB,EAAE;EAChC,MAAM,SAAS,MAAM,KAAK,sBAAsB,IAAI,WAAW;AAC/D,OAAK,MAAM,SAAS,QAAQ;GAC1B,MAAM,WAAW,MAAM,KAAK,mBAAmB,OAAO,KAAK,WAAW;AACtE,QAAK,MAAM,WAAW,SAEpB,OAAM,KAAK,KACT;IACE,gBAAgB;IAChB;IACA,iBAAiB,EACf,QAAQ,sBACT;IACF,EACD,YAAY,SAAS,4BAA4B,CAClD;AAEH,eAAY,KAAK,GAAG,SAAS;;AAE/B,SAAO;;;;;;;;CAST,MAAM,sBACJ,eACA,YACiB;EAEjB,MAAM,SAASC,uCAAe,aAC5B,4RAOD;EAMD,MAAM,iBALQ,MAAM,KAAK,MAAM,OAAO,CAAC,IACrC,eACA,YAAY,SAAS,uBAAuB,CAC7C,EAE2B,MAAM;AAElC,MAAI,KAAK,QACP,SAAQ,IAAI,qBAAqB,cAAc;EAEjD,MAAM,QAAQ,cAAc,MAAM,YAAY;AAC9C,MAAI,MAGF,QAFuB,WAAW,MAAM,GAAG,GACV,KAAM,KAAK;MAG5C,QAAO;;;;;;;;;CAWX,MAAM,sBACJ,OACA,YACmB;EACnB,MAAM,SAASA,uCAAe,aAC5B,oMAID;EAGD,MAAM,iBADe,KAAK,gBAAgB,iBAAiB,CAAC,MAAM,CAAC,MAAM,CAEtE,KAAK,MAA+B,EAAE,YAAY,CAClD,KAAK,KAAK;EACb,MAAM,SAAS,MAAM,KAAK,MAAM,OAAO,CAAC,IACtC,gBACA,YAAY,SAAS,oBAAoB,CAC1C;AACD,SAAO,2BAA2B,UAAU,OAAO;;;;;;;;;;CAWrD,MAAM,mBACJ,OACA,KACA,YACmB;EAEnB,MAAM,SAASA,uCAAe,aAC5B,uKAID;EAGD,MAAM,qBADkB,MAAM,KAAK,cAAc,OAAO,KAAK,WAAW,EAErE,KAAK,QAAQ,UAAU,GAAG,QAAQ,EAAE,IAAI,OAAO,cAAc,CAC7D,KAAK,KAAK;EACb,MAAM,SAAS,MAAM,KAAK,MAAM,OAAO,CAAC,KACtC;GACE;GACA,oBAAoB;GACrB,EACD,YAAY,SAAS,sBAAsB,CAC5C;AACD,SAAO,2BAA2B,UAAU,OAAO,OAAO;;;;;;;CAQ5D,OAAO,UAAU,MAAwB;AAEvC,SAAO,KAAK,MAAM,KAAK,CAAC,KAAK,MAAM,EAAE,MAAM,CAAC;;;;;;;;;CAW9C,MAAM,cACJ,aACA,MACA,YACqB;AACrB,SAAO,KAAK,gBAAgB,OAC1B,aACA,YAAY,SAAS,mBAAmB,CACzC;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA6BL,IAAa,wBAAb,cAA2CC,kCAAW;CACpD;CAEA;CAEA;CAEA;CAEA,AAAQ,iBAAiB;CAEzB,aAAa;CAEb,6BAA6B;CAE7B,eAAe;CAEf,sBAAsB;CAEtB,4BAA4B;CAE5B,wBAAwB;CAExB,SAAS;CAET;CAEA,YACE,KACA,iBACA,QACA;AACA,SAAO;AACP,OAAK,MAAM;AACX,OAAK,kBAAkB;AACvB,OAAK,UAAU,QAAQ,WAAW,KAAK;AACvC,OAAK,sBACH,QAAQ,uBAAuB,KAAK;AACtC,OAAK,iBAAiB,QAAQ,kBAAkB,KAAK;AACrD,OAAK,cAAc,IAAI,2BAA2B,KAAK,iBAAiB;GACtE,qBAAqB,QAAQ;GAC7B,kBAAkB,QAAQ;GAC3B,CAAC;;;;;;CAOJ,yBAAiC;AAC/B,SAAO,KAAK;;;;;;CAOd,gCAAwC;AACtC,SAAO,KAAK;;;;;;CAOd,kBAA0B;AACxB,SAAO,KAAK;;;;;;CAOd,oBAA4B;AAC1B,SAAO,KAAK;;CAGd,IAAI,aAAuB;AAEzB,SAAO,CAAC,KAAK,qBAAqB,KAAK,sBAAsB;;;;;;;;;;CAW/D,MAAM,UACJ,eACA,KACA,UACA,WACA;AACA,SAAO,KAAK,YAAY,KACtB;GAAE,gBAAgB;GAAe;GAAK,iBAAiB;GAAU,EACjE,UACD;;;;;;;CAQH,qBAAqB,kBAAsC;AACzD,MAAI,CAAC,iBAAiB,OACpB,QAAO;EAET,MAAM,iCAAiB,IAAI,KAAK;EAChC,MAAM,UAAU,EAAE;AAClB,OAAK,MAAM,UAAU,kBAAkB;AACrC,OAAI,OAAO,eAAe,eACxB;AAEF,kBAAe,IAAI,OAAO,YAAY;GACtC,MAAM,cAAc,OAAO,SAAS,WAAW,eAAe,SAAS;IACrE,OAAO;IACP,KAAK;IACL,MAAM;IACN,MAAM;IACN,QAAQ;IACR,QAAQ;IACT,CAAC;AACF,WAAQ,KAAK,GAAG,YAAY,IAAI,OAAO,YAAY,MAAM,GAAG;;AAG9D,SADsB,QAAQ,KAAK,QAAQ,GAAG,MAAM,CAAC,KAAK,KAAK;;;;;;;CASjE,qBAAqB,kBAAsC;AAIzD,SAHsB,iBACnB,KAAK,QAAQ,GAAG,IAAI,cAAc,CAClC,KAAK,KAAK;;;;;;;CASf,MAAM,sBAAsB,gBAAyC;EAEnE,MAAM,SAAS,EAAE;AACjB,OAAK,MAAM,OAAO,KAAK,gBACpB,iBAAiB,CACjB,OAAO,CACP,SAAS,EAAE;AACZ,OAAI,kBAAkB,KAAK,gBAAgB;AACzC,QAAI,KAAK,QACP,SAAQ,IAAI,mDAAmD;AAEjE;;AAGF,qBAAkB,MAAM,KAAK,IAAI,aAAa,IAAI,YAAY;AAC9D,OAAI,iBAAiB,KAAK,eACxB,QAAO,KAAK,IAAI;;AAGpB,SAAO,KAAK,qBAAqB,OAAO;;CAG1C,IAAI,kBAA4B;AAE9B,SAAO,EAAE;;;;;;;CAQX,MAAM,oBACJ,QACiC;EACjC,MAAM,UAAU,OAAO,KAAK;EAC5B,MAAM,MAAM,OAAO,KAAK;AACxB,MAAI,YAAY,QAAW;GACzB,MAAM,oBACJ,MAAM,QAAQ,IACZ,QAAQ,KAAK,UACX,KAAK,YAAY,cAAc,OAAO,IAAI,CAC3C,CACF,EACD,MAAM;AACR,UAAO;KACJ,KAAK,sBAAsB,KAAK,qBAAqB,iBAAiB;KACtE,KAAK,4BACJ,KAAK,qBAAqB,iBAAiB;IAC9C;;EAEH,MAAM,0BAA0B,OAAO,KAAK;AAC5C,MAAI,4BAA4B,OAC9B,QAAO,GACJ,KAAK,wBAAwB,MAAM,KAAK,sBACvC,wBACD,EACF;AAEH,SAAO,EAAE;;;;;;;;CASX,MAAM,YACJ,SACA,SACe;EAEf,MAAM,MAAM,QAAQ,KAAK;EACzB,MAAM,MAAM,QAAQ,KAAK;AACzB,MAAI,IACF,OAAM,KAAK,UAAU,KAAK,KAAK,EAAE,CAAC;;;;;;CAQtC,QAAc"}

package/dist/experimental/generative_agents/generative_agent_memory.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"generative_agent_memory.d.cts","names":["BaseLanguageModelInterface","PromptTemplate","Document","ChainValues","BaseMemory","InputValues","OutputValues","CallbackManagerForChainRun","Callbacks","TimeWeightedVectorStoreRetriever","BaseChain","LLMChain","GenerativeAgentMemoryConfig","GenerativeAgentMemoryChain","Omit","Promise","Date","GenerativeAgentMemory","Record"],"sources":["../../../src/experimental/generative_agents/generative_agent_memory.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { Document } from \"@langchain/core/documents\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BaseMemory, InputValues, OutputValues } from \"@langchain/core/memory\";\nimport { CallbackManagerForChainRun, Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { TimeWeightedVectorStoreRetriever } from \"../../retrievers/time_weighted.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nexport type GenerativeAgentMemoryConfig = {\n    reflectionThreshold?: number;\n    importanceWeight?: number;\n    verbose?: boolean;\n    maxTokensLimit?: number;\n};\n/**\n * Class that manages the memory of a generative agent in LangChain. It\n * extends the `BaseChain` class and has methods for adding observations\n * or memories to the agent's memory, scoring the importance of a memory,\n * reflecting on recent events to add synthesized memories, and generating\n * insights on a topic of reflection based on pertinent memories.\n */\ndeclare class GenerativeAgentMemoryChain extends BaseChain {\n    static lc_name(): string;\n    reflecting: boolean;\n    reflectionThreshold?: number;\n    importanceWeight: number;\n    memoryRetriever: TimeWeightedVectorStoreRetriever;\n    llm: BaseLanguageModelInterface;\n    verbose: boolean;\n    private aggregateImportance;\n    constructor(llm: BaseLanguageModelInterface, memoryRetriever: TimeWeightedVectorStoreRetriever, config: Omit<GenerativeAgentMemoryConfig, \"maxTokensLimit\">);\n    _chainType(): string;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    /**\n     * Method that creates a new LLMChain with the given prompt.\n     * @param prompt The PromptTemplate to use for the new LLMChain.\n     * @returns A new LLMChain instance.\n     */\n    chain(prompt: PromptTemplate): LLMChain;\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<{\n        output: number;\n    }>;\n    /**\n     * Method that pauses the agent to reflect on recent events and generate\n     * new insights.\n     * @param now The current date.\n     * @param runManager The CallbackManagerForChainRun to use for the reflection.\n     * @returns An array of new insights as strings.\n     */\n    pauseToReflect(now?: Date, runManager?: CallbackManagerForChainRun): Promise<string[]>;\n    /**\n     * Method that scores the importance of a given memory.\n     * @param memoryContent The content of the memory to score.\n     * @param runManager The CallbackManagerForChainRun to use for scoring.\n     * @returns The importance score of the memory as a number.\n     */\n    scoreMemoryImportance(memoryContent: string, runManager?: CallbackManagerForChainRun): Promise<number>;\n    /**\n     * Method that retrieves the topics of reflection based on the last K\n     * memories.\n     * @param lastK The number of most recent memories to consider for generating topics.\n     * @param runManager The CallbackManagerForChainRun to use for retrieving topics.\n     * @returns An array of topics of reflection as strings.\n     */\n    getTopicsOfReflection(lastK: number, runManager?: CallbackManagerForChainRun): Promise<string[]>;\n    /**\n     * Method that generates insights on a given topic of reflection based on\n     * pertinent memories.\n     * @param topic The topic of reflection.\n     * @param now The current date.\n     * @param runManager The CallbackManagerForChainRun to use for generating insights.\n     * @returns An array of insights as strings.\n     */\n    getInsightsOnTopic(topic: string, now?: Date, runManager?: CallbackManagerForChainRun): Promise<string[]>;\n    /**\n     * Method that parses a newline-separated string into a list of strings.\n     * @param text The newline-separated string to parse.\n     * @returns An array of strings.\n     */\n    static parseList(text: string): string[];\n    /**\n     * Method that fetches memories related to a given observation.\n     * @param observation The observation to fetch memories for.\n     * @param _now The current date.\n     * @param runManager The CallbackManagerForChainRun to use for fetching memories.\n     * @returns An array of Document instances representing the fetched memories.\n     */\n    fetchMemories(observation: string, _now?: Date, runManager?: CallbackManagerForChainRun): Promise<Document[]>;\n}\n/**\n * Class that manages the memory of a generative agent in LangChain. It\n * extends the `BaseMemory` class and has methods for adding a memory,\n * formatting memories, getting memories until a token limit is reached,\n * loading memory variables, saving the context of a model run to memory,\n * and clearing memory contents.\n * @example\n * ```typescript\n * const createNewMemoryRetriever = async () => {\n *   const vectorStore = new MemoryVectorStore(new OpenAIEmbeddings());\n *   const retriever = new TimeWeightedVectorStoreRetriever({\n *     vectorStore,\n *     otherScoreKeys: [\"importance\"],\n *     k: 15,\n *   });\n *   return retriever;\n * };\n * const tommiesMemory = new GenerativeAgentMemory(\n *   llm,\n *   await createNewMemoryRetriever(),\n *   { reflectionThreshold: 8 },\n * );\n * const summary = await tommiesMemory.getSummary();\n * ```\n */\nexport declare class GenerativeAgentMemory extends BaseMemory {\n    llm: BaseLanguageModelInterface;\n    memoryRetriever: TimeWeightedVectorStoreRetriever;\n    verbose: boolean;\n    reflectionThreshold?: number;\n    private maxTokensLimit;\n    queriesKey: string;\n    mostRecentMemoriesTokenKey: string;\n    addMemoryKey: string;\n    relevantMemoriesKey: string;\n    relevantMemoriesSimpleKey: string;\n    mostRecentMemoriesKey: string;\n    nowKey: string;\n    memoryChain: GenerativeAgentMemoryChain;\n    constructor(llm: BaseLanguageModelInterface, memoryRetriever: TimeWeightedVectorStoreRetriever, config?: GenerativeAgentMemoryConfig);\n    /**\n     * Method that returns the key for relevant memories.\n     * @returns The key for relevant memories as a string.\n     */\n    getRelevantMemoriesKey(): string;\n    /**\n     * Method that returns the key for the most recent memories token.\n     * @returns The key for the most recent memories token as a string.\n     */\n    getMostRecentMemoriesTokenKey(): string;\n    /**\n     * Method that returns the key for adding a memory.\n     * @returns The key for adding a memory as a string.\n     */\n    getAddMemoryKey(): string;\n    /**\n     * Method that returns the key for the current time.\n     * @returns The key for the current time as a string.\n     */\n    getCurrentTimeKey(): string;\n    get memoryKeys(): string[];\n    /**\n     * Method that adds a memory to the agent's memory.\n     * @param memoryContent The content of the memory to add.\n     * @param now The current date.\n     * @param metadata The metadata for the memory.\n     * @param callbacks The Callbacks to use for adding the memory.\n     * @returns The result of the memory addition.\n     */\n    addMemory(memoryContent: string, now?: Date, metadata?: Record<string, unknown>, callbacks?: Callbacks): Promise<ChainValues>;\n    /**\n     * Method that formats the given relevant memories in detail.\n     * @param relevantMemories The relevant memories to format.\n     * @returns The formatted memories as a string.\n     */\n    formatMemoriesDetail(relevantMemories: Document[]): string;\n    /**\n     * Method that formats the given relevant memories in a simple manner.\n     * @param relevantMemories The relevant memories to format.\n     * @returns The formatted memories as a string.\n     */\n    formatMemoriesSimple(relevantMemories: Document[]): string;\n    /**\n     * Method that retrieves memories until a token limit is reached.\n     * @param consumedTokens The number of tokens consumed so far.\n     * @returns The memories as a string.\n     */\n    getMemoriesUntilLimit(consumedTokens: number): Promise<string>;\n    get memoryVariables(): string[];\n    /**\n     * Method that loads memory variables based on the given inputs.\n     * @param inputs The inputs to use for loading memory variables.\n     * @returns An object containing the loaded memory variables.\n     */\n    loadMemoryVariables(inputs: InputValues): Promise<Record<string, string>>;\n    /**\n     * Method that saves the context of a model run to memory.\n     * @param _inputs The inputs of the model run.\n     * @param outputs The outputs of the model run.\n     * @returns Nothing.\n     */\n    saveContext(_inputs: InputValues, outputs: OutputValues): Promise<void>;\n    /**\n     * Method that clears the memory contents.\n     * @returns Nothing.\n     */\n    clear(): void;\n}\nexport {};\n//# sourceMappingURL=generative_agent_memory.d.ts.map"],"mappings":";;;;;;;;;;;KASYY,2BAAAA;;EAAAA,gBAAAA,CAAAA,EAAAA,MAAAA;EAaEC,OAAAA,CAAAA,EAAAA,OAAAA;EAKOJ,cAAAA,CAAAA,EAAAA,MAAAA;CACZT;;;;;;;;cANKa,0BAAAA,SAAmCH,SAAAA,CAmBLH;EAA6BQ,OAAAA,OAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAUhDC,UAAAA,EAAAA,OAAAA;EAAmBT,mBAAAA,CAAAA,EAAAA,MAAAA;EAA6BQ,gBAAAA,EAAAA,MAAAA;EAOXR,eAAAA,EA/BzCE,gCA+ByCF;EAA6BQ,GAAAA,EA9BlFf,0BA8BkFe;EAQrCR,OAAAA,EAAAA,OAAAA;EAA6BQ,QAAAA,mBAAAA;EASvCC,WAAAA,CAAAA,GAAAA,EA5CvBhB,0BA4CuBgB,EAAAA,eAAAA,EA5CsBP,gCA4CtBO,EAAAA,MAAAA,EA5CgEF,IA4ChEE,CA5CqEJ,2BA4CrEI,EAAAA,gBAAAA,CAAAA;EAAmBT,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAA6BQ,IAAAA,SAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAc9CC,IAAAA,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAmBT;;;;AAnEP;EA8FrCU,KAAAA,CAAAA,MAAAA,EA5EHhB,cA4EwB,CAAA,EA5EPU,QA4EO;EACjCX,KAAAA,CAAAA,MAAAA,EA5ESG,WA4ETH,EAAAA,UAAAA,CAAAA,EA5EmCO,0BA4EnCP,CAAAA,EA5EgEe,OA4EhEf,CAAAA;IACYS,MAAAA,EAAAA,MAAAA;EAWJI,CAAAA,CAAAA;EACIb;;;;;;;EA8BwFe,cAAAA,CAAAA,GAAAA,CAAAA,EA7GpFC,IA6GoFD,EAAAA,UAAAA,CAAAA,EA7GjER,0BA6GiEQ,CAAAA,EA7GpCA,OA6GoCA,CAAAA,MAAAA,EAAAA,CAAAA;EAMlEb;;;;;;EA0BlBG,qBAAAA,CAAAA,aAAAA,EAAAA,MAAAA,EAAAA,UAAAA,CAAAA,EAtIqCE,0BAsIrCF,CAAAA,EAtIkEU,OAsIlEV,CAAAA,MAAAA,CAAAA;EAAsBC;;;AA5Ec;;;;oDAlDPC,6BAA6BQ;;;;;;;;;0CASvCC,mBAAmBT,6BAA6BQ;;;;;;;;;;;;;;4CAc9CC,mBAAmBT,6BAA6BQ,QAAQb;;;;;;;;;;;;;;;;;;;;;;;;;;;cA2BjFe,qBAAAA,SAA8Bb,UAAAA;OAC1CJ;mBACYS;;;;;;;;;;;eAWJI;mBACIb,6CAA6CS,2CAA2CG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yCA8BlEI,iBAAiBE,qCAAqCV,YAAYO,QAAQZ;;;;;;yCAM1ED;;;;;;yCAMAA;;;;;;iDAMQa;;;;;;;8BAOnBV,gBAAcU,QAAQG;;;;;;;uBAO7Bb,wBAAsBC,eAAeS"}
+{"version":3,"file":"generative_agent_memory.d.cts","names":[],"sources":["../../../src/experimental/generative_agents/generative_agent_memory.ts"],"mappings":";;;;;;;;;;;KAaY,2BAAA;EACV,mBAAA;EACA,gBAAA;EACA,OAAA;EACA,cAAA;AAAA;;;;;;;AACA;cASI,0BAAA,SAAmC,SAAA;EAAA,OAChC,OAAA,CAAA;EAIP,UAAA;EAEA,mBAAA;EAEA,gBAAA;EAEA,eAAA,EAAiB,gCAAA;EAEjB,GAAA,EAAK,0BAAA;EAEL,OAAA;EAAA,QAEQ,mBAAA;EAER,WAAA,CACE,GAAA,EAAK,0BAAA,EACL,eAAA,EAAiB,gCAAA,EACjB,MAAA,EAAQ,IAAA,CAAK,2BAAA;EAUf,UAAA,CAAA;EAAA,IAII,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EA4DI;;;;;EAnDR,KAAA,CAAM,MAAA,EAAQ,cAAA,GAAiB,QAAA;EAUzB,KAAA,CAAM,MAAA,EAAQ,WAAA,EAAa,UAAA,GAAa,0BAAA,GAA0B,OAAA;;;EAsJrE;;;;;;;EA9GG,cAAA,CACJ,GAAA,GAAM,IAAA,EACN,UAAA,GAAa,0BAAA,GACZ,OAAA;EAtGoC;;;;;;EAsIjC,qBAAA,CACJ,aAAA,UACA,UAAA,GAAa,0BAAA,GACZ,OAAA;EA5HH;;;;;;;EAkKM,qBAAA,CACJ,KAAA,UACA,UAAA,GAAa,0BAAA,GACZ,OAAA;EA7JD;;;;;;;;EAwLI,kBAAA,CACJ,KAAA,UACA,GAAA,GAAM,IAAA,EACN,UAAA,GAAa,0BAAA,GACZ,OAAA;EAhKG;;;;;EAAA,OA4LC,SAAA,CAAU,IAAA;EAlLgB;;;;;;;EA+L3B,aAAA,CACJ,WAAA,UACA,IAAA,GAAO,IAAA,EACP,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,QAAA;AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;cAiCA,qBAAA,SAA8B,UAAA;EACzC,GAAA,EAAK,0BAAA;EAEL,eAAA,EAAiB,gCAAA;EAEjB,OAAA;EAEA,mBAAA;EAAA,QAEQ,cAAA;EAER,UAAA;EAEA,0BAAA;EAEA,YAAA;EAEA,mBAAA;EAEA,yBAAA;EAEA,qBAAA;EAEA,MAAA;EAEA,WAAA,EAAa,0BAAA;EAEb,WAAA,CACE,GAAA,EAAK,0BAAA,EACL,eAAA,EAAiB,gCAAA,EACjB,MAAA,GAAS,2BAAA;EAgEG;;;;EA7Cd,sBAAA,CAAA;EAoGqD;;;;EA5FrD,6BAAA,CAAA;EAkKW;;;;EA1JX,eAAA,CAAA;EAjEyC;;;;EAyEzC,iBAAA,CAAA;EAAA,IAII,UAAA,CAAA;EAtEJ;;;;;;;;EAmFM,SAAA,CACJ,aAAA,UACA,GAAA,GAAM,IAAA,EACN,QAAA,GAAW,MAAA,mBACX,SAAA,GAAY,SAAA,GAAS,OAAA,CAAA,WAAA;EArEvB;;;;;EAkFA,oBAAA,CAAqB,gBAAA,EAAkB,QAAA;EA9ErC;;;;;EA4GF,oBAAA,CAAqB,gBAAA,EAAkB,QAAA;EAhEvC;;;;;EA4EM,qBAAA,CAAsB,cAAA,WAAyB,OAAA;EAAA,IAsBjD,eAAA,CAAA;EA9EF;;;;;EAwFI,mBAAA,CACJ,MAAA,EAAQ,aAAA,GACP,OAAA,CAAQ,MAAA;EA5E4B;;;;;;EA8GjC,WAAA,CACJ,OAAA,EAAS,aAAA,EACT,OAAA,EAAS,YAAA,GACR,OAAA;EAvEkD;;;;EAoFrD,KAAA,CAAA;AAAA"}

package/dist/experimental/generative_agents/generative_agent_memory.d.ts

@@ -1,12 +1,12 @@
 import { BaseChain } from "../../chains/base.js";
 import { LLMChain } from "../../chains/llm_chain.js";
 import { TimeWeightedVectorStoreRetriever } from "../../retrievers/time_weighted.js";
-import { Document } from "@langchain/core/documents";
 import { PromptTemplate } from "@langchain/core/prompts";
 import { BaseLanguageModelInterface } from "@langchain/core/language_models/base";
 import { CallbackManagerForChainRun, Callbacks } from "@langchain/core/callbacks/manager";
 import { BaseMemory, InputValues, OutputValues } from "@langchain/core/memory";
 import { ChainValues } from "@langchain/core/utils/types";
+import { Document } from "@langchain/core/documents";
 
 //#region src/experimental/generative_agents/generative_agent_memory.d.ts
 type GenerativeAgentMemoryConfig = {

package/dist/experimental/generative_agents/generative_agent_memory.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"generative_agent_memory.d.ts","names":["BaseLanguageModelInterface","PromptTemplate","Document","ChainValues","BaseMemory","InputValues","OutputValues","CallbackManagerForChainRun","Callbacks","TimeWeightedVectorStoreRetriever","BaseChain","LLMChain","GenerativeAgentMemoryConfig","GenerativeAgentMemoryChain","Omit","Promise","Date","GenerativeAgentMemory","Record"],"sources":["../../../src/experimental/generative_agents/generative_agent_memory.d.ts"],"sourcesContent":["import type { BaseLanguageModelInterface } from \"@langchain/core/language_models/base\";\nimport { PromptTemplate } from \"@langchain/core/prompts\";\nimport { Document } from \"@langchain/core/documents\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\nimport { BaseMemory, InputValues, OutputValues } from \"@langchain/core/memory\";\nimport { CallbackManagerForChainRun, Callbacks } from \"@langchain/core/callbacks/manager\";\nimport { TimeWeightedVectorStoreRetriever } from \"../../retrievers/time_weighted.js\";\nimport { BaseChain } from \"../../chains/base.js\";\nimport { LLMChain } from \"../../chains/llm_chain.js\";\nexport type GenerativeAgentMemoryConfig = {\n    reflectionThreshold?: number;\n    importanceWeight?: number;\n    verbose?: boolean;\n    maxTokensLimit?: number;\n};\n/**\n * Class that manages the memory of a generative agent in LangChain. It\n * extends the `BaseChain` class and has methods for adding observations\n * or memories to the agent's memory, scoring the importance of a memory,\n * reflecting on recent events to add synthesized memories, and generating\n * insights on a topic of reflection based on pertinent memories.\n */\ndeclare class GenerativeAgentMemoryChain extends BaseChain {\n    static lc_name(): string;\n    reflecting: boolean;\n    reflectionThreshold?: number;\n    importanceWeight: number;\n    memoryRetriever: TimeWeightedVectorStoreRetriever;\n    llm: BaseLanguageModelInterface;\n    verbose: boolean;\n    private aggregateImportance;\n    constructor(llm: BaseLanguageModelInterface, memoryRetriever: TimeWeightedVectorStoreRetriever, config: Omit<GenerativeAgentMemoryConfig, \"maxTokensLimit\">);\n    _chainType(): string;\n    get inputKeys(): string[];\n    get outputKeys(): string[];\n    /**\n     * Method that creates a new LLMChain with the given prompt.\n     * @param prompt The PromptTemplate to use for the new LLMChain.\n     * @returns A new LLMChain instance.\n     */\n    chain(prompt: PromptTemplate): LLMChain;\n    _call(values: ChainValues, runManager?: CallbackManagerForChainRun): Promise<{\n        output: number;\n    }>;\n    /**\n     * Method that pauses the agent to reflect on recent events and generate\n     * new insights.\n     * @param now The current date.\n     * @param runManager The CallbackManagerForChainRun to use for the reflection.\n     * @returns An array of new insights as strings.\n     */\n    pauseToReflect(now?: Date, runManager?: CallbackManagerForChainRun): Promise<string[]>;\n    /**\n     * Method that scores the importance of a given memory.\n     * @param memoryContent The content of the memory to score.\n     * @param runManager The CallbackManagerForChainRun to use for scoring.\n     * @returns The importance score of the memory as a number.\n     */\n    scoreMemoryImportance(memoryContent: string, runManager?: CallbackManagerForChainRun): Promise<number>;\n    /**\n     * Method that retrieves the topics of reflection based on the last K\n     * memories.\n     * @param lastK The number of most recent memories to consider for generating topics.\n     * @param runManager The CallbackManagerForChainRun to use for retrieving topics.\n     * @returns An array of topics of reflection as strings.\n     */\n    getTopicsOfReflection(lastK: number, runManager?: CallbackManagerForChainRun): Promise<string[]>;\n    /**\n     * Method that generates insights on a given topic of reflection based on\n     * pertinent memories.\n     * @param topic The topic of reflection.\n     * @param now The current date.\n     * @param runManager The CallbackManagerForChainRun to use for generating insights.\n     * @returns An array of insights as strings.\n     */\n    getInsightsOnTopic(topic: string, now?: Date, runManager?: CallbackManagerForChainRun): Promise<string[]>;\n    /**\n     * Method that parses a newline-separated string into a list of strings.\n     * @param text The newline-separated string to parse.\n     * @returns An array of strings.\n     */\n    static parseList(text: string): string[];\n    /**\n     * Method that fetches memories related to a given observation.\n     * @param observation The observation to fetch memories for.\n     * @param _now The current date.\n     * @param runManager The CallbackManagerForChainRun to use for fetching memories.\n     * @returns An array of Document instances representing the fetched memories.\n     */\n    fetchMemories(observation: string, _now?: Date, runManager?: CallbackManagerForChainRun): Promise<Document[]>;\n}\n/**\n * Class that manages the memory of a generative agent in LangChain. It\n * extends the `BaseMemory` class and has methods for adding a memory,\n * formatting memories, getting memories until a token limit is reached,\n * loading memory variables, saving the context of a model run to memory,\n * and clearing memory contents.\n * @example\n * ```typescript\n * const createNewMemoryRetriever = async () => {\n *   const vectorStore = new MemoryVectorStore(new OpenAIEmbeddings());\n *   const retriever = new TimeWeightedVectorStoreRetriever({\n *     vectorStore,\n *     otherScoreKeys: [\"importance\"],\n *     k: 15,\n *   });\n *   return retriever;\n * };\n * const tommiesMemory = new GenerativeAgentMemory(\n *   llm,\n *   await createNewMemoryRetriever(),\n *   { reflectionThreshold: 8 },\n * );\n * const summary = await tommiesMemory.getSummary();\n * ```\n */\nexport declare class GenerativeAgentMemory extends BaseMemory {\n    llm: BaseLanguageModelInterface;\n    memoryRetriever: TimeWeightedVectorStoreRetriever;\n    verbose: boolean;\n    reflectionThreshold?: number;\n    private maxTokensLimit;\n    queriesKey: string;\n    mostRecentMemoriesTokenKey: string;\n    addMemoryKey: string;\n    relevantMemoriesKey: string;\n    relevantMemoriesSimpleKey: string;\n    mostRecentMemoriesKey: string;\n    nowKey: string;\n    memoryChain: GenerativeAgentMemoryChain;\n    constructor(llm: BaseLanguageModelInterface, memoryRetriever: TimeWeightedVectorStoreRetriever, config?: GenerativeAgentMemoryConfig);\n    /**\n     * Method that returns the key for relevant memories.\n     * @returns The key for relevant memories as a string.\n     */\n    getRelevantMemoriesKey(): string;\n    /**\n     * Method that returns the key for the most recent memories token.\n     * @returns The key for the most recent memories token as a string.\n     */\n    getMostRecentMemoriesTokenKey(): string;\n    /**\n     * Method that returns the key for adding a memory.\n     * @returns The key for adding a memory as a string.\n     */\n    getAddMemoryKey(): string;\n    /**\n     * Method that returns the key for the current time.\n     * @returns The key for the current time as a string.\n     */\n    getCurrentTimeKey(): string;\n    get memoryKeys(): string[];\n    /**\n     * Method that adds a memory to the agent's memory.\n     * @param memoryContent The content of the memory to add.\n     * @param now The current date.\n     * @param metadata The metadata for the memory.\n     * @param callbacks The Callbacks to use for adding the memory.\n     * @returns The result of the memory addition.\n     */\n    addMemory(memoryContent: string, now?: Date, metadata?: Record<string, unknown>, callbacks?: Callbacks): Promise<ChainValues>;\n    /**\n     * Method that formats the given relevant memories in detail.\n     * @param relevantMemories The relevant memories to format.\n     * @returns The formatted memories as a string.\n     */\n    formatMemoriesDetail(relevantMemories: Document[]): string;\n    /**\n     * Method that formats the given relevant memories in a simple manner.\n     * @param relevantMemories The relevant memories to format.\n     * @returns The formatted memories as a string.\n     */\n    formatMemoriesSimple(relevantMemories: Document[]): string;\n    /**\n     * Method that retrieves memories until a token limit is reached.\n     * @param consumedTokens The number of tokens consumed so far.\n     * @returns The memories as a string.\n     */\n    getMemoriesUntilLimit(consumedTokens: number): Promise<string>;\n    get memoryVariables(): string[];\n    /**\n     * Method that loads memory variables based on the given inputs.\n     * @param inputs The inputs to use for loading memory variables.\n     * @returns An object containing the loaded memory variables.\n     */\n    loadMemoryVariables(inputs: InputValues): Promise<Record<string, string>>;\n    /**\n     * Method that saves the context of a model run to memory.\n     * @param _inputs The inputs of the model run.\n     * @param outputs The outputs of the model run.\n     * @returns Nothing.\n     */\n    saveContext(_inputs: InputValues, outputs: OutputValues): Promise<void>;\n    /**\n     * Method that clears the memory contents.\n     * @returns Nothing.\n     */\n    clear(): void;\n}\nexport {};\n//# sourceMappingURL=generative_agent_memory.d.ts.map"],"mappings":";;;;;;;;;;;KASYY,2BAAAA;;EAAAA,gBAAAA,CAAAA,EAAAA,MAAAA;EAaEC,OAAAA,CAAAA,EAAAA,OAAAA;EAKOJ,cAAAA,CAAAA,EAAAA,MAAAA;CACZT;;;;;;;;cANKa,0BAAAA,SAAmCH,SAAAA,CAmBLH;EAA6BQ,OAAAA,OAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAUhDC,UAAAA,EAAAA,OAAAA;EAAmBT,mBAAAA,CAAAA,EAAAA,MAAAA;EAA6BQ,gBAAAA,EAAAA,MAAAA;EAOXR,eAAAA,EA/BzCE,gCA+ByCF;EAA6BQ,GAAAA,EA9BlFf,0BA8BkFe;EAQrCR,OAAAA,EAAAA,OAAAA;EAA6BQ,QAAAA,mBAAAA;EASvCC,WAAAA,CAAAA,GAAAA,EA5CvBhB,0BA4CuBgB,EAAAA,eAAAA,EA5CsBP,gCA4CtBO,EAAAA,MAAAA,EA5CgEF,IA4ChEE,CA5CqEJ,2BA4CrEI,EAAAA,gBAAAA,CAAAA;EAAmBT,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAA6BQ,IAAAA,SAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAc9CC,IAAAA,UAAAA,CAAAA,CAAAA,EAAAA,MAAAA,EAAAA;EAAmBT;;;;AAnEP;EA8FrCU,KAAAA,CAAAA,MAAAA,EA5EHhB,cA4EwB,CAAA,EA5EPU,QA4EO;EACjCX,KAAAA,CAAAA,MAAAA,EA5ESG,WA4ETH,EAAAA,UAAAA,CAAAA,EA5EmCO,0BA4EnCP,CAAAA,EA5EgEe,OA4EhEf,CAAAA;IACYS,MAAAA,EAAAA,MAAAA;EAWJI,CAAAA,CAAAA;EACIb;;;;;;;EA8BwFe,cAAAA,CAAAA,GAAAA,CAAAA,EA7GpFC,IA6GoFD,EAAAA,UAAAA,CAAAA,EA7GjER,0BA6GiEQ,CAAAA,EA7GpCA,OA6GoCA,CAAAA,MAAAA,EAAAA,CAAAA;EAMlEb;;;;;;EA0BlBG,qBAAAA,CAAAA,aAAAA,EAAAA,MAAAA,EAAAA,UAAAA,CAAAA,EAtIqCE,0BAsIrCF,CAAAA,EAtIkEU,OAsIlEV,CAAAA,MAAAA,CAAAA;EAAsBC;;;AA5Ec;;;;oDAlDPC,6BAA6BQ;;;;;;;;;0CASvCC,mBAAmBT,6BAA6BQ;;;;;;;;;;;;;;4CAc9CC,mBAAmBT,6BAA6BQ,QAAQb;;;;;;;;;;;;;;;;;;;;;;;;;;;cA2BjFe,qBAAAA,SAA8Bb,UAAAA;OAC1CJ;mBACYS;;;;;;;;;;;eAWJI;mBACIb,6CAA6CS,2CAA2CG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yCA8BlEI,iBAAiBE,qCAAqCV,YAAYO,QAAQZ;;;;;;yCAM1ED;;;;;;yCAMAA;;;;;;iDAMQa;;;;;;;8BAOnBV,cAAcU,QAAQG;;;;;;;uBAO7Bb,sBAAsBC,eAAeS"}
+{"version":3,"file":"generative_agent_memory.d.ts","names":[],"sources":["../../../src/experimental/generative_agents/generative_agent_memory.ts"],"mappings":";;;;;;;;;;;KAaY,2BAAA;EACV,mBAAA;EACA,gBAAA;EACA,OAAA;EACA,cAAA;AAAA;;;;;;;AACA;cASI,0BAAA,SAAmC,SAAA;EAAA,OAChC,OAAA,CAAA;EAIP,UAAA;EAEA,mBAAA;EAEA,gBAAA;EAEA,eAAA,EAAiB,gCAAA;EAEjB,GAAA,EAAK,0BAAA;EAEL,OAAA;EAAA,QAEQ,mBAAA;EAER,WAAA,CACE,GAAA,EAAK,0BAAA,EACL,eAAA,EAAiB,gCAAA,EACjB,MAAA,EAAQ,IAAA,CAAK,2BAAA;EAUf,UAAA,CAAA;EAAA,IAII,SAAA,CAAA;EAAA,IAIA,UAAA,CAAA;EA4DI;;;;;EAnDR,KAAA,CAAM,MAAA,EAAQ,cAAA,GAAiB,QAAA;EAUzB,KAAA,CAAM,MAAA,EAAQ,WAAA,EAAa,UAAA,GAAa,0BAAA,GAA0B,OAAA;;;EAsJrE;;;;;;;EA9GG,cAAA,CACJ,GAAA,GAAM,IAAA,EACN,UAAA,GAAa,0BAAA,GACZ,OAAA;EAtGoC;;;;;;EAsIjC,qBAAA,CACJ,aAAA,UACA,UAAA,GAAa,0BAAA,GACZ,OAAA;EA5HH;;;;;;;EAkKM,qBAAA,CACJ,KAAA,UACA,UAAA,GAAa,0BAAA,GACZ,OAAA;EA7JD;;;;;;;;EAwLI,kBAAA,CACJ,KAAA,UACA,GAAA,GAAM,IAAA,EACN,UAAA,GAAa,0BAAA,GACZ,OAAA;EAhKG;;;;;EAAA,OA4LC,SAAA,CAAU,IAAA;EAlLgB;;;;;;;EA+L3B,aAAA,CACJ,WAAA,UACA,IAAA,GAAO,IAAA,EACP,UAAA,GAAa,0BAAA,GACZ,OAAA,CAAQ,QAAA;AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;cAiCA,qBAAA,SAA8B,UAAA;EACzC,GAAA,EAAK,0BAAA;EAEL,eAAA,EAAiB,gCAAA;EAEjB,OAAA;EAEA,mBAAA;EAAA,QAEQ,cAAA;EAER,UAAA;EAEA,0BAAA;EAEA,YAAA;EAEA,mBAAA;EAEA,yBAAA;EAEA,qBAAA;EAEA,MAAA;EAEA,WAAA,EAAa,0BAAA;EAEb,WAAA,CACE,GAAA,EAAK,0BAAA,EACL,eAAA,EAAiB,gCAAA,EACjB,MAAA,GAAS,2BAAA;EAgEG;;;;EA7Cd,sBAAA,CAAA;EAoGqD;;;;EA5FrD,6BAAA,CAAA;EAkKW;;;;EA1JX,eAAA,CAAA;EAjEyC;;;;EAyEzC,iBAAA,CAAA;EAAA,IAII,UAAA,CAAA;EAtEJ;;;;;;;;EAmFM,SAAA,CACJ,aAAA,UACA,GAAA,GAAM,IAAA,EACN,QAAA,GAAW,MAAA,mBACX,SAAA,GAAY,SAAA,GAAS,OAAA,CAAA,WAAA;EArEvB;;;;;EAkFA,oBAAA,CAAqB,gBAAA,EAAkB,QAAA;EA9ErC;;;;;EA4GF,oBAAA,CAAqB,gBAAA,EAAkB,QAAA;EAhEvC;;;;;EA4EM,qBAAA,CAAsB,cAAA,WAAyB,OAAA;EAAA,IAsBjD,eAAA,CAAA;EA9EF;;;;;EAwFI,mBAAA,CACJ,MAAA,EAAQ,WAAA,GACP,OAAA,CAAQ,MAAA;EA5E4B;;;;;;EA8GjC,WAAA,CACJ,OAAA,EAAS,WAAA,EACT,OAAA,EAAS,YAAA,GACR,OAAA;EAvEkD;;;;EAoFrD,KAAA,CAAA;AAAA"}

package/dist/experimental/generative_agents/generative_agent_memory.js

@@ -1,8 +1,8 @@
 import { BaseChain } from "../../chains/base.js";
 import { LLMChain } from "../../chains/llm_chain.js";
-import { Document } from "@langchain/core/documents";
 import { PromptTemplate } from "@langchain/core/prompts";
 import { BaseMemory } from "@langchain/core/memory";
+import { Document } from "@langchain/core/documents";
 
 //#region src/experimental/generative_agents/generative_agent_memory.ts
 /**
@@ -50,13 +50,12 @@ var GenerativeAgentMemoryChain = class GenerativeAgentMemoryChain extends BaseCh
 	* @returns A new LLMChain instance.
 	*/
 	chain(prompt) {
-		const chain = new LLMChain({
+		return new LLMChain({
 			llm: this.llm,
 			prompt,
 			verbose: this.verbose,
 			outputKey: "output"
 		});
-		return chain;
 	}
 	async _call(values, runManager) {
 		const { memory_content: memoryContent, now } = values;
@@ -109,15 +108,11 @@ var GenerativeAgentMemoryChain = class GenerativeAgentMemoryChain extends BaseCh
 	*/
 	async scoreMemoryImportance(memoryContent, runManager) {
 		const prompt = PromptTemplate.fromTemplate("On the scale of 1 to 10, where 1 is purely mundane (e.g., brushing teeth, making bed) and 10 is extremely poignant (e.g., a break up, college acceptance), rate the likely poignancy of the following piece of memory. Respond with a single integer.\nMemory: {memory_content}\nRating: ");
-		const score = await this.chain(prompt).run(memoryContent, runManager?.getChild("determine_importance"));
-		const strippedScore = score.trim();
+		const strippedScore = (await this.chain(prompt).run(memoryContent, runManager?.getChild("determine_importance"))).trim();
 		if (this.verbose) console.log("Importance score:", strippedScore);
 		const match = strippedScore.match(/^\D*(\d+)/);
-		if (match) {
-			const capturedNumber = parseFloat(match[1]);
-			const result = capturedNumber / 10 * this.importanceWeight;
-			return result;
-		} else return 0;
+		if (match) return parseFloat(match[1]) / 10 * this.importanceWeight;
+		else return 0;
 	}
 	/**
 	* Method that retrieves the topics of reflection based on the last K
@@ -128,8 +123,7 @@ var GenerativeAgentMemoryChain = class GenerativeAgentMemoryChain extends BaseCh
 	*/
 	async getTopicsOfReflection(lastK, runManager) {
 		const prompt = PromptTemplate.fromTemplate("{observations}\n\nGiven only the information above, what are the 3 most salient high-level questions we can answer about the subjects in the statements? Provide each question on a new line.\n\n");
-		const observations = this.memoryRetriever.getMemoryStream().slice(-lastK);
-		const observationStr = observations.map((o) => o.pageContent).join("\n");
+		const observationStr = this.memoryRetriever.getMemoryStream().slice(-lastK).map((o) => o.pageContent).join("\n");
 		const result = await this.chain(prompt).run(observationStr, runManager?.getChild("reflection_topics"));
 		return GenerativeAgentMemoryChain.parseList(result);
 	}
@@ -143,8 +137,7 @@ var GenerativeAgentMemoryChain = class GenerativeAgentMemoryChain extends BaseCh
 	*/
 	async getInsightsOnTopic(topic, now, runManager) {
 		const prompt = PromptTemplate.fromTemplate("Statements about {topic}\n{related_statements}\n\nWhat 5 high-level insights can you infer from the above statements? (example format: insight (because of 1, 5, 3))");
-		const relatedMemories = await this.fetchMemories(topic, now, runManager);
-		const relatedStatements = relatedMemories.map((memory, index) => `${index + 1}. ${memory.pageContent}`).join("\n");
+		const relatedStatements = (await this.fetchMemories(topic, now, runManager)).map((memory, index) => `${index + 1}. ${memory.pageContent}`).join("\n");
 		const result = await this.chain(prompt).call({
 			topic,
 			related_statements: relatedStatements
@@ -289,8 +282,7 @@ var GenerativeAgentMemory = class extends BaseMemory {
 			});
 			content.push(`${createdTime}: ${memory.pageContent.trim()}`);
 		}
-		const joinedContent = content.map((mem) => `${mem}`).join("\n");
-		return joinedContent;
+		return content.map((mem) => `${mem}`).join("\n");
 	}
 	/**
 	* Method that formats the given relevant memories in a simple manner.
@@ -298,8 +290,7 @@ var GenerativeAgentMemory = class extends BaseMemory {
 	* @returns The formatted memories as a string.
 	*/
 	formatMemoriesSimple(relevantMemories) {
-		const joinedContent = relevantMemories.map((mem) => `${mem.pageContent}`).join("; ");
-		return joinedContent;
+		return relevantMemories.map((mem) => `${mem.pageContent}`).join("; ");
 	}
 	/**
 	* Method that retrieves memories until a token limit is reached.