SimpleAI: 自托管AI API服务项目教程

1. 项目介绍

SimpleAI是一个开源项目，旨在提供一个自托管的AI API服务，允许用户轻松地部署自己的模型，并通过标准的API接口进行访问。它兼容开放的AI客户端，可以作为一个替代方案来实验不同的模型，创建性能基准，以及处理那些不能完全依赖外部服务的特定用例。

2. 项目快速启动

要快速启动SimpleAI项目，请按照以下步骤操作：

首先，确保你的系统中已安装Python 3.9或更高版本。然后，你可以通过以下方式安装SimpleAI：

# 从源代码安装
pip install git+https://github.com/lhenault/simpleAI

# 或者从PyPi安装
pip install simple_ai_server

安装完成后，初始化项目配置文件：

simple_ai init

这会创建一个models.toml文件，用于声明你的模型。然后，启动服务：

simple_ai serve [--host 127.0.0.1] [--port 8080]

你可以通过浏览器访问http://127.0.0.1:8080查看文档和尝试API。

3. 应用案例和最佳实践

3.1 集成模型

SimpleAI通过gRPC协议查询模型，以便将API与模型推断分离，并支持多种语言。以下是一个简单的Python模型集成示例：

import logging
from dataclasses import dataclass
from simple_ai.api.grpc.embedding.server import serve, LanguageModelServicer

@dataclass(unsafe_hash=True)
class EmbeddingModel:
    def embed(self, inputs: list = []) -> list:
        # 实现embed方法
        return [[]]

if __name__ == '__main__':
    model_servicer = LanguageModelServicer(model=EmbeddingModel())
    serve(address='[::]:50051', model_servicer=model_servicer)

3.2 声明模型

在你的models.toml文件中添加新模型，例如：

[llama-7B-4b]
[llama-7B-4b.metadata]
owned_by = 'Meta/ggerganov'
permission = []
description = 'C++实现LlaMA模型，70亿参数，4位量化'
[llama-7B-4b.network]
url = 'localhost:50051'
type = 'gRPC'

3.3 使用API

你可以使用curl或者OpenAI的Python客户端来调用API：

curl -X POST http://127.0.0.1:8080/edits -H 'accept: application/json' -H 'Content-Type: application/json' -d '{
    "model": "alpaca-lora-7B",
    "instruction": "Make this message nicer and more formal",
    "input": "This meeting was useless and should have been a bloody email",
    "top_p": 1,
    "n": 1,
    "temperature": 1,
    "max_tokens": 256
}'

或者使用Python客户端：

import openai

openai.api_key = 'Free the models'
openai.api_base = "http://127.0.0.1:8080"

print(openai.Model.list())
completion = openai.Completion.create(model="llama-7B", prompt="Hello everyone this is")

4. 典型生态项目

SimpleAI可以作为各种机器学习和MLOps生态项目的一部分，支持LLM（大型语言模型）等应用。你可以探索将其集成到更广泛的工作流程中，例如数据标注、模型训练、模型评估和部署等。

以上是SimpleAI项目的简要介绍、快速启动指南、应用案例和生态项目的一篇教程。希望对你有所帮助！