22 个版本

0.1.22 2024 年 2 月 22 日
0.1.21 2023 年 11 月 7 日
0.1.19 2023 年 6 月 16 日
0.1.18 2023 年 5 月 6 日
0.1.9 2023 年 3 月 28 日

#254 in 文本处理

Download history 10/week @ 2024-03-31 1/week @ 2024-04-07 7/week @ 2024-06-30 100/week @ 2024-07-07

每月 107 次下载

Apache-2.0

355KB
6.5K SLoC

Rust 中的 AI

提供一系列第三方 API,供公众访问 AI 功能。

警告:此 crate 不适合生产环境。

支持列表

OpenAI

类别 变体 测试模型
聊天补全 - gpt-4-turbo-preview, gpt-4-0125-preview, gpt-4-1106-vision-preview, gpt-4-1106-preview, gpt-4-vision-preview, gpt-4, gpt-4-0314, gpt-4-0613, gpt-3.5-turbo-0125, gpt-3.5-turbo-1106, gpt-3.5-turbo, gpt-3.5-turbo-16k, gpt-3.5-turbo-16k-0613, gpt-3.5-turbo-0613, gpt-3.5-turbo-0301, text-davinci-003, text-davinci-002, code-davinci-002[^note_3]
补全 - gpt-4-turbo-previewgpt-4-0125-previewgpt-4-1106-vision-previewgpt-4-1106-previewgpt-4-vision-previewgpt-4gpt-4-0314gpt-4-0613gpt-3.5-turbo-0125gpt-3.5-turbo-1106gpt-3.5-turbogpt-3.5-turbo-16kgpt-3.5-turbo-16k-0613gpt-3.5-turbo-0613gpt-3.5-turbo-0301text-davinci-003
编辑 - text-davinci-edit-001code-davinci-edit-001 -
图像 生成 dall-e-2dall-e-3 -
图像 编辑 dall-e-2 -
图像 变体 dall-e-2 -
嵌入 - text-embedding-ada-002[^note_4] -
音频 转录 whisper-1 -
音频 翻译 whisper-1 -
审查 - text-moderation-latest[^note_2],text-moderation-stable -

注意

  • 更新于2024年2月22日。
  • 目前不支持OpenAI的微调端点。
  • 目前不支持OpenAI的助手端点。

Azure

类别 能力 端点
认知服务 语音 文本到语音,语音列表,语音到文本(批量)

注意

  • Azure CN目前不支持此存储库。

使用

在您的当前工作目录或此crate的根目录中创建一个config.yml。内容如下

openai:
  api_key: sk-XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
  org_id: org-xxx
  base_endpoint: https://api.openai.com
azure:
  speech:
    key: 4c7eXXXXXXXXXXXXXXXXXXXXXXX54c32
    region: westus

如果您希望从环境变量中加载配置,设置内容为RUST_AI_CONFIG,例如chat-completion-streamed所做的那样。

[^note_2]:OpenAI的审查响应表明使用text-moderation-004模型(2023年3月23日)。但开发者无法使用其API端点指定除text-moderation-latesttext-moderation-stable之外的变体。

[^note_3]:测试了GPT-4/GPT-4-0314/GPT-4-0613,GPT-4-32K/GPT-4-32K-0314未测试,因为开发者目前只能访问8K令牌的版本(2023年5月6日)。

[^note_4]:您可能会注意到嵌入API背后的实际模型是text-embedding-ada-002-v2(2023年3月23日)。

依赖项

~14-28MB
~498K SLoC