Skip to main content
Open In ColabOpen on GitHub

OctoAI

OctoAI 提供便捷的高效计算访问,并允许用户将他们选择的AI模型集成到应用程序中。OctoAI 计算服务帮助您轻松运行、调整和扩展AI应用程序。

本示例介绍了如何使用LangChain与OctoAILLM端点进行交互

设置

要运行我们的示例应用程序,有两个简单的步骤需要采取:

  1. 您的OctoAI账户页面获取API令牌。

  2. 在下面的代码单元格中粘贴您的API密钥。

注意:如果你想使用不同的LLM模型,你可以通过遵循从Python构建容器从容器创建自定义端点来容器化模型并自己创建一个自定义的OctoAI端点,然后更新你的OCTOAI_API_BASE环境变量。

import os

os.environ["OCTOAI_API_TOKEN"] = "OCTOAI_API_TOKEN"
from langchain.chains import LLMChain
from langchain_community.llms.octoai_endpoint import OctoAIEndpoint
from langchain_core.prompts import PromptTemplate

示例

template = """Below is an instruction that describes a task. Write a response that appropriately completes the request.\n Instruction:\n{question}\n Response: """
prompt = PromptTemplate.from_template(template)
llm = OctoAIEndpoint(
model_name="llama-2-13b-chat-fp16",
max_tokens=200,
presence_penalty=0,
temperature=0.1,
top_p=0.9,
)
question = "Who was Leonardo da Vinci?"

chain = prompt | llm

print(chain.invoke(question))

列奥纳多·达·芬奇是一位真正的文艺复兴时期的多才多艺者。他于1452年出生在意大利的芬奇,以其在艺术、科学、工程和数学等多个领域的工作而闻名。他被认为是有史以来最伟大的画家之一,他最著名的作品包括《蒙娜丽莎》和《最后的晚餐》。除了艺术,达·芬奇还对工程学和解剖学做出了重大贡献,他的机器设计和发明超前了几个世纪。他还以其广泛的日记和绘画而闻名,这些作品为他的思想和理念提供了宝贵的见解。达·芬奇的遗产至今仍在激励和影响着世界各地的艺术家、科学家和思想家。


这个页面有帮助吗?