llms_openai.py

from langchain.llms import OpenAI

llm = OpenAI(model_name="text-davinci-003", n=2, best_of=2)

# 保存llm配置
# llm.save("./txt/llm.json")

# 生成文本, get_num_tokens用于计算输入文本的token。
result = llm("随机一份双色球号码")
print()
print("随机一份双色球号码. Number of Tokens：%s"%(llm.get_num_tokens("随机一份双色球号码")), result)

'''
# generate,输入是一个list， 返回多条详细回复list
list = llm.generate(["随机一份双色球号码","取出红球号码"])
print("双色球generate：", list)
'''