Source code for langchain.memory.utils

from typing import Any, Dict, List


[docs]def get_prompt_input_key(inputs: Dict[str, Any], memory_variables: List[str]) -> str: """获取提示输入键。 参数: 输入:字典[str, 任意] 内存变量:列表[str] 返回: 一个提示输入键。 """ # "stop" is a special key that can be passed as input but is not used to # format the prompt. prompt_input_keys = list(set(inputs).difference(memory_variables + ["stop"])) if len(prompt_input_keys) != 1: raise ValueError(f"One input key expected got {prompt_input_keys}") return prompt_input_keys[0]