Skip to main content
Open In ColabOpen on GitHub

OctoAI

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

本示例将介绍如何使用 LangChain 与 OctoAI 的 LLM 端点 进行交互。

设置​

要运行我们的示例应用程序，只需执行两个简单的步骤：

  1. 从您的 OctoAI 账户页面获取 API Token。

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

注意：如果您想使用不同的 LLM 模型，可以将其容器化并自行创建自定义 OctoAI 端点。请遵循从 Python 构建容器和从容器创建自定义端点的说明，然后更新您的 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 年出生在意大利的芬奇镇，以其在艺术、科学、工程和数学等多个领域的贡献而闻名。他被认为是史上最伟大的画家之一，其最著名的作品包括《蒙娜丽莎》和《最后的晚餐》。除了艺术，达·芬奇在工程学和解剖学方面也做出了重要贡献，他设计的机器和发明比他所处的时代领先数个世纪。他还以其详尽的笔记和图纸而闻名，这些为我们深入了解他的思想和想法提供了宝贵的见解。达·芬奇的遗产至今仍在激励和影响着世界各地的艺术家、科学家和思想家。