为了将一个自定义的语言模型集成到 LangChain 中,你需要创建一个类来继承 langchain_core.language_models.llms.LLM
类,并实现特定的方法。下面是一些关键点,可以帮助你构思如何集成你的语言模型:
继承 LLM
类
你需要从 langchain_core.language_models.llms.LLM
类继承,并且至少实现 _call
方法。这个基类提供了一些通用的功能,比如回调管理等。
实现 _call
方法
这是最重要的方法,它定义了如何调用你的语言模型来生成文本。方法签名如下:
def _call(
self,
prompt: str,
stop: Optional[List[str]] = None,
run_manager: Optional[CallbackManagerForLLMRun] = None,
**kwargs: Any,
) -> str:
prompt
: 用户提供的文本提示。stop
: 一个可选的字符串列表,这些字符串如果出现在生成的文本中,将会停止生成过程。run_manager
: 一个可选的回调管理器,用于跟踪和报告生成过程中的状态。**kwargs
: 其他可变的关键字参数,这些参数可能会被模型使用。
其他常用方法和属性
_llm_type
属性: 返回一个字符串标识你的语言模型的类型。_identifying_params
属性: 返回一个字典,其中包含识别该模型实例的重要参数。_check_validity_of_input
方法: 用于检查输入的