forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
760 B
Python
29 lines
760 B
Python
"""Wrappers on top of large language models APIs."""
|
|
from typing import Dict, Type
|
|
|
|
from langchain.llms.ai21 import AI21
|
|
from langchain.llms.base import BaseLLM
|
|
from langchain.llms.cohere import Cohere
|
|
from langchain.llms.huggingface_hub import HuggingFaceHub
|
|
from langchain.llms.huggingface_pipeline import HuggingFacePipeline
|
|
from langchain.llms.nlpcloud import NLPCloud
|
|
from langchain.llms.openai import OpenAI
|
|
|
|
__all__ = [
|
|
"Cohere",
|
|
"NLPCloud",
|
|
"OpenAI",
|
|
"HuggingFaceHub",
|
|
"HuggingFacePipeline",
|
|
"AI21",
|
|
]
|
|
|
|
type_to_cls_dict: Dict[str, Type[BaseLLM]] = {
|
|
"ai21": AI21,
|
|
"cohere": Cohere,
|
|
"huggingface_hub": HuggingFaceHub,
|
|
"nlpcloud": NLPCloud,
|
|
"openai": OpenAI,
|
|
"huggingface_pipeline": HuggingFacePipeline,
|
|
}
|