Banana

Banana 专注于构建机器学习基础设施。

这个示例介绍了如何使用 LangChain 与 Banana 模型进行交互。

# 安装包 https://docs.banana.dev/banana-docs/core-concepts/sdks/python
pip install banana-dev

# 获取新的令牌：https://app.banana.dev/
# 我们需要两个令牌，不仅仅是一个 `api_key`：`BANANA_API_KEY` 和 `YOUR_MODEL_KEY`

import os
from getpass import getpass

os.environ["BANANA_API_KEY"] = "YOUR_API_KEY"
# 或者
# BANANA_API_KEY = getpass()

from langchain.llms import Banana
from langchain import PromptTemplate, LLMChain

template = """问题：{question}

答案：让我们一步一步地思考。"""

prompt = PromptTemplate(template=template, input_variables=["question"])

llm = Banana(model_key="YOUR_MODEL_KEY")

llm_chain = LLMChain(prompt=prompt, llm=llm)

question = "贾斯汀·比伯出生的那一年，哪个 NFL 球队赢得了超级碗？"

llm_chain.run(question)