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]