fix failing test but requires verification

Author: Pouyanpi

tests/test_configs/with_custom_llm/custom_llm.py

@@ -19,7 +19,7 @@
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
 )
-from langchain_core.language_models.base import BaseLLM
+from langchain_core.language_models.llms import BaseLLM
 
 
 class CustomLLM(BaseLLM):
@@ -28,14 +28,48 @@ def _call(
         prompt: str,
         stop: Optional[List[str]] = None,
         run_manager: Optional[CallbackManagerForLLMRun] = None,
-    ) -> str: ...
+        **kwargs,
+    ) -> str:
+        return "Custom LLM response"
 
     async def _acall(
         self,
         prompt: str,
         stop: Optional[List[str]] = None,
         run_manager: Optional[AsyncCallbackManagerForLLMRun] = None,
-    ) -> str: ...
+        **kwargs,
+    ) -> str:
+        return "Custom LLM response"
+
+    def _generate(
+        self,
+        prompts: List[str],
+        stop: Optional[List[str]] = None,
+        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        **kwargs,
+    ):
+        from langchain_core.outputs import Generation, LLMResult
+
+        generations = [
+            [Generation(text=self._call(prompt, stop, run_manager, **kwargs))]
+            for prompt in prompts
+        ]
+        return LLMResult(generations=generations)
+
+    async def _agenerate(
+        self,
+        prompts: List[str],
+        stop: Optional[List[str]] = None,
+        run_manager: Optional[AsyncCallbackManagerForLLMRun] = None,
+        **kwargs,
+    ):
+        from langchain_core.outputs import Generation, LLMResult
+
+        generations = [
+            [Generation(text=await self._acall(prompt, stop, run_manager, **kwargs))]
+            for prompt in prompts
+        ]
+        return LLMResult(generations=generations)
 
     @property
     def _llm_type(self) -> str: