mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-18 18:34:27 -04:00
Leonid Ganeline
1c81883d42
This PR targets the `API Reference` documentation.
- Several classes and functions missed `docstrings`. These docstrings
were created.
- In several places this
```
except ImportError:
raise ValueError(
```
was replaced to
```
except ImportError:
raise ImportError(
```
23 lines
678 B
Python
23 lines
678 B
Python
from typing import Any, Dict, List
|
|
|
|
from langchain.schema import get_buffer_string # noqa: 401
|
|
|
|
|
|
def get_prompt_input_key(inputs: Dict[str, Any], memory_variables: List[str]) -> str:
|
|
"""
|
|
Get the prompt input key.
|
|
|
|
Args:
|
|
inputs: Dict[str, Any]
|
|
memory_variables: List[str]
|
|
|
|
Returns:
|
|
A prompt input key.
|
|
"""
|
|
# "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]
|