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 FakeLLM(LLM, BaseModel): | |
"""Fake LLM wrapper for testing purposes.""" | |
queries: Optional[Mapping] = None | |
def _llm_type(self) -> str: | |
"""Return type of llm.""" | |
return "fake" | |
def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str: | |
"""First try to lookup in queries, else return 'foo' or 'bar'.""" | |
if self.queries is not None: | |
return self.queries[prompt] | |
if stop is None: | |
return "foo" | |
else: | |
return "bar" | |
def _identifying_params(self) -> Mapping[str, Any]: | |
return {} | |