diff --git a/langchain/llms/base.py b/langchain/llms/base.py index ae8bb97fd7a..3cf6f437a5e 100644 --- a/langchain/llms/base.py +++ b/langchain/llms/base.py @@ -496,7 +496,7 @@ class LLM(BaseLLM): """Run the LLM on the given prompt and input.""" new_arg_supported = inspect.signature(self._acall).parameters.get("run_manager") - async def get_completion(prompt): + async def get_completion(prompt: str) -> str: return ( await self._acall(prompt, stop=stop, run_manager=run_manager) if new_arg_supported