Source code for langchain_community.llms.openlm

from typing import Any, Dict

from langchain_core.pydantic_v1 import root_validator

from langchain_community.llms.openai import BaseOpenAI

[docs]class OpenLM(BaseOpenAI): """OpenLM models."""
[docs] @classmethod def is_lc_serializable(cls) -> bool: return False
@property def _invocation_params(self) -> Dict[str, Any]: return {**{"model": self.model_name}, **super()._invocation_params} @root_validator() def validate_environment(cls, values: Dict) -> Dict: try: import openlm values["client"] = openlm.Completion except ImportError: raise ImportError( "Could not import openlm python package. " "Please install it with `pip install openlm`." ) if values["streaming"]: raise ValueError("Streaming not supported with openlm") return values