mirror of
https://github.com/hwchase17/langchain
synced 2024-11-08 07:10:35 +00:00
1987f905ed
- **Description:** [AS-IS] When dealing with a yaml file, the extension must be .yaml. [TO-BE] In the absence of extension length constraints in the OS, the extension of the YAML file is yaml, but control over the yml extension must still be made. It's as if it's an error because it's a .jpg extension in jpeg support. - **Issue:** - - **Dependencies:** no dependencies required for this change,
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
"""Base interface for loading large language model APIs."""
|
|
import json
|
|
from pathlib import Path
|
|
from typing import Union
|
|
|
|
import yaml
|
|
from langchain_core.language_models.llms import BaseLLM
|
|
|
|
from langchain_community.llms import get_type_to_cls_dict
|
|
|
|
|
|
def load_llm_from_config(config: dict) -> BaseLLM:
|
|
"""Load LLM from Config Dict."""
|
|
if "_type" not in config:
|
|
raise ValueError("Must specify an LLM Type in config")
|
|
config_type = config.pop("_type")
|
|
|
|
type_to_cls_dict = get_type_to_cls_dict()
|
|
|
|
if config_type not in type_to_cls_dict:
|
|
raise ValueError(f"Loading {config_type} LLM not supported")
|
|
|
|
llm_cls = type_to_cls_dict[config_type]()
|
|
return llm_cls(**config)
|
|
|
|
|
|
def load_llm(file: Union[str, Path]) -> BaseLLM:
|
|
"""Load LLM from file."""
|
|
# Convert file to Path object.
|
|
if isinstance(file, str):
|
|
file_path = Path(file)
|
|
else:
|
|
file_path = file
|
|
# Load from either json or yaml.
|
|
if file_path.suffix == ".json":
|
|
with open(file_path) as f:
|
|
config = json.load(f)
|
|
elif file_path.suffix.endswith((".yaml", ".yml")):
|
|
with open(file_path, "r") as f:
|
|
config = yaml.safe_load(f)
|
|
else:
|
|
raise ValueError("File type must be json or yaml")
|
|
# Load the LLM from the config now.
|
|
return load_llm_from_config(config)
|