Spaces:
Runtime error
Runtime error
| """Fake LLM wrapper for testing purposes.""" | |
| from typing import Any, List, Mapping, Optional | |
| from pydantic import BaseModel | |
| from langchain.llms.base import LLM | |
| class FakeListLLM(LLM, BaseModel): | |
| """Fake LLM wrapper for testing purposes.""" | |
| responses: List | |
| i: int = 0 | |
| def _llm_type(self) -> str: | |
| """Return type of llm.""" | |
| return "fake-list" | |
| def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str: | |
| """First try to lookup in queries, else return 'foo' or 'bar'.""" | |
| response = self.responses[self.i] | |
| self.i += 1 | |
| return response | |
| def _identifying_params(self) -> Mapping[str, Any]: | |
| return {} | |