赞
踩
本笔记:讲述Chain的不同调用方式。
所有从 Chain
继承的类中都提供了几种运行链逻辑的方法。最直接的方法是使用 __call__
;
说明:__call__
这个是python版中的方法已经实现了。
chat = ChatOpenAI(temperature=0)
prompt_template = "Tell me a {adjective} joke"
llm_chain = LLMChain(llm=chat, prompt=PromptTemplate.from_template(prompt_template))
llm_chain(inputs={"adjective": "corny"})
结果:
{'adjective': 'corny',
'text': 'Why did the tomato turn red? Because it saw the salad dressing!'}
默认情况下,__call__
返回的是键值对:输入和输出。我们可以通过将return_only_outputs
设置为True
:仅返回输出键值对。
如果我们想Chain
只输出一个key(即:output_keys
中只有一个元素),则可以使用run
方法。
请注意:run
方法输出一个字符串而不是键值对(或者叫:字典)。
# llm_chain only has one output key, so we can use run
# llm_chain 输出只有一个key的情况下,我们可以使用run方法。
llm_chain.output_keys
# 结果
['text']
则,使用run
方法:
llm_chain.run({"adjective": "corny"})
# 结果
'Why did the tomato turn red? Because it saw the salad dressing!'
在只有一个输入key的情况下,可以直接输入字符串,无需指定输入映射:
# These two are equivalent 这两个是等价的
llm_chain.run({"adjective": "corny"})
llm_chain.run("corny")
# These two are also equivalent 这两个也是等价的
llm_chain("corny")
llm_chain({"adjective": "corny"})
结果:
{'adjective': 'corny',
'text': 'Why did the tomato turn red? Because it saw the salad dressing!'}
我们可以通过其run
方法轻松地将Chain
对象作为Tool
集成Agent
中。请参阅此处的示例。
这里给个理解代码:
tools.append( Tool.from_function( func=llm_math_chain.run, name="Calculator", description="useful for when you need to answer questions about math", args_schema=CalculatorInput # coroutine= ... <- you can specify an async method if desired as well ) ) # Construct the agent. We will use the default agent type here. # See documentation for a full list of options. # 集成到了agent中 agent = initialize_agent( tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True ) agent.run( "Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?" )
要调用Chain的方法。
llm_chain
llm_chain = LLMChain(llm=chat, prompt=PromptTemplate.from_template(prompt_template))
llm_chain(xxx)
,即可。默认情况下,输入和输出都会打印出来llm_chain(inputs={"adjective": "corny"})
# 结果
{'adjective': 'corny',
'text': 'Why did the tomato turn red? Because it saw the salad dressing!'}
run
方法。llm_chain("corny")
llm_chain.run("corny")
参考地址:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。