Skip to main content
Open on GitHub

Shale Protocol

页岩协议 为开源大型语言模型提供生产级推理API。它是一个即插即用的API,因为它托管在高度可扩展的GPU云基础设施上。

我们的免费层级支持每个密钥每天最多 1K 次请求,因为我们希望消除任何人使用大型语言模型开始构建生成式 AI 应用的障碍。

通过Shale协议,开发者/研究人员可以免费创建应用程序并探索开源LLM的能力。

本页面介绍了如何将 Shale-Serve API 与 LangChain 结合使用。

截至2023年6月,API默认支持Vicuna-13B。我们计划在未来的版本中支持更多大型语言模型,例如Falcon-40B。

如何

2. 使用 https://shale.live/v1 作为 OpenAI API 的替代品

例如

from langchain_openai import OpenAI
from langchain_core.prompts import PromptTemplate
from langchain_core.output_parsers import StrOutputParser

import os
os.environ['OPENAI_API_BASE'] = "https://shale.live/v1"
os.environ['OPENAI_API_KEY'] = "ENTER YOUR API KEY"

llm = OpenAI()

template = """Question: {question}

# Answer: Let's think step by step."""

prompt = PromptTemplate.from_template(template)


llm_chain = prompt | llm | StrOutputParser()

question = "What NFL team won the Super Bowl in the year Justin Beiber was born?"

llm_chain.invoke(question)