# 導(dǎo)入LLM包裝器
from langchain import OpenAI, ConversationChain
from langchain.agents import initialize_agent
from langchain.agents import load_tools
from langchain.chains import LLMChain
from langchain.prompts import PromptTemplate

llm = OpenAI(model_name="text-davinci-003", temperature=0.9) // 這些都是OpenAI的參數(shù)
text = "What would be a good company name for a company that makes colorful socks?"
print(llm(text)) 
// 以上就是打印調(diào)用OpenAI接口的返回值，相當(dāng)于接口的封裝，實(shí)現(xiàn)的代碼可以看看github.com/hwchase17/langchain/llms/openai.py的OpenAIChat

prompt = PromptTemplate(
    input_variables=["product"],
    template="What is a good name for a company that makes {product}?",
)
// 以上是兩個(gè)參數(shù)，一個(gè)輸入變量，一個(gè)模板字符串，實(shí)現(xiàn)的代碼可以看看github.com/hwchase17/langchain/prompts
// PromptTemplate實(shí)際是基于StringPromptTemplate，可以支持字符串類型的模板，也可以支持文件類型的模板

What is a good name for a company that makes colorful socks?

from langchain.chains import LLMChain
chain = LLMChain(llm=llm, prompt=prompt) // 通過(guò)LLM的llm變量，Prompt Templates的prompt生成LLMChain
chain.run("colorful socks") // 實(shí)際這里就變成了實(shí)際問題：What is a good name for a company that makes colorful socks？

# 導(dǎo)入一些tools，比如llm-math
# llm-math是langchain里面的能做數(shù)學(xué)計(jì)算的模塊
tools = load_tools(["llm-math"], llm=llm)
# 初始化tools，models 和使用的agent
agent = initialize_agent(
    tools, llm, agent="zero-shot-react-description", verbose=True)
text = "12 raised to the 3 power and result raised to 2 power?"
print("input text: ", text)
agent.run(text)

> Entering new AgentExecutor chain...
 I need to use the calculator for this
Action: Calculator
Action Input: 12^3
Observation: Answer: 1728
Thought: I need to then raise the previous result to the second power
Action: Calculator
Action Input: 1728^2
Observation: Answer: 2985984
Thought: I now know the final answer
Final Answer: 2985984
> Finished chain.

# ConversationChain用法
llm = OpenAI(temperature=0)
# 將verbose設(shè)置為True，以便我們可以看到提示
conversation = ConversationChain(llm=llm, verbose=True)
print("input text: conversation")
conversation.predict(input="Hi there!")
conversation.predict(
  input="I'm doing well! Just having a conversation with an AI.")

# 導(dǎo)入LLM包裝器
from langchain import OpenAI, ConversationChain
from langchain.agents import initialize_agent
from langchain.agents import load_tools
from langchain.chains import LLMChain
from langchain.prompts import PromptTemplate
# 初始化包裝器，temperature越高結(jié)果越隨機(jī)
llm = OpenAI(temperature=0.9)
# 進(jìn)行調(diào)用
text = "What would be a good company name for a company that makes colorful socks?"
print("input text: ", text)
print(llm(text))
prompt = PromptTemplate(
  input_variables=["product"],
  template="What is a good name for a company that makes {product}?",
)
print("input text: product")
print(prompt.format(product="colorful socks"))
chain = LLMChain(llm=llm, prompt=prompt)
chain.run("colorful socks")
# 導(dǎo)入一些tools，比如llm-math
# llm-math是langchain里面的能做數(shù)學(xué)計(jì)算的模塊
tools = load_tools(["llm-math"], llm=llm)
# 初始化tools，models 和使用的agent
agent = initialize_agent(tools,
                         llm,
                         agent="zero-shot-react-description",
                         verbose=True)
text = "12 raised to the 3 power and result raised to 2 power?"
print("input text: ", text)
agent.run(text)
# ConversationChain用法
llm = OpenAI(temperature=0)
# 將verbose設(shè)置為True，以便我們可以看到提示
conversation = ConversationChain(llm=llm, verbose=True)
print("input text: conversation")
conversation.predict(input="Hi there!")
conversation.predict(
  input="I'm doing well! Just having a conversation with an AI.")