mirror of
https://github.com/LLM-Red-Team/glm-free-api.git
synced 2025-04-20 03:29:23 +08:00
Compare commits
27 Commits
Author | SHA1 | Date | |
---|---|---|---|
|
2f12a5daef | ||
|
2ce738b1fc | ||
|
719e3b682f | ||
|
57b042d187 | ||
|
05ecba5cc2 | ||
|
a969acd6fb | ||
|
54805f2475 | ||
|
b402f99960 | ||
|
c1f5e9ae78 | ||
|
fe3f0784c8 | ||
|
3237198289 | ||
|
f56e582ec6 | ||
|
d53a39f45a | ||
|
6519a575fa | ||
|
72b3698757 | ||
|
620f6202de | ||
|
90a9a702ed | ||
|
34282ad837 | ||
|
53f3365872 | ||
|
cb044beae0 | ||
|
9b650fab98 | ||
|
afc9c96d02 | ||
|
506de6f791 | ||
|
6f48e347eb | ||
|
4a450b8848 | ||
|
756c89aaef | ||
|
a2770fea93 |
@ -4,7 +4,7 @@ WORKDIR /app
|
|||||||
|
|
||||||
COPY . /app
|
COPY . /app
|
||||||
|
|
||||||
RUN yarn install --registry https://registry.npmmirror.com/ && yarn run build
|
RUN yarn install --registry https://registry.npmmirror.com/ --ignore-engines && yarn run build
|
||||||
|
|
||||||
FROM node:lts-alpine
|
FROM node:lts-alpine
|
||||||
|
|
||||||
|
107
README.md
107
README.md
@ -1,15 +1,19 @@
|
|||||||
# GLM AI Free 服务
|
# GLM AI Free 服务
|
||||||
|
|
||||||
|
<hr>
|
||||||
|
|
||||||
|
<span>[ 中文 | <a href="README_EN.md">English</a> ]</span>
|
||||||
|
|
||||||
[](LICENSE)
|
[](LICENSE)
|
||||||

|

|
||||||

|

|
||||||

|

|
||||||
|
|
||||||
支持高速流式输出、支持多轮对话、支持智能体对话、支持AI绘图、支持联网搜索、支持长文档解读、支持图像解析,零配置部署,多路token支持,自动清理会话痕迹。
|
支持GLM-4-Plus高速流式输出、支持多轮对话、支持智能体对话、支持Zero思考推理模型、支持视频生成、支持AI绘图、支持联网搜索、支持长文档解读、支持图像解析,零配置部署,多路token支持,自动清理会话痕迹。
|
||||||
|
|
||||||
与ChatGPT接口完全兼容。
|
与ChatGPT接口完全兼容。
|
||||||
|
|
||||||
还有以下六个free-api欢迎关注:
|
还有以下十个free-api欢迎关注:
|
||||||
|
|
||||||
Moonshot AI(Kimi.ai)接口转API [kimi-free-api](https://github.com/LLM-Red-Team/kimi-free-api)
|
Moonshot AI(Kimi.ai)接口转API [kimi-free-api](https://github.com/LLM-Red-Team/kimi-free-api)
|
||||||
|
|
||||||
@ -19,26 +23,34 @@ Moonshot AI(Kimi.ai)接口转API [kimi-free-api](https://github.com/LLM-Red-
|
|||||||
|
|
||||||
秘塔AI (Metaso) 接口转API [metaso-free-api](https://github.com/LLM-Red-Team/metaso-free-api)
|
秘塔AI (Metaso) 接口转API [metaso-free-api](https://github.com/LLM-Red-Team/metaso-free-api)
|
||||||
|
|
||||||
|
字节跳动(豆包)接口转API [doubao-free-api](https://github.com/LLM-Red-Team/doubao-free-api)
|
||||||
|
|
||||||
|
字节跳动(即梦AI)接口转API [jimeng-free-api](https://github.com/LLM-Red-Team/jimeng-free-api)
|
||||||
|
|
||||||
讯飞星火(Spark)接口转API [spark-free-api](https://github.com/LLM-Red-Team/spark-free-api)
|
讯飞星火(Spark)接口转API [spark-free-api](https://github.com/LLM-Red-Team/spark-free-api)
|
||||||
|
|
||||||
聆心智能 (Emohaa) 接口转API [emohaa-free-api](https://github.com/LLM-Red-Team/emohaa-free-api)
|
MiniMax(海螺AI)接口转API [hailuo-free-api](https://github.com/LLM-Red-Team/hailuo-free-api)
|
||||||
|
|
||||||
|
深度求索(DeepSeek)接口转API [deepseek-free-api](https://github.com/LLM-Red-Team/deepseek-free-api)
|
||||||
|
|
||||||
|
聆心智能 (Emohaa) 接口转API [emohaa-free-api](https://github.com/LLM-Red-Team/emohaa-free-api)(当前不可用)
|
||||||
|
|
||||||
## 目录
|
## 目录
|
||||||
|
|
||||||
* [免责声明](#免责声明)
|
* [免责声明](#免责声明)
|
||||||
* [在线体验](#在线体验)
|
|
||||||
* [效果示例](#效果示例)
|
* [效果示例](#效果示例)
|
||||||
* [接入准备](#接入准备)
|
* [接入准备](#接入准备)
|
||||||
* [智能体接入](#智能体接入)
|
* [智能体接入](#智能体接入)
|
||||||
* [多账号接入](#多账号接入)
|
* [多账号接入](#多账号接入)
|
||||||
* [Docker部署](#Docker部署)
|
* [Docker部署](#Docker部署)
|
||||||
* [Docker-compose部署](#Docker-compose部署)
|
* [Docker-compose部署](#Docker-compose部署)
|
||||||
* [Render部署](#Render部署)
|
* [Render部署](#Render部署)
|
||||||
* [Vercel部署](#Vercel部署)
|
* [Vercel部署](#Vercel部署)
|
||||||
* [原生部署](#原生部署)
|
* [原生部署](#原生部署)
|
||||||
* [推荐使用客户端](#推荐使用客户端)
|
* [推荐使用客户端](#推荐使用客户端)
|
||||||
* [接口列表](#接口列表)
|
* [接口列表](#接口列表)
|
||||||
* [对话补全](#对话补全)
|
* [对话补全](#对话补全)
|
||||||
|
* [视频生成](#视频生成)
|
||||||
* [AI绘图](#AI绘图)
|
* [AI绘图](#AI绘图)
|
||||||
* [文档解读](#文档解读)
|
* [文档解读](#文档解读)
|
||||||
* [图像解析](#图像解析)
|
* [图像解析](#图像解析)
|
||||||
@ -60,12 +72,6 @@ Moonshot AI(Kimi.ai)接口转API [kimi-free-api](https://github.com/LLM-Red-
|
|||||||
|
|
||||||
**仅限自用,禁止对外提供服务或商用,避免对官方造成服务压力,否则风险自担!**
|
**仅限自用,禁止对外提供服务或商用,避免对官方造成服务压力,否则风险自担!**
|
||||||
|
|
||||||
## 在线体验
|
|
||||||
|
|
||||||
此链接仅临时测试功能,只有一路并发,如果遇到异常请稍后重试,建议自行部署使用。
|
|
||||||
|
|
||||||
https://udify.app/chat/Pe89TtaX3rKXM8NS
|
|
||||||
|
|
||||||
## 效果示例
|
## 效果示例
|
||||||
|
|
||||||
### 验明正身Demo
|
### 验明正身Demo
|
||||||
@ -88,6 +94,10 @@ https://udify.app/chat/Pe89TtaX3rKXM8NS
|
|||||||
|
|
||||||

|

|
||||||
|
|
||||||
|
### 视频生成Demo
|
||||||
|
|
||||||
|
[点击预览](https://sfile.chatglm.cn/testpath/video/c1f59468-32fa-58c3-bd9d-ab4230cfe3ca_0.mp4)
|
||||||
|
|
||||||
### AI绘图Demo
|
### AI绘图Demo
|
||||||
|
|
||||||

|

|
||||||
@ -276,9 +286,11 @@ Authorization: Bearer [refresh_token]
|
|||||||
请求数据:
|
请求数据:
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
// 如果使用智能体请填写智能体ID到此处,否则可以乱填
|
// 默认模型:glm-4-plus
|
||||||
"model": "glm4",
|
// zero思考推理模型:glm-4-zero / glm-4-think
|
||||||
// 目前多轮对话基于消息合并实现,某些场景可能导致能力下降且token最高为4096
|
// 如果使用智能体请填写智能体ID到此处
|
||||||
|
"model": "glm-4-plus",
|
||||||
|
// 目前多轮对话基于消息合并实现,某些场景可能导致能力下降且受单轮最大token数限制
|
||||||
// 如果您想获得原生的多轮对话体验,可以传入首轮消息获得的id,来接续上下文
|
// 如果您想获得原生的多轮对话体验,可以传入首轮消息获得的id,来接续上下文
|
||||||
// "conversation_id": "65f6c28546bae1f0fbb532de",
|
// "conversation_id": "65f6c28546bae1f0fbb532de",
|
||||||
"messages": [
|
"messages": [
|
||||||
@ -295,9 +307,9 @@ Authorization: Bearer [refresh_token]
|
|||||||
响应数据:
|
响应数据:
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
// conversation_id,你可以传入到下一轮对话来接续上下文
|
// 如果想获得原生多轮对话体验,此id,你可以传入到下一轮对话的conversation_id来接续上下文
|
||||||
"id": "65f6c28546bae1f0fbb532de",
|
"id": "65f6c28546bae1f0fbb532de",
|
||||||
"model": "glm4",
|
"model": "glm-4",
|
||||||
"object": "chat.completion",
|
"object": "chat.completion",
|
||||||
"choices": [
|
"choices": [
|
||||||
{
|
{
|
||||||
@ -318,9 +330,64 @@ Authorization: Bearer [refresh_token]
|
|||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### 视频生成
|
||||||
|
|
||||||
|
视频生成接口
|
||||||
|
|
||||||
|
**如果您的账号未开通VIP,可能会因排队导致生成耗时较久**
|
||||||
|
|
||||||
|
**POST /v1/videos/generations**
|
||||||
|
|
||||||
|
header 需要设置 Authorization 头部:
|
||||||
|
|
||||||
|
```
|
||||||
|
Authorization: Bearer [refresh_token]
|
||||||
|
```
|
||||||
|
|
||||||
|
请求数据:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
// 模型名称
|
||||||
|
// cogvideox:默认官方视频模型
|
||||||
|
// cogvideox-pro:先生成图像再作为参考图像生成视频,作为视频首帧引导视频效果,但耗时更长
|
||||||
|
"model": "cogvideox",
|
||||||
|
// 视频生成提示词
|
||||||
|
"prompt": "一只可爱的猫走在花丛中",
|
||||||
|
// 支持使用图像URL或者BASE64_URL作为视频首帧参考图像(如果使用cogvideox-pro则会忽略此参数)
|
||||||
|
// "image_url": "https://sfile.chatglm.cn/testpath/b5341945-3839-522c-b4ab-a6268cb131d5_0.png",
|
||||||
|
// 支持设置视频风格:卡通3D/黑白老照片/油画/电影感
|
||||||
|
// "video_style": "油画",
|
||||||
|
// 支持设置情感氛围:温馨和谐/生动活泼/紧张刺激/凄凉寂寞
|
||||||
|
// "emotional_atmosphere": "生动活泼",
|
||||||
|
// 支持设置运镜方式:水平/垂直/推近/拉远
|
||||||
|
// "mirror_mode": "水平"
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
响应数据:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"created": 1722103836,
|
||||||
|
"data": [
|
||||||
|
{
|
||||||
|
// 对话ID,目前没啥用
|
||||||
|
"conversation_id": "66a537ec0603e53bccb8900a",
|
||||||
|
// 封面URL
|
||||||
|
"cover_url": "https://sfile.chatglm.cn/testpath/video_cover/c1f59468-32fa-58c3-bd9d-ab4230cfe3ca_cover_0.png",
|
||||||
|
// 视频URL
|
||||||
|
"video_url": "https://sfile.chatglm.cn/testpath/video/c1f59468-32fa-58c3-bd9d-ab4230cfe3ca_0.mp4",
|
||||||
|
// 视频时长
|
||||||
|
"video_duration": "6s",
|
||||||
|
// 视频分辨率
|
||||||
|
"resolution": "1440 × 960"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
### AI绘图
|
### AI绘图
|
||||||
|
|
||||||
对话补全接口,与openai的 [images-create-api](https://platform.openai.com/docs/api-reference/images/create) 兼容。
|
图像生成接口,与openai的 [images-create-api](https://platform.openai.com/docs/api-reference/images/create) 兼容。
|
||||||
|
|
||||||
**POST /v1/images/generations**
|
**POST /v1/images/generations**
|
||||||
|
|
||||||
@ -367,7 +434,7 @@ Authorization: Bearer [refresh_token]
|
|||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
// 如果使用智能体请填写智能体ID到此处,否则可以乱填
|
// 如果使用智能体请填写智能体ID到此处,否则可以乱填
|
||||||
"model": "glm4",
|
"model": "glm-4",
|
||||||
"messages": [
|
"messages": [
|
||||||
{
|
{
|
||||||
"role": "user",
|
"role": "user",
|
||||||
@ -394,7 +461,7 @@ Authorization: Bearer [refresh_token]
|
|||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
"id": "cnmuo7mcp7f9hjcmihn0",
|
"id": "cnmuo7mcp7f9hjcmihn0",
|
||||||
"model": "glm4",
|
"model": "glm-4",
|
||||||
"object": "chat.completion",
|
"object": "chat.completion",
|
||||||
"choices": [
|
"choices": [
|
||||||
{
|
{
|
||||||
|
596
README_EN.md
Normal file
596
README_EN.md
Normal file
@ -0,0 +1,596 @@
|
|||||||
|
# GLM AI Free Service
|
||||||
|
|
||||||
|
[](LICENSE)
|
||||||
|

|
||||||
|

|
||||||
|

|
||||||
|
|
||||||
|
Supports high-speed streaming output, multi-turn dialogues, internet search, long document reading, image analysis, zero-configuration deployment, multi-token support, and automatic session trace cleanup.
|
||||||
|
|
||||||
|
Fully compatible with the ChatGPT interface.
|
||||||
|
|
||||||
|
Also, the following free APIs are available for your attention:
|
||||||
|
|
||||||
|
Moonshot AI (Kimi.ai) API to API [kimi-free-api](https://github.com/LLM-Red-Team/kimi-free-api/tree/master)
|
||||||
|
|
||||||
|
StepFun (StepChat) API to API [step-free-api](https://github.com/LLM-Red-Team/step-free-api)
|
||||||
|
|
||||||
|
Ali Tongyi (Qwen) API to API [qwen-free-api](https://github.com/LLM-Red-Team/qwen-free-api)
|
||||||
|
|
||||||
|
ZhipuAI (ChatGLM) API to API [glm-free-api](https://github.com/LLM-Red-Team/glm-free-api)
|
||||||
|
|
||||||
|
ByteDance (Doubao) API to API [doubao-free-api](https://github.com/LLM-Red-Team/doubao-free-api)
|
||||||
|
|
||||||
|
Meta Sota (metaso) API to API [metaso-free-api](https://github.com/LLM-Red-Team/metaso-free-api)
|
||||||
|
|
||||||
|
Iflytek Spark (Spark) API to API [spark-free-api](https://github.com/LLM-Red-Team/spark-free-api)
|
||||||
|
|
||||||
|
MiniMax(Hailuo)API to API [hailuo-free-api](https://github.com/LLM-Red-Team/hailuo-free-api)
|
||||||
|
|
||||||
|
DeepSeek(DeepSeek)API to API [deepseek-free-api](https://github.com/LLM-Red-Team/deepseek-free-api)
|
||||||
|
|
||||||
|
Lingxin Intelligence (Emohaa) API to API [emohaa-free-api](https://github.com/LLM-Red-Team/emohaa-free-api) (OUT OF ORDER)
|
||||||
|
|
||||||
|
## Table of Contents
|
||||||
|
|
||||||
|
* [Announcement](#Announcement)
|
||||||
|
* [Online Experience](#Online-Experience)
|
||||||
|
* [Effect Examples](#Effect-Examples)
|
||||||
|
* [Access Preparation](#Access-Preparation)
|
||||||
|
* [Agent Access](#Agent-Access)
|
||||||
|
* [Multiple Account Access](#Multiple-Account-Access)
|
||||||
|
* [Docker Deployment](#Docker-Deployment)
|
||||||
|
* [Docker-compose Deployment](#Docker-compose-Deployment)
|
||||||
|
* [Render Deployment](#Render-Deployment)
|
||||||
|
* [Vercel Deployment](#Vercel-Deployment)
|
||||||
|
* [Native Deployment](#Native-Deployment)
|
||||||
|
* [Recommended Clients](#Recommended-Clients)
|
||||||
|
* [Interface List](#Interface-List)
|
||||||
|
* [Conversation Completion](#Conversation-Completion)
|
||||||
|
* [Video Generation](#Video-Generation)
|
||||||
|
* [AI Drawing](#AI-Drawing)
|
||||||
|
* [Document Interpretation](#Document-Interpretation)
|
||||||
|
* [Image Analysis](#Image-Analysis)
|
||||||
|
* [Refresh_token Survival Detection](#Refresh_token-Survival-Detection)
|
||||||
|
* [Notification](#Notification)
|
||||||
|
* [Nginx Anti-generation Optimization](#Nginx-Anti-generation-Optimization)
|
||||||
|
* [Token Statistics](#Token-Statistics)
|
||||||
|
* [Star History](#star-history)
|
||||||
|
|
||||||
|
## Announcement
|
||||||
|
|
||||||
|
**This API is unstable. So we highly recommend you go to the [Zhipu](https://open.bigmodel.cn/) use the offical API, avoiding banned.**
|
||||||
|
|
||||||
|
**This organization and individuals do not accept any financial donations and transactions. This project is purely for research, communication, and learning purposes!**
|
||||||
|
|
||||||
|
**For personal use only, it is forbidden to provide services or commercial use externally to avoid causing service pressure on the official, otherwise, bear the risk yourself!**
|
||||||
|
|
||||||
|
**For personal use only, it is forbidden to provide services or commercial use externally to avoid causing service pressure on the official, otherwise, bear the risk yourself!**
|
||||||
|
|
||||||
|
**For personal use only, it is forbidden to provide services or commercial use externally to avoid causing service pressure on the official, otherwise, bear the risk yourself!**
|
||||||
|
|
||||||
|
## Online Experience
|
||||||
|
This link is only for temporary testing of functions and cannot be used for a long time. For long-term use, please deploy by yourself.
|
||||||
|
|
||||||
|
https://udify.app/chat/Pe89TtaX3rKXM8NS
|
||||||
|
|
||||||
|
## Effect Examples
|
||||||
|
|
||||||
|
### Identity Verification
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
### AI-Agent
|
||||||
|
|
||||||
|
Agent link:[Comments Generator](https://chatglm.cn/main/gdetail/65c046a531d3fcb034918abe)
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
### Combined with Dify workflow
|
||||||
|
|
||||||
|
Experience link:https://udify.app/chat/m46YgeVLNzFh4zRs
|
||||||
|
|
||||||
|
<img width="390" alt="image" src="https://github.com/LLM-Red-Team/glm-free-api/assets/20235341/4773b9f6-b1ca-460c-b3a7-c56bdb1f0659">
|
||||||
|
|
||||||
|
### Multi-turn Dialogue
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
### Video Generation
|
||||||
|
|
||||||
|
[View](https://sfile.chatglm.cn/testpath/video/c1f59468-32fa-58c3-bd9d-ab4230cfe3ca_0.mp4)
|
||||||
|
|
||||||
|
### AI Drawing
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
### Internet Search
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
### Long Document Reading
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
### Using Code
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
### Image Analysis
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
## Access Preparation
|
||||||
|
|
||||||
|
Obtain `refresh_token` from [Zhipu](https://chatglm.cn/)
|
||||||
|
|
||||||
|
Enter Zhipu Qingyan and start a random conversation, then press F12 to open the developer tools. Find the value of `tongyi_sso_ticket` in Application > Cookies, which will be used as the Bearer Token value for Authorization: `Authorization: Bearer TOKEN`
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
### Agent Access
|
||||||
|
|
||||||
|
Open a window of Agent Chat, the ID in the url is the ID of the Agent, which is the parameter of `model`.
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
|
### Multiple Account Access
|
||||||
|
|
||||||
|
You can provide multiple account chatglm_refresh_tokens and use `,` to join them:
|
||||||
|
|
||||||
|
`Authorization: Bearer TOKEN1,TOKEN2,TOKEN3`
|
||||||
|
|
||||||
|
The service will pick one each time a request is made.
|
||||||
|
|
||||||
|
## Docker Deployment
|
||||||
|
|
||||||
|
Please prepare a server with a public IP and open port 8000.
|
||||||
|
|
||||||
|
Pull the image and start the service
|
||||||
|
|
||||||
|
```shell
|
||||||
|
docker run -it -d --init --name step-free-api -p 8000:8000 -e TZ=Asia/Shanghai vinlic/step-free-api:latest
|
||||||
|
```
|
||||||
|
|
||||||
|
check real-time service logs
|
||||||
|
|
||||||
|
```shell
|
||||||
|
docker logs -f glm-free-api
|
||||||
|
```
|
||||||
|
|
||||||
|
Restart service
|
||||||
|
|
||||||
|
```shell
|
||||||
|
docker restart glm-free-api
|
||||||
|
```
|
||||||
|
|
||||||
|
Shut down service
|
||||||
|
|
||||||
|
```shell
|
||||||
|
docker stop glm-free-api
|
||||||
|
```
|
||||||
|
|
||||||
|
### Docker-compose Deployment
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
version: '3'
|
||||||
|
|
||||||
|
services:
|
||||||
|
glm-free-api:
|
||||||
|
container_name: glm-free-api
|
||||||
|
image: vinlic/glm-free-api:latest
|
||||||
|
restart: always
|
||||||
|
ports:
|
||||||
|
- "8000:8000"
|
||||||
|
environment:
|
||||||
|
- TZ=Asia/Shanghai
|
||||||
|
```
|
||||||
|
|
||||||
|
### Render Deployment
|
||||||
|
|
||||||
|
**Attention: Some deployment regions may not be able to connect to Kimi. If container logs show request timeouts or connection failures (Singapore has been tested and found unavailable), please switch to another deployment region!**
|
||||||
|
|
||||||
|
**Attention: Container instances for free accounts will automatically stop after a period of inactivity, which may result in a 50-second or longer delay during the next request. It is recommended to check [Render Container Keepalive](https://github.com/LLM-Red-Team/free-api-hub/#Render%E5%AE%B9%E5%99%A8%E4%BF%9D%E6%B4%BB)**
|
||||||
|
|
||||||
|
1. Fork this project to your GitHub account.
|
||||||
|
|
||||||
|
2. Visit [Render](https://dashboard.render.com/) and log in with your GitHub account.
|
||||||
|
|
||||||
|
3. Build your Web Service (`New+` -> `Build and deploy from a Git repository` -> `Connect your forked project` -> `Select deployment region` -> `Choose instance type as Free` -> `Create Web Service`).
|
||||||
|
|
||||||
|
4. After the build is complete, copy the assigned domain and append the URL to access it.
|
||||||
|
|
||||||
|
### Vercel Deployment
|
||||||
|
|
||||||
|
**Note: Vercel free accounts have a request response timeout of 10 seconds, but interface responses are usually longer, which may result in a 504 timeout error from Vercel!**
|
||||||
|
|
||||||
|
Please ensure that Node.js environment is installed first.
|
||||||
|
|
||||||
|
```shell
|
||||||
|
npm i -g vercel --registry http://registry.npmmirror.com
|
||||||
|
vercel login
|
||||||
|
git clone https://github.com/LLM-Red-Team/glm-free-api
|
||||||
|
cd glm-free-api
|
||||||
|
vercel --prod
|
||||||
|
```
|
||||||
|
|
||||||
|
## Native Deployment
|
||||||
|
|
||||||
|
Please prepare a server with a public IP and open port 8000.
|
||||||
|
|
||||||
|
Please install the Node.js environment and configure the environment variables first, and confirm that the node command is available.
|
||||||
|
|
||||||
|
Install dependencies
|
||||||
|
|
||||||
|
```shell
|
||||||
|
npm i
|
||||||
|
```
|
||||||
|
|
||||||
|
Install PM2 for process guarding
|
||||||
|
|
||||||
|
```shell
|
||||||
|
npm i -g pm2
|
||||||
|
```
|
||||||
|
|
||||||
|
Compile and build. When you see the dist directory, the build is complete.
|
||||||
|
|
||||||
|
```shell
|
||||||
|
npm run build
|
||||||
|
```
|
||||||
|
|
||||||
|
Start service
|
||||||
|
|
||||||
|
```shell
|
||||||
|
pm2 start dist/index.js --name "glm-free-api"
|
||||||
|
```
|
||||||
|
|
||||||
|
View real-time service logs
|
||||||
|
|
||||||
|
```shell
|
||||||
|
pm2 logs glm-free-api
|
||||||
|
```
|
||||||
|
|
||||||
|
Restart service
|
||||||
|
|
||||||
|
```shell
|
||||||
|
pm2 reload glm-free-api
|
||||||
|
```
|
||||||
|
|
||||||
|
Shut down service
|
||||||
|
|
||||||
|
```shell
|
||||||
|
pm2 stop glm-free-api
|
||||||
|
```
|
||||||
|
|
||||||
|
## Recommended Clients
|
||||||
|
|
||||||
|
Using the following second-developed clients for free-api series projects is faster and easier, and supports document/image uploads!
|
||||||
|
|
||||||
|
[Clivia](https://github.com/Yanyutin753/lobe-chat)'s modified LobeChat [https://github.com/Yanyutin753/lobe-chat](https://github.com/Yanyutin753/lobe-chat)
|
||||||
|
|
||||||
|
[Time@](https://github.com/SuYxh)'s modified ChatGPT Web [https://github.com/SuYxh/chatgpt-web-sea](https://github.com/SuYxh/chatgpt-web-sea)
|
||||||
|
|
||||||
|
## interface List
|
||||||
|
|
||||||
|
Currently, the `/v1/chat/completions` interface compatible with openai is supported. You can use the client access interface compatible with openai or other clients, or use online services such as [dify](https://dify.ai/) Access and use.
|
||||||
|
|
||||||
|
### Conversation Completion
|
||||||
|
|
||||||
|
Conversation completion interface, compatible with openai's [chat-completions-api](https://platform.openai.com/docs/guides/text-generation/chat-completions-api).
|
||||||
|
|
||||||
|
**POST /v1/chat/completions**
|
||||||
|
|
||||||
|
The header needs to set the Authorization header:
|
||||||
|
|
||||||
|
```
|
||||||
|
Authorization: Bearer [refresh_token]
|
||||||
|
```
|
||||||
|
|
||||||
|
Request data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
// Default model: glm-4-plus
|
||||||
|
// zero thinking model: glm-4-zero / glm-4-think
|
||||||
|
// If using the Agent, fill in the Agent ID here
|
||||||
|
"model": "glm-4",
|
||||||
|
// Currently, multi-round conversations are realized based on message merging, which in some scenarios may lead to capacity degradation and is limited by the maximum number of tokens in a single round.
|
||||||
|
// If you want a native multi-round dialog experience, you can pass in the ids obtained from the last round of messages to pick up the context
|
||||||
|
// "conversation_id": "65f6c28546bae1f0fbb532de",
|
||||||
|
"messages": [
|
||||||
|
{
|
||||||
|
"role": "user",
|
||||||
|
"content": "Who RU?"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
// If using SSE stream, please set it to true, the default is false
|
||||||
|
"stream": false
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
Response data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"id": "65f6c28546bae1f0fbb532de",
|
||||||
|
"model": "glm-4",
|
||||||
|
"object": "chat.completion",
|
||||||
|
"choices": [
|
||||||
|
{
|
||||||
|
"index": 0,
|
||||||
|
"message": {
|
||||||
|
"role": "assistant",
|
||||||
|
"content": "My name is Zhipu Qingyan."
|
||||||
|
},
|
||||||
|
"finish_reason": "stop"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"usage": {
|
||||||
|
"prompt_tokens": 1,
|
||||||
|
"completion_tokens": 1,
|
||||||
|
"total_tokens": 2
|
||||||
|
},
|
||||||
|
"created": 1710152062
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Video Generation
|
||||||
|
|
||||||
|
Video API
|
||||||
|
|
||||||
|
**If you're not VIP, you will wait in line for a long time.**
|
||||||
|
|
||||||
|
**POST /v1/videos/generations**
|
||||||
|
|
||||||
|
The header needs to set the Authorization header:
|
||||||
|
|
||||||
|
```
|
||||||
|
Authorization: Bearer [refresh_token]
|
||||||
|
```
|
||||||
|
|
||||||
|
Request data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
// 模型名称
|
||||||
|
// cogvideox:默认官方视频模型
|
||||||
|
// cogvideox-pro:先生成图像再作为参考图像生成视频,作为视频首帧引导视频效果,但耗时更长
|
||||||
|
"model": "cogvideox",
|
||||||
|
// 视频生成提示词
|
||||||
|
"prompt": "一只可爱的猫走在花丛中",
|
||||||
|
// 支持使用图像URL或者BASE64_URL作为视频首帧参考图像(如果使用cogvideox-pro则会忽略此参数)
|
||||||
|
// "image_url": "https://sfile.chatglm.cn/testpath/b5341945-3839-522c-b4ab-a6268cb131d5_0.png",
|
||||||
|
// 支持设置视频风格:卡通3D/黑白老照片/油画/电影感
|
||||||
|
// "video_style": "油画",
|
||||||
|
// 支持设置情感氛围:温馨和谐/生动活泼/紧张刺激/凄凉寂寞
|
||||||
|
// "emotional_atmosphere": "生动活泼",
|
||||||
|
// 支持设置运镜方式:水平/垂直/推近/拉远
|
||||||
|
// "mirror_mode": "水平"
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
Response data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"created": 1722103836,
|
||||||
|
"data": [
|
||||||
|
{
|
||||||
|
// 对话ID,目前没啥用
|
||||||
|
"conversation_id": "66a537ec0603e53bccb8900a",
|
||||||
|
// 封面URL
|
||||||
|
"cover_url": "https://sfile.chatglm.cn/testpath/video_cover/c1f59468-32fa-58c3-bd9d-ab4230cfe3ca_cover_0.png",
|
||||||
|
// 视频URL
|
||||||
|
"video_url": "https://sfile.chatglm.cn/testpath/video/c1f59468-32fa-58c3-bd9d-ab4230cfe3ca_0.mp4",
|
||||||
|
// 视频时长
|
||||||
|
"video_duration": "6s",
|
||||||
|
// 视频分辨率
|
||||||
|
"resolution": "1440 × 960"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### AI Drawing
|
||||||
|
|
||||||
|
This format is compatible with the [gpt-4-vision-preview](https://platform.openai.com/docs/guides/vision) API format.
|
||||||
|
|
||||||
|
**POST /v1/images/generations**
|
||||||
|
|
||||||
|
The header needs to set the Authorization header:
|
||||||
|
|
||||||
|
```
|
||||||
|
Authorization: Bearer [refresh_token]
|
||||||
|
```
|
||||||
|
|
||||||
|
Request data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
// 如果使用智能体请填写智能体ID到此处,否则可以乱填
|
||||||
|
"model": "cogview-3",
|
||||||
|
"prompt": "A cute cat"
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
Response data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"created": 1711507449,
|
||||||
|
"data": [
|
||||||
|
{
|
||||||
|
"url": "https://sfile.chatglm.cn/testpath/5e56234b-34ae-593c-ba4e-3f7ba77b5768_0.png"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Document Interpretation
|
||||||
|
|
||||||
|
Provide an accessible file URL or BASE64_URL to parse.
|
||||||
|
|
||||||
|
**POST /v1/chat/completions**
|
||||||
|
|
||||||
|
The header needs to set the Authorization header:
|
||||||
|
|
||||||
|
```
|
||||||
|
Authorization: Bearer [refresh_token]
|
||||||
|
```
|
||||||
|
|
||||||
|
Request data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
// 如果使用智能体请填写智能体ID到此处,否则可以乱填
|
||||||
|
"model": "glm-4",
|
||||||
|
"messages": [
|
||||||
|
{
|
||||||
|
"role": "user",
|
||||||
|
"content": [
|
||||||
|
{
|
||||||
|
"type": "file",
|
||||||
|
"file_url": {
|
||||||
|
"url": "https://mj101-1317487292.cos.ap-shanghai.myqcloud.com/ai/test.pdf"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "text",
|
||||||
|
"text": "文档里说了什么?"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
],
|
||||||
|
// 如果使用SSE流请设置为true,默认false
|
||||||
|
"stream": false
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
Response data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"id": "cnmuo7mcp7f9hjcmihn0",
|
||||||
|
"model": "glm-4",
|
||||||
|
"object": "chat.completion",
|
||||||
|
"choices": [
|
||||||
|
{
|
||||||
|
"index": 0,
|
||||||
|
"message": {
|
||||||
|
"role": "assistant",
|
||||||
|
"content": "根据文档内容,我总结如下:\n\n这是一份关于希腊罗马时期的魔法咒语和仪式的文本,包含几个魔法仪式:\n\n1. 一个涉及面包、仪式场所和特定咒语的仪式,用于使某人爱上你。\n\n2. 一个针对女神赫卡忒的召唤仪式,用来折磨某人直到她自愿来到你身边。\n\n3. 一个通过念诵爱神阿芙罗狄蒂的秘密名字,连续七天进行仪式,来赢得一个美丽女子的心。\n\n4. 一个通过燃烧没药并念诵咒语,让一个女子对你产生强烈欲望的仪式。\n\n这些仪式都带有魔法和迷信色彩,使用各种咒语和象征性行为来影响人的感情和意愿。"
|
||||||
|
},
|
||||||
|
"finish_reason": "stop"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"usage": {
|
||||||
|
"prompt_tokens": 1,
|
||||||
|
"completion_tokens": 1,
|
||||||
|
"total_tokens": 2
|
||||||
|
},
|
||||||
|
"created": 100920
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Image Analysis
|
||||||
|
|
||||||
|
Provide an accessible image URL or BASE64_URL to parse.
|
||||||
|
|
||||||
|
This format is compatible with the [gpt-4-vision-preview](https://platform.openai.com/docs/guides/vision) API format. You can also use this format to transmit documents for parsing.
|
||||||
|
|
||||||
|
**POST /v1/chat/completions**
|
||||||
|
|
||||||
|
The header needs to set the Authorization header:
|
||||||
|
|
||||||
|
```
|
||||||
|
Authorization: Bearer [refresh_token]
|
||||||
|
```
|
||||||
|
|
||||||
|
Request data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"model": "65c046a531d3fcb034918abe",
|
||||||
|
"messages": [
|
||||||
|
{
|
||||||
|
"role": "user",
|
||||||
|
"content": [
|
||||||
|
{
|
||||||
|
"type": "image_url",
|
||||||
|
"image_url": {
|
||||||
|
"url": "http://1255881664.vod2.myqcloud.com/6a0cd388vodbj1255881664/7b97ce1d3270835009240537095/uSfDwh6ZpB0A.png"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "text",
|
||||||
|
"text": "图像描述了什么?"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"stream": false
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
Response data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"id": "65f6c28546bae1f0fbb532de",
|
||||||
|
"model": "glm",
|
||||||
|
"object": "chat.completion",
|
||||||
|
"choices": [
|
||||||
|
{
|
||||||
|
"index": 0,
|
||||||
|
"message": {
|
||||||
|
"role": "assistant",
|
||||||
|
"content": "图片中展示的是一个蓝色背景下的logo,具体地,左边是一个由多个蓝色的圆点组成的圆形图案,右边是“智谱·AI”四个字,字体颜色为蓝色。"
|
||||||
|
},
|
||||||
|
"finish_reason": "stop"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"usage": {
|
||||||
|
"prompt_tokens": 1,
|
||||||
|
"completion_tokens": 1,
|
||||||
|
"total_tokens": 2
|
||||||
|
},
|
||||||
|
"created": 1710670469
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Refresh_token Survival Detection
|
||||||
|
|
||||||
|
Check whether refresh_token is alive. If live is not true, otherwise it is false. Please do not call this interface frequently (less than 10 minutes).
|
||||||
|
|
||||||
|
**POST /token/check**
|
||||||
|
|
||||||
|
Request data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"token": "eyJhbGciOiJIUzUxMiIsInR5cCI6IkpXVCJ9..."
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
Response data:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"live": true
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
## Notification
|
||||||
|
|
||||||
|
### Nginx Anti-generation Optimization
|
||||||
|
|
||||||
|
If you are using Nginx reverse proxy `glm-free-api`, please add the following configuration items to optimize the output effect of the stream and optimize the experience.
|
||||||
|
|
||||||
|
```nginx
|
||||||
|
# Turn off proxy buffering. When set to off, Nginx will immediately send client requests to the backend server and immediately send responses received from the backend server back to the client.
|
||||||
|
proxy_buffering off;
|
||||||
|
# Enable chunked transfer encoding. Chunked transfer encoding allows servers to send data in chunks for dynamically generated content without knowing the size of the content in advance.
|
||||||
|
chunked_transfer_encoding on;
|
||||||
|
# Turn on TCP_NOPUSH, which tells Nginx to send as much data as possible before sending the packet to the client. This is usually used in conjunction with sendfile to improve network efficiency.
|
||||||
|
tcp_nopush on;
|
||||||
|
# Turn on TCP_NODELAY, which tells Nginx not to delay sending data and to send small data packets immediately. In some cases, this can reduce network latency.
|
||||||
|
tcp_nodelay on;
|
||||||
|
#Set the timeout to keep the connection, here it is set to 120 seconds. If there is no further communication between client and server during this time, the connection will be closed.
|
||||||
|
keepalive_timeout 120;
|
||||||
|
```
|
||||||
|
|
||||||
|
### Token Statistics
|
||||||
|
|
||||||
|
Since the inference side is not in glm-free-api, the token cannot be counted and will be returned as a fixed number!!!!!
|
||||||
|
|
||||||
|
## Star History
|
||||||
|
|
||||||
|
[](https://star-history.com/#LLM-Red-Team/glm-free-api&Date)
|
@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "glm-free-api",
|
"name": "glm-free-api",
|
||||||
"version": "0.0.26",
|
"version": "0.0.35",
|
||||||
"description": "GLM Free API Server",
|
"description": "GLM Free API Server",
|
||||||
"type": "module",
|
"type": "module",
|
||||||
"main": "dist/index.js",
|
"main": "dist/index.js",
|
||||||
@ -13,8 +13,8 @@
|
|||||||
"dist/"
|
"dist/"
|
||||||
],
|
],
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"dev": "tsup src/index.ts --format cjs,esm --sourcemap --dts --publicDir public --watch --onSuccess \"node dist/index.js\"",
|
"dev": "tsup src/index.ts --format cjs,esm --sourcemap --dts --publicDir public --watch --onSuccess \"node --enable-source-maps dist/index.js\"",
|
||||||
"start": "node dist/index.js",
|
"start": "node --enable-source-maps dist/index.js",
|
||||||
"build": "tsup src/index.ts --format cjs,esm --sourcemap --dts --clean --publicDir public"
|
"build": "tsup src/index.ts --format cjs,esm --sourcemap --dts --clean --publicDir public"
|
||||||
},
|
},
|
||||||
"author": "Vinlic",
|
"author": "Vinlic",
|
||||||
@ -38,6 +38,7 @@
|
|||||||
"mime": "^4.0.1",
|
"mime": "^4.0.1",
|
||||||
"minimist": "^1.2.8",
|
"minimist": "^1.2.8",
|
||||||
"randomstring": "^1.3.0",
|
"randomstring": "^1.3.0",
|
||||||
|
"sharp": "^0.33.4",
|
||||||
"uuid": "^9.0.1",
|
"uuid": "^9.0.1",
|
||||||
"yaml": "^2.3.4"
|
"yaml": "^2.3.4"
|
||||||
},
|
},
|
||||||
|
@ -7,5 +7,6 @@ export default {
|
|||||||
API_FILE_EXECEEDS_SIZE: [-2004, '远程文件超出大小'],
|
API_FILE_EXECEEDS_SIZE: [-2004, '远程文件超出大小'],
|
||||||
API_CHAT_STREAM_PUSHING: [-2005, '已有对话流正在输出'],
|
API_CHAT_STREAM_PUSHING: [-2005, '已有对话流正在输出'],
|
||||||
API_CONTENT_FILTERED: [-2006, '内容由于合规问题已被阻止生成'],
|
API_CONTENT_FILTERED: [-2006, '内容由于合规问题已被阻止生成'],
|
||||||
API_IMAGE_GENERATION_FAILED: [-2007, '图像生成失败']
|
API_IMAGE_GENERATION_FAILED: [-2007, '图像生成失败'],
|
||||||
|
API_VIDEO_GENERATION_FAILED: [-2008, '视频生成失败'],
|
||||||
}
|
}
|
@ -2,6 +2,8 @@ import { PassThrough } from "stream";
|
|||||||
import path from "path";
|
import path from "path";
|
||||||
import _ from "lodash";
|
import _ from "lodash";
|
||||||
import mime from "mime";
|
import mime from "mime";
|
||||||
|
import sharp from "sharp";
|
||||||
|
import fs from "fs-extra";
|
||||||
import FormData from "form-data";
|
import FormData from "form-data";
|
||||||
import axios, { AxiosResponse } from "axios";
|
import axios, { AxiosResponse } from "axios";
|
||||||
|
|
||||||
@ -15,6 +17,8 @@ import util from "@/lib/util.ts";
|
|||||||
const MODEL_NAME = "glm";
|
const MODEL_NAME = "glm";
|
||||||
// 默认的智能体ID,GLM4
|
// 默认的智能体ID,GLM4
|
||||||
const DEFAULT_ASSISTANT_ID = "65940acff94777010aa6b796";
|
const DEFAULT_ASSISTANT_ID = "65940acff94777010aa6b796";
|
||||||
|
// zero推理模型智能体ID
|
||||||
|
const ZERO_ASSISTANT_ID = "676411c38945bbc58a905d31";
|
||||||
// access_token有效期
|
// access_token有效期
|
||||||
const ACCESS_TOKEN_EXPIRES = 3600;
|
const ACCESS_TOKEN_EXPIRES = 3600;
|
||||||
// 最大重试次数
|
// 最大重试次数
|
||||||
@ -163,14 +167,14 @@ async function removeConversation(
|
|||||||
*
|
*
|
||||||
* @param messages 参考gpt系列消息格式,多轮对话请完整提供上下文
|
* @param messages 参考gpt系列消息格式,多轮对话请完整提供上下文
|
||||||
* @param refreshToken 用于刷新access_token的refresh_token
|
* @param refreshToken 用于刷新access_token的refresh_token
|
||||||
* @param assistantId 智能体ID,默认使用GLM4原版
|
* @param model 智能体ID,默认使用GLM4原版
|
||||||
* @param retryCount 重试次数
|
* @param retryCount 重试次数
|
||||||
*/
|
*/
|
||||||
async function createCompletion(
|
async function createCompletion(
|
||||||
messages: any[],
|
messages: any[],
|
||||||
refreshToken: string,
|
refreshToken: string,
|
||||||
assistantId = DEFAULT_ASSISTANT_ID,
|
model = MODEL_NAME,
|
||||||
refConvId = '',
|
refConvId = "",
|
||||||
retryCount = 0
|
retryCount = 0
|
||||||
) {
|
) {
|
||||||
return (async () => {
|
return (async () => {
|
||||||
@ -180,10 +184,20 @@ async function createCompletion(
|
|||||||
const refFileUrls = extractRefFileUrls(messages);
|
const refFileUrls = extractRefFileUrls(messages);
|
||||||
const refs = refFileUrls.length
|
const refs = refFileUrls.length
|
||||||
? await Promise.all(
|
? await Promise.all(
|
||||||
refFileUrls.map((fileUrl) => uploadFile(fileUrl, refreshToken))
|
refFileUrls.map((fileUrl) => uploadFile(fileUrl, refreshToken))
|
||||||
)
|
)
|
||||||
: [];
|
: [];
|
||||||
|
|
||||||
|
// 如果引用对话ID不正确则重置引用
|
||||||
|
if (!/[0-9a-zA-Z]{24}/.test(refConvId)) refConvId = "";
|
||||||
|
|
||||||
|
let assistantId = /^[a-z0-9]{24,}$/.test(model) ? model : DEFAULT_ASSISTANT_ID;
|
||||||
|
|
||||||
|
if(model.indexOf('think') != -1 || model.indexOf('zero') != -1) {
|
||||||
|
assistantId = ZERO_ASSISTANT_ID;
|
||||||
|
logger.info('使用思考模型');
|
||||||
|
}
|
||||||
|
|
||||||
// 请求流
|
// 请求流
|
||||||
const token = await acquireToken(refreshToken);
|
const token = await acquireToken(refreshToken);
|
||||||
const result = await axios.post(
|
const result = await axios.post(
|
||||||
@ -195,8 +209,11 @@ async function createCompletion(
|
|||||||
meta_data: {
|
meta_data: {
|
||||||
channel: "",
|
channel: "",
|
||||||
draft_id: "",
|
draft_id: "",
|
||||||
|
if_plus_model: true,
|
||||||
input_question_type: "xxxx",
|
input_question_type: "xxxx",
|
||||||
is_test: false,
|
is_test: false,
|
||||||
|
platform: "pc",
|
||||||
|
quote_log_id: ""
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@ -217,7 +234,7 @@ async function createCompletion(
|
|||||||
}
|
}
|
||||||
);
|
);
|
||||||
if (result.headers["content-type"].indexOf("text/event-stream") == -1) {
|
if (result.headers["content-type"].indexOf("text/event-stream") == -1) {
|
||||||
result.data.on("data", buffer => logger.error(buffer.toString()));
|
result.data.on("data", (buffer) => logger.error(buffer.toString()));
|
||||||
throw new APIException(
|
throw new APIException(
|
||||||
EX.API_REQUEST_FAILED,
|
EX.API_REQUEST_FAILED,
|
||||||
`Stream response Content-Type invalid: ${result.headers["content-type"]}`
|
`Stream response Content-Type invalid: ${result.headers["content-type"]}`
|
||||||
@ -226,14 +243,14 @@ async function createCompletion(
|
|||||||
|
|
||||||
const streamStartTime = util.timestamp();
|
const streamStartTime = util.timestamp();
|
||||||
// 接收流为输出文本
|
// 接收流为输出文本
|
||||||
const answer = await receiveStream(result.data);
|
const answer = await receiveStream(model, result.data);
|
||||||
logger.success(
|
logger.success(
|
||||||
`Stream has completed transfer ${util.timestamp() - streamStartTime}ms`
|
`Stream has completed transfer ${util.timestamp() - streamStartTime}ms`
|
||||||
);
|
);
|
||||||
|
|
||||||
// 异步移除会话
|
// 异步移除会话
|
||||||
removeConversation(answer.id, refreshToken, assistantId).catch((err) =>
|
removeConversation(answer.id, refreshToken, assistantId).catch(
|
||||||
!refConvId && console.error(err)
|
(err) => !refConvId && console.error(err)
|
||||||
);
|
);
|
||||||
|
|
||||||
return answer;
|
return answer;
|
||||||
@ -246,7 +263,7 @@ async function createCompletion(
|
|||||||
return createCompletion(
|
return createCompletion(
|
||||||
messages,
|
messages,
|
||||||
refreshToken,
|
refreshToken,
|
||||||
assistantId,
|
model,
|
||||||
refConvId,
|
refConvId,
|
||||||
retryCount + 1
|
retryCount + 1
|
||||||
);
|
);
|
||||||
@ -261,14 +278,14 @@ async function createCompletion(
|
|||||||
*
|
*
|
||||||
* @param messages 参考gpt系列消息格式,多轮对话请完整提供上下文
|
* @param messages 参考gpt系列消息格式,多轮对话请完整提供上下文
|
||||||
* @param refreshToken 用于刷新access_token的refresh_token
|
* @param refreshToken 用于刷新access_token的refresh_token
|
||||||
* @param assistantId 智能体ID,默认使用GLM4原版
|
* @param model 智能体ID,默认使用GLM4原版
|
||||||
* @param retryCount 重试次数
|
* @param retryCount 重试次数
|
||||||
*/
|
*/
|
||||||
async function createCompletionStream(
|
async function createCompletionStream(
|
||||||
messages: any[],
|
messages: any[],
|
||||||
refreshToken: string,
|
refreshToken: string,
|
||||||
assistantId = DEFAULT_ASSISTANT_ID,
|
model = MODEL_NAME,
|
||||||
refConvId = '',
|
refConvId = "",
|
||||||
retryCount = 0
|
retryCount = 0
|
||||||
) {
|
) {
|
||||||
return (async () => {
|
return (async () => {
|
||||||
@ -278,10 +295,20 @@ async function createCompletionStream(
|
|||||||
const refFileUrls = extractRefFileUrls(messages);
|
const refFileUrls = extractRefFileUrls(messages);
|
||||||
const refs = refFileUrls.length
|
const refs = refFileUrls.length
|
||||||
? await Promise.all(
|
? await Promise.all(
|
||||||
refFileUrls.map((fileUrl) => uploadFile(fileUrl, refreshToken))
|
refFileUrls.map((fileUrl) => uploadFile(fileUrl, refreshToken))
|
||||||
)
|
)
|
||||||
: [];
|
: [];
|
||||||
|
|
||||||
|
// 如果引用对话ID不正确则重置引用
|
||||||
|
if (!/[0-9a-zA-Z]{24}/.test(refConvId)) refConvId = "";
|
||||||
|
|
||||||
|
let assistantId = /^[a-z0-9]{24,}$/.test(model) ? model : DEFAULT_ASSISTANT_ID;
|
||||||
|
|
||||||
|
if(model.indexOf('think') != -1 || model.indexOf('zero') != -1) {
|
||||||
|
assistantId = ZERO_ASSISTANT_ID;
|
||||||
|
logger.info('使用思考模型');
|
||||||
|
}
|
||||||
|
|
||||||
// 请求流
|
// 请求流
|
||||||
const token = await acquireToken(refreshToken);
|
const token = await acquireToken(refreshToken);
|
||||||
const result = await axios.post(
|
const result = await axios.post(
|
||||||
@ -293,8 +320,11 @@ async function createCompletionStream(
|
|||||||
meta_data: {
|
meta_data: {
|
||||||
channel: "",
|
channel: "",
|
||||||
draft_id: "",
|
draft_id: "",
|
||||||
|
if_plus_model: true,
|
||||||
input_question_type: "xxxx",
|
input_question_type: "xxxx",
|
||||||
is_test: false,
|
is_test: false,
|
||||||
|
platform: "pc",
|
||||||
|
quote_log_id: ""
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@ -320,7 +350,7 @@ async function createCompletionStream(
|
|||||||
`Invalid response Content-Type:`,
|
`Invalid response Content-Type:`,
|
||||||
result.headers["content-type"]
|
result.headers["content-type"]
|
||||||
);
|
);
|
||||||
result.data.on("data", buffer => logger.error(buffer.toString()));
|
result.data.on("data", (buffer) => logger.error(buffer.toString()));
|
||||||
const transStream = new PassThrough();
|
const transStream = new PassThrough();
|
||||||
transStream.end(
|
transStream.end(
|
||||||
`data: ${JSON.stringify({
|
`data: ${JSON.stringify({
|
||||||
@ -346,13 +376,13 @@ async function createCompletionStream(
|
|||||||
|
|
||||||
const streamStartTime = util.timestamp();
|
const streamStartTime = util.timestamp();
|
||||||
// 创建转换流将消息格式转换为gpt兼容格式
|
// 创建转换流将消息格式转换为gpt兼容格式
|
||||||
return createTransStream(result.data, (convId: string) => {
|
return createTransStream(model, result.data, (convId: string) => {
|
||||||
logger.success(
|
logger.success(
|
||||||
`Stream has completed transfer ${util.timestamp() - streamStartTime}ms`
|
`Stream has completed transfer ${util.timestamp() - streamStartTime}ms`
|
||||||
);
|
);
|
||||||
// 流传输结束后异步移除会话
|
// 流传输结束后异步移除会话
|
||||||
removeConversation(convId, refreshToken, assistantId).catch((err) =>
|
removeConversation(convId, refreshToken, assistantId).catch(
|
||||||
!refConvId && console.error(err)
|
(err) => !refConvId && console.error(err)
|
||||||
);
|
);
|
||||||
});
|
});
|
||||||
})().catch((err) => {
|
})().catch((err) => {
|
||||||
@ -364,7 +394,7 @@ async function createCompletionStream(
|
|||||||
return createCompletionStream(
|
return createCompletionStream(
|
||||||
messages,
|
messages,
|
||||||
refreshToken,
|
refreshToken,
|
||||||
assistantId,
|
model,
|
||||||
refConvId,
|
refConvId,
|
||||||
retryCount + 1
|
retryCount + 1
|
||||||
);
|
);
|
||||||
@ -383,7 +413,10 @@ async function generateImages(
|
|||||||
return (async () => {
|
return (async () => {
|
||||||
logger.info(prompt);
|
logger.info(prompt);
|
||||||
const messages = [
|
const messages = [
|
||||||
{ role: "user", content: prompt.indexOf('画') == -1 ? `请画:${prompt}` : prompt },
|
{
|
||||||
|
role: "user",
|
||||||
|
content: prompt.indexOf("画") == -1 ? `请画:${prompt}` : prompt,
|
||||||
|
},
|
||||||
];
|
];
|
||||||
// 请求流
|
// 请求流
|
||||||
const token = await acquireToken(refreshToken);
|
const token = await acquireToken(refreshToken);
|
||||||
@ -396,8 +429,11 @@ async function generateImages(
|
|||||||
meta_data: {
|
meta_data: {
|
||||||
channel: "",
|
channel: "",
|
||||||
draft_id: "",
|
draft_id: "",
|
||||||
|
if_plus_model: true,
|
||||||
input_question_type: "xxxx",
|
input_question_type: "xxxx",
|
||||||
is_test: false,
|
is_test: false,
|
||||||
|
platform: "pc",
|
||||||
|
quote_log_id: ""
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
@ -450,6 +486,180 @@ async function generateImages(
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async function generateVideos(
|
||||||
|
model = "cogvideox",
|
||||||
|
prompt: string,
|
||||||
|
refreshToken: string,
|
||||||
|
options: {
|
||||||
|
imageUrl: string;
|
||||||
|
videoStyle: string;
|
||||||
|
emotionalAtmosphere: string;
|
||||||
|
mirrorMode: string;
|
||||||
|
audioId: string;
|
||||||
|
},
|
||||||
|
refConvId = "",
|
||||||
|
retryCount = 0
|
||||||
|
) {
|
||||||
|
return (async () => {
|
||||||
|
logger.info(prompt);
|
||||||
|
|
||||||
|
// 如果引用对话ID不正确则重置引用
|
||||||
|
if (!/[0-9a-zA-Z]{24}/.test(refConvId)) refConvId = "";
|
||||||
|
|
||||||
|
const sourceList = [];
|
||||||
|
if (model == "cogvideox-pro") {
|
||||||
|
const imageUrls = await generateImages(undefined, prompt, refreshToken);
|
||||||
|
options.imageUrl = imageUrls[0];
|
||||||
|
}
|
||||||
|
if (options.imageUrl) {
|
||||||
|
const { source_id: sourceId } = await uploadFile(
|
||||||
|
options.imageUrl,
|
||||||
|
refreshToken,
|
||||||
|
true
|
||||||
|
);
|
||||||
|
sourceList.push(sourceId);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 发起生成请求
|
||||||
|
let token = await acquireToken(refreshToken);
|
||||||
|
const result = await axios.post(
|
||||||
|
`https://chatglm.cn/chatglm/video-api/v1/chat`,
|
||||||
|
{
|
||||||
|
conversation_id: refConvId,
|
||||||
|
prompt,
|
||||||
|
source_list: sourceList.length > 0 ? sourceList : undefined,
|
||||||
|
advanced_parameter_extra: {
|
||||||
|
emotional_atmosphere: options.emotionalAtmosphere,
|
||||||
|
mirror_mode: options.mirrorMode,
|
||||||
|
video_style: options.videoStyle,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
headers: {
|
||||||
|
Authorization: `Bearer ${token}`,
|
||||||
|
Referer: "https://chatglm.cn/video",
|
||||||
|
"X-Device-Id": util.uuid(false),
|
||||||
|
"X-Request-Id": util.uuid(false),
|
||||||
|
...FAKE_HEADERS,
|
||||||
|
},
|
||||||
|
// 30秒超时
|
||||||
|
timeout: 30000,
|
||||||
|
validateStatus: () => true,
|
||||||
|
}
|
||||||
|
);
|
||||||
|
const { result: _result } = checkResult(result, refreshToken);
|
||||||
|
const { chat_id: chatId, conversation_id: convId } = _result;
|
||||||
|
|
||||||
|
// 轮询生成进度
|
||||||
|
const startTime = util.unixTimestamp();
|
||||||
|
const results = [];
|
||||||
|
while (true) {
|
||||||
|
if (util.unixTimestamp() - startTime > 600)
|
||||||
|
throw new APIException(EX.API_VIDEO_GENERATION_FAILED);
|
||||||
|
const token = await acquireToken(refreshToken);
|
||||||
|
const result = await axios.get(
|
||||||
|
`https://chatglm.cn/chatglm/video-api/v1/chat/status/${chatId}`,
|
||||||
|
{
|
||||||
|
headers: {
|
||||||
|
Authorization: `Bearer ${token}`,
|
||||||
|
Referer: "https://chatglm.cn/video",
|
||||||
|
"X-Device-Id": util.uuid(false),
|
||||||
|
"X-Request-Id": util.uuid(false),
|
||||||
|
...FAKE_HEADERS,
|
||||||
|
},
|
||||||
|
// 30秒超时
|
||||||
|
timeout: 30000,
|
||||||
|
validateStatus: () => true,
|
||||||
|
}
|
||||||
|
);
|
||||||
|
const { result: _result } = checkResult(result, refreshToken);
|
||||||
|
const {
|
||||||
|
status,
|
||||||
|
msg,
|
||||||
|
plan,
|
||||||
|
cover_url,
|
||||||
|
video_url,
|
||||||
|
video_duration,
|
||||||
|
resolution,
|
||||||
|
} = _result;
|
||||||
|
if (status != "init" && status != "processing") {
|
||||||
|
if (status != "finished")
|
||||||
|
throw new APIException(EX.API_VIDEO_GENERATION_FAILED);
|
||||||
|
let videoUrl = video_url;
|
||||||
|
if (options.audioId) {
|
||||||
|
const [key, id] = options.audioId.split("-");
|
||||||
|
const token = await acquireToken(refreshToken);
|
||||||
|
const result = await axios.post(
|
||||||
|
`https://chatglm.cn/chatglm/video-api/v1/static/composite_video`,
|
||||||
|
{
|
||||||
|
chat_id: chatId,
|
||||||
|
key,
|
||||||
|
audio_id: id,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
headers: {
|
||||||
|
Authorization: `Bearer ${token}`,
|
||||||
|
Referer: "https://chatglm.cn/video",
|
||||||
|
"X-Device-Id": util.uuid(false),
|
||||||
|
"X-Request-Id": util.uuid(false),
|
||||||
|
...FAKE_HEADERS,
|
||||||
|
},
|
||||||
|
// 30秒超时
|
||||||
|
timeout: 30000,
|
||||||
|
validateStatus: () => true,
|
||||||
|
}
|
||||||
|
);
|
||||||
|
const { result: _result } = checkResult(result, refreshToken);
|
||||||
|
videoUrl = _result.url;
|
||||||
|
}
|
||||||
|
results.push({
|
||||||
|
conversation_id: convId,
|
||||||
|
cover_url,
|
||||||
|
video_url: videoUrl,
|
||||||
|
video_duration,
|
||||||
|
resolution,
|
||||||
|
});
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
await new Promise((resolve) => setTimeout(resolve, 1000));
|
||||||
|
}
|
||||||
|
|
||||||
|
//https://chatglm.cn/chatglm/video-api/v1/reference/audio_group
|
||||||
|
|
||||||
|
axios
|
||||||
|
.delete(`https://chatglm.cn/chatglm/video-api/v1/chat/${chatId}`, {
|
||||||
|
headers: {
|
||||||
|
Authorization: `Bearer ${token}`,
|
||||||
|
Referer: "https://chatglm.cn/video",
|
||||||
|
"X-Device-Id": util.uuid(false),
|
||||||
|
"X-Request-Id": util.uuid(false),
|
||||||
|
...FAKE_HEADERS,
|
||||||
|
},
|
||||||
|
validateStatus: () => true,
|
||||||
|
})
|
||||||
|
.catch((err) => logger.error("移除视频生成记录失败:", err));
|
||||||
|
|
||||||
|
return results;
|
||||||
|
})().catch((err) => {
|
||||||
|
if (retryCount < MAX_RETRY_COUNT) {
|
||||||
|
logger.error(`Video generation error: ${err.message}`);
|
||||||
|
logger.warn(`Try again after ${RETRY_DELAY / 1000}s...`);
|
||||||
|
return (async () => {
|
||||||
|
await new Promise((resolve) => setTimeout(resolve, RETRY_DELAY));
|
||||||
|
return generateVideos(
|
||||||
|
model,
|
||||||
|
prompt,
|
||||||
|
refreshToken,
|
||||||
|
options,
|
||||||
|
refConvId,
|
||||||
|
retryCount + 1
|
||||||
|
);
|
||||||
|
})();
|
||||||
|
}
|
||||||
|
throw err;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 提取消息中引用的文件URL
|
* 提取消息中引用的文件URL
|
||||||
*
|
*
|
||||||
@ -496,46 +706,43 @@ function extractRefFileUrls(messages: any[]) {
|
|||||||
* @param isRefConv 是否为引用会话
|
* @param isRefConv 是否为引用会话
|
||||||
*/
|
*/
|
||||||
function messagesPrepare(messages: any[], refs: any[], isRefConv = false) {
|
function messagesPrepare(messages: any[], refs: any[], isRefConv = false) {
|
||||||
// 检查最新消息是否含有"type": "image_url"或"type": "file",如果有则注入消息
|
|
||||||
let latestMessage = messages[messages.length - 1];
|
|
||||||
let hasFileOrImage =
|
|
||||||
Array.isArray(latestMessage.content) &&
|
|
||||||
latestMessage.content.some(
|
|
||||||
(v) => typeof v === "object" && ["file", "image_url"].includes(v["type"])
|
|
||||||
);
|
|
||||||
if (hasFileOrImage) {
|
|
||||||
let newFileMessage = {
|
|
||||||
content: "关注用户最新发送文件和消息",
|
|
||||||
role: "system",
|
|
||||||
};
|
|
||||||
messages.splice(messages.length - 1, 0, newFileMessage);
|
|
||||||
logger.info("注入提升尾部文件注意力system prompt");
|
|
||||||
} else {
|
|
||||||
// 由于注入会导致设定污染,暂时注释
|
|
||||||
// let newTextMessage = {
|
|
||||||
// content: "关注用户最新的消息",
|
|
||||||
// role: "system",
|
|
||||||
// };
|
|
||||||
// messages.splice(messages.length - 1, 0, newTextMessage);
|
|
||||||
// logger.info("注入提升尾部消息注意力system prompt");
|
|
||||||
}
|
|
||||||
|
|
||||||
let content;
|
let content;
|
||||||
if(isRefConv || messages.length < 2) {
|
if (isRefConv || messages.length < 2) {
|
||||||
content = messages.reduce((content, message) => {
|
content = messages.reduce((content, message) => {
|
||||||
if (_.isArray(message.content)) {
|
if (_.isArray(message.content)) {
|
||||||
return (
|
return message.content.reduce((_content, v) => {
|
||||||
message.content.reduce((_content, v) => {
|
if (!_.isObject(v) || v["type"] != "text") return _content;
|
||||||
if (!_.isObject(v) || v["type"] != "text") return _content;
|
return _content + (v["text"] || "") + "\n";
|
||||||
return _content + (v["text"] || "") + "\n";
|
}, content);
|
||||||
}, content)
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
return content + `${message.content}\n`;
|
return content + `${message.content}\n`;
|
||||||
}, "");
|
}, "");
|
||||||
logger.info("\n透传内容:\n" + content);
|
logger.info("\n透传内容:\n" + content);
|
||||||
}
|
} else {
|
||||||
else {
|
// 检查最新消息是否含有"type": "image_url"或"type": "file",如果有则注入消息
|
||||||
|
let latestMessage = messages[messages.length - 1];
|
||||||
|
let hasFileOrImage =
|
||||||
|
Array.isArray(latestMessage.content) &&
|
||||||
|
latestMessage.content.some(
|
||||||
|
(v) =>
|
||||||
|
typeof v === "object" && ["file", "image_url"].includes(v["type"])
|
||||||
|
);
|
||||||
|
if (hasFileOrImage) {
|
||||||
|
let newFileMessage = {
|
||||||
|
content: "关注用户最新发送文件和消息",
|
||||||
|
role: "system",
|
||||||
|
};
|
||||||
|
messages.splice(messages.length - 1, 0, newFileMessage);
|
||||||
|
logger.info("注入提升尾部文件注意力system prompt");
|
||||||
|
} else {
|
||||||
|
// 由于注入会导致设定污染,暂时注释
|
||||||
|
// let newTextMessage = {
|
||||||
|
// content: "关注用户最新的消息",
|
||||||
|
// role: "system",
|
||||||
|
// };
|
||||||
|
// messages.splice(messages.length - 1, 0, newTextMessage);
|
||||||
|
// logger.info("注入提升尾部消息注意力system prompt");
|
||||||
|
}
|
||||||
content = (
|
content = (
|
||||||
messages.reduce((content, message) => {
|
messages.reduce((content, message) => {
|
||||||
const role = message.role
|
const role = message.role
|
||||||
@ -543,12 +750,10 @@ function messagesPrepare(messages: any[], refs: any[], isRefConv = false) {
|
|||||||
.replace("assistant", "<|assistant|>")
|
.replace("assistant", "<|assistant|>")
|
||||||
.replace("user", "<|user|>");
|
.replace("user", "<|user|>");
|
||||||
if (_.isArray(message.content)) {
|
if (_.isArray(message.content)) {
|
||||||
return (
|
return message.content.reduce((_content, v) => {
|
||||||
message.content.reduce((_content, v) => {
|
if (!_.isObject(v) || v["type"] != "text") return _content;
|
||||||
if (!_.isObject(v) || v["type"] != "text") return _content;
|
return _content + (`${role}\n` + v["text"] || "") + "\n";
|
||||||
return _content + (`${role}\n` + v["text"] || "") + "\n";
|
}, content);
|
||||||
}, content)
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
return (content += `${role}\n${message.content}\n`);
|
return (content += `${role}\n${message.content}\n`);
|
||||||
}, "") + "<|assistant|>\n"
|
}, "") + "<|assistant|>\n"
|
||||||
@ -575,19 +780,19 @@ function messagesPrepare(messages: any[], refs: any[], isRefConv = false) {
|
|||||||
...(fileRefs.length == 0
|
...(fileRefs.length == 0
|
||||||
? []
|
? []
|
||||||
: [
|
: [
|
||||||
{
|
{
|
||||||
type: "file",
|
type: "file",
|
||||||
file: fileRefs,
|
file: fileRefs,
|
||||||
},
|
},
|
||||||
]),
|
]),
|
||||||
...(imageRefs.length == 0
|
...(imageRefs.length == 0
|
||||||
? []
|
? []
|
||||||
: [
|
: [
|
||||||
{
|
{
|
||||||
type: "image",
|
type: "image",
|
||||||
image: imageRefs,
|
image: imageRefs,
|
||||||
},
|
},
|
||||||
]),
|
]),
|
||||||
],
|
],
|
||||||
},
|
},
|
||||||
];
|
];
|
||||||
@ -625,8 +830,13 @@ async function checkFileUrl(fileUrl: string) {
|
|||||||
*
|
*
|
||||||
* @param fileUrl 文件URL
|
* @param fileUrl 文件URL
|
||||||
* @param refreshToken 用于刷新access_token的refresh_token
|
* @param refreshToken 用于刷新access_token的refresh_token
|
||||||
|
* @param isVideoImage 是否是用于视频图像
|
||||||
*/
|
*/
|
||||||
async function uploadFile(fileUrl: string, refreshToken: string) {
|
async function uploadFile(
|
||||||
|
fileUrl: string,
|
||||||
|
refreshToken: string,
|
||||||
|
isVideoImage: boolean = false
|
||||||
|
) {
|
||||||
// 预检查远程文件URL可用性
|
// 预检查远程文件URL可用性
|
||||||
await checkFileUrl(fileUrl);
|
await checkFileUrl(fileUrl);
|
||||||
|
|
||||||
@ -653,6 +863,22 @@ async function uploadFile(fileUrl: string, refreshToken: string) {
|
|||||||
// 获取文件的MIME类型
|
// 获取文件的MIME类型
|
||||||
mimeType = mimeType || mime.getType(filename);
|
mimeType = mimeType || mime.getType(filename);
|
||||||
|
|
||||||
|
if (isVideoImage) {
|
||||||
|
const im = sharp(fileData).resize(1440, null, {
|
||||||
|
fit: "inside", // 保持宽高比
|
||||||
|
});
|
||||||
|
const metadata = await im.metadata();
|
||||||
|
const cropHeight = metadata.height > 960 ? 960 : metadata.height;
|
||||||
|
fileData = await im
|
||||||
|
.extract({
|
||||||
|
width: 1440,
|
||||||
|
height: cropHeight,
|
||||||
|
left: 0,
|
||||||
|
top: (metadata.height - cropHeight) / 2,
|
||||||
|
})
|
||||||
|
.toBuffer();
|
||||||
|
}
|
||||||
|
|
||||||
const formData = new FormData();
|
const formData = new FormData();
|
||||||
formData.append("file", fileData, {
|
formData.append("file", fileData, {
|
||||||
filename,
|
filename,
|
||||||
@ -663,7 +889,9 @@ async function uploadFile(fileUrl: string, refreshToken: string) {
|
|||||||
const token = await acquireToken(refreshToken);
|
const token = await acquireToken(refreshToken);
|
||||||
let result = await axios.request({
|
let result = await axios.request({
|
||||||
method: "POST",
|
method: "POST",
|
||||||
url: "https://chatglm.cn/chatglm/backend-api/assistant/file_upload",
|
url: isVideoImage
|
||||||
|
? "https://chatglm.cn/chatglm/video-api/v1/static/upload"
|
||||||
|
: "https://chatglm.cn/chatglm/backend-api/assistant/file_upload",
|
||||||
data: formData,
|
data: formData,
|
||||||
// 100M限制
|
// 100M限制
|
||||||
maxBodyLength: FILE_MAX_SIZE,
|
maxBodyLength: FILE_MAX_SIZE,
|
||||||
@ -671,7 +899,9 @@ async function uploadFile(fileUrl: string, refreshToken: string) {
|
|||||||
timeout: 60000,
|
timeout: 60000,
|
||||||
headers: {
|
headers: {
|
||||||
Authorization: `Bearer ${token}`,
|
Authorization: `Bearer ${token}`,
|
||||||
Referer: `https://chatglm.cn/`,
|
Referer: isVideoImage
|
||||||
|
? "https://chatglm.cn/video"
|
||||||
|
: "https://chatglm.cn/",
|
||||||
...FAKE_HEADERS,
|
...FAKE_HEADERS,
|
||||||
...formData.getHeaders(),
|
...formData.getHeaders(),
|
||||||
},
|
},
|
||||||
@ -699,14 +929,15 @@ function checkResult(result: AxiosResponse, refreshToken: string) {
|
|||||||
/**
|
/**
|
||||||
* 从流接收完整的消息内容
|
* 从流接收完整的消息内容
|
||||||
*
|
*
|
||||||
|
* @param model 模型
|
||||||
* @param stream 消息流
|
* @param stream 消息流
|
||||||
*/
|
*/
|
||||||
async function receiveStream(stream: any): Promise<any> {
|
async function receiveStream(model: string, stream: any): Promise<any> {
|
||||||
return new Promise((resolve, reject) => {
|
return new Promise((resolve, reject) => {
|
||||||
// 消息初始化
|
// 消息初始化
|
||||||
const data = {
|
const data = {
|
||||||
id: "",
|
id: "",
|
||||||
model: MODEL_NAME,
|
model,
|
||||||
object: "chat.completion",
|
object: "chat.completion",
|
||||||
choices: [
|
choices: [
|
||||||
{
|
{
|
||||||
@ -718,12 +949,16 @@ async function receiveStream(stream: any): Promise<any> {
|
|||||||
usage: { prompt_tokens: 1, completion_tokens: 1, total_tokens: 2 },
|
usage: { prompt_tokens: 1, completion_tokens: 1, total_tokens: 2 },
|
||||||
created: util.unixTimestamp(),
|
created: util.unixTimestamp(),
|
||||||
};
|
};
|
||||||
|
const isSilentModel = model.indexOf('silent') != -1;
|
||||||
|
let thinkingText = "";
|
||||||
let toolCall = false;
|
let toolCall = false;
|
||||||
let codeGenerating = false;
|
let codeGenerating = false;
|
||||||
let textChunkLength = 0;
|
let textChunkLength = 0;
|
||||||
let codeTemp = "";
|
let codeTemp = "";
|
||||||
let lastExecutionOutput = "";
|
let lastExecutionOutput = "";
|
||||||
let textOffset = 0;
|
let textOffset = 0;
|
||||||
|
let refContent = "";
|
||||||
|
logger.info(`是否静默模型: ${isSilentModel}`);
|
||||||
const parser = createParser((event) => {
|
const parser = createParser((event) => {
|
||||||
try {
|
try {
|
||||||
if (event.type !== "event") return;
|
if (event.type !== "event") return;
|
||||||
@ -751,6 +986,7 @@ async function receiveStream(stream: any): Promise<any> {
|
|||||||
textChunkLength = 0;
|
textChunkLength = 0;
|
||||||
innerStr += "\n";
|
innerStr += "\n";
|
||||||
}
|
}
|
||||||
|
|
||||||
if (type == "text") {
|
if (type == "text") {
|
||||||
if (toolCall) {
|
if (toolCall) {
|
||||||
innerStr += "\n";
|
innerStr += "\n";
|
||||||
@ -759,20 +995,24 @@ async function receiveStream(stream: any): Promise<any> {
|
|||||||
}
|
}
|
||||||
if (partStatus == "finish") textChunkLength = text.length;
|
if (partStatus == "finish") textChunkLength = text.length;
|
||||||
return innerStr + text;
|
return innerStr + text;
|
||||||
} else if (
|
} else if (type == "text_thinking" && !isSilentModel) {
|
||||||
|
if (toolCall) {
|
||||||
|
innerStr += "\n";
|
||||||
|
textOffset++;
|
||||||
|
toolCall = false;
|
||||||
|
}
|
||||||
|
thinkingText = text;
|
||||||
|
return innerStr;
|
||||||
|
}else if (
|
||||||
type == "quote_result" &&
|
type == "quote_result" &&
|
||||||
status == "finish" &&
|
status == "finish" &&
|
||||||
meta_data &&
|
meta_data &&
|
||||||
_.isArray(meta_data.metadata_list)
|
_.isArray(meta_data.metadata_list) &&
|
||||||
|
!isSilentModel
|
||||||
) {
|
) {
|
||||||
const searchText =
|
refContent = meta_data.metadata_list.reduce((meta, v) => {
|
||||||
meta_data.metadata_list.reduce(
|
return meta + `${v.title} - ${v.url}\n`;
|
||||||
(meta, v) => meta + `检索 ${v.title}(${v.url}) ...`,
|
}, refContent);
|
||||||
""
|
|
||||||
) + "\n";
|
|
||||||
textOffset += searchText.length;
|
|
||||||
toolCall = true;
|
|
||||||
return innerStr + searchText;
|
|
||||||
} else if (
|
} else if (
|
||||||
type == "image" &&
|
type == "image" &&
|
||||||
_.isArray(image) &&
|
_.isArray(image) &&
|
||||||
@ -790,7 +1030,7 @@ async function receiveStream(stream: any): Promise<any> {
|
|||||||
textOffset += imageText.length;
|
textOffset += imageText.length;
|
||||||
toolCall = true;
|
toolCall = true;
|
||||||
return innerStr + imageText;
|
return innerStr + imageText;
|
||||||
} else if (type == "code" && partStatus == "init") {
|
} else if (type == "code" && status == "init") {
|
||||||
let codeHead = "";
|
let codeHead = "";
|
||||||
if (!codeGenerating) {
|
if (!codeGenerating) {
|
||||||
codeGenerating = true;
|
codeGenerating = true;
|
||||||
@ -802,7 +1042,7 @@ async function receiveStream(stream: any): Promise<any> {
|
|||||||
return innerStr + codeHead + chunk;
|
return innerStr + codeHead + chunk;
|
||||||
} else if (
|
} else if (
|
||||||
type == "code" &&
|
type == "code" &&
|
||||||
partStatus == "finish" &&
|
status == "finish" &&
|
||||||
codeGenerating
|
codeGenerating
|
||||||
) {
|
) {
|
||||||
const codeFooter = "\n```\n";
|
const codeFooter = "\n```\n";
|
||||||
@ -813,7 +1053,7 @@ async function receiveStream(stream: any): Promise<any> {
|
|||||||
} else if (
|
} else if (
|
||||||
type == "execution_output" &&
|
type == "execution_output" &&
|
||||||
_.isString(content) &&
|
_.isString(content) &&
|
||||||
partStatus == "done" &&
|
status == "finish" &&
|
||||||
lastExecutionOutput != content
|
lastExecutionOutput != content
|
||||||
) {
|
) {
|
||||||
lastExecutionOutput = content;
|
lastExecutionOutput = content;
|
||||||
@ -831,8 +1071,16 @@ async function receiveStream(stream: any): Promise<any> {
|
|||||||
);
|
);
|
||||||
data.choices[0].message.content += chunk;
|
data.choices[0].message.content += chunk;
|
||||||
} else {
|
} else {
|
||||||
|
if(thinkingText)
|
||||||
|
data.choices[0].message.content = `[思考开始]\n${thinkingText}[思考结束]\n\n${data.choices[0].message.content}`;
|
||||||
data.choices[0].message.content =
|
data.choices[0].message.content =
|
||||||
data.choices[0].message.content.replace(/【\d+†source】/g, "");
|
data.choices[0].message.content.replace(
|
||||||
|
/【\d+†(来源|源|source)】/g,
|
||||||
|
""
|
||||||
|
) +
|
||||||
|
(refContent
|
||||||
|
? `\n\n搜索结果来自:\n${refContent.replace(/\n$/, "")}`
|
||||||
|
: "");
|
||||||
resolve(data);
|
resolve(data);
|
||||||
}
|
}
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
@ -852,18 +1100,22 @@ async function receiveStream(stream: any): Promise<any> {
|
|||||||
*
|
*
|
||||||
* 将流格式转换为gpt兼容流格式
|
* 将流格式转换为gpt兼容流格式
|
||||||
*
|
*
|
||||||
|
* @param model 模型
|
||||||
* @param stream 消息流
|
* @param stream 消息流
|
||||||
* @param endCallback 传输结束回调
|
* @param endCallback 传输结束回调
|
||||||
*/
|
*/
|
||||||
function createTransStream(stream: any, endCallback?: Function) {
|
function createTransStream(model: string, stream: any, endCallback?: Function) {
|
||||||
// 消息创建时间
|
// 消息创建时间
|
||||||
const created = util.unixTimestamp();
|
const created = util.unixTimestamp();
|
||||||
// 创建转换流
|
// 创建转换流
|
||||||
const transStream = new PassThrough();
|
const transStream = new PassThrough();
|
||||||
|
const isSilentModel = model.indexOf('silent') != -1;
|
||||||
let content = "";
|
let content = "";
|
||||||
|
let thinking = false;
|
||||||
let toolCall = false;
|
let toolCall = false;
|
||||||
let codeGenerating = false;
|
let codeGenerating = false;
|
||||||
let textChunkLength = 0;
|
let textChunkLength = 0;
|
||||||
|
let thinkingText = "";
|
||||||
let codeTemp = "";
|
let codeTemp = "";
|
||||||
let lastExecutionOutput = "";
|
let lastExecutionOutput = "";
|
||||||
let textOffset = 0;
|
let textOffset = 0;
|
||||||
@ -871,7 +1123,7 @@ function createTransStream(stream: any, endCallback?: Function) {
|
|||||||
transStream.write(
|
transStream.write(
|
||||||
`data: ${JSON.stringify({
|
`data: ${JSON.stringify({
|
||||||
id: "",
|
id: "",
|
||||||
model: MODEL_NAME,
|
model,
|
||||||
object: "chat.completion.chunk",
|
object: "chat.completion.chunk",
|
||||||
choices: [
|
choices: [
|
||||||
{
|
{
|
||||||
@ -909,6 +1161,11 @@ function createTransStream(stream: any, endCallback?: Function) {
|
|||||||
innerStr += "\n";
|
innerStr += "\n";
|
||||||
}
|
}
|
||||||
if (type == "text") {
|
if (type == "text") {
|
||||||
|
if(thinking) {
|
||||||
|
innerStr += "[思考结束]\n\n"
|
||||||
|
textOffset = thinkingText.length + 8;
|
||||||
|
thinking = false;
|
||||||
|
}
|
||||||
if (toolCall) {
|
if (toolCall) {
|
||||||
innerStr += "\n";
|
innerStr += "\n";
|
||||||
textOffset++;
|
textOffset++;
|
||||||
@ -916,11 +1173,26 @@ function createTransStream(stream: any, endCallback?: Function) {
|
|||||||
}
|
}
|
||||||
if (partStatus == "finish") textChunkLength = text.length;
|
if (partStatus == "finish") textChunkLength = text.length;
|
||||||
return innerStr + text;
|
return innerStr + text;
|
||||||
|
} else if (type == "text_thinking" && !isSilentModel) {
|
||||||
|
if(!thinking) {
|
||||||
|
innerStr += "[思考开始]\n";
|
||||||
|
textOffset = 7;
|
||||||
|
thinking = true;
|
||||||
|
}
|
||||||
|
if (toolCall) {
|
||||||
|
innerStr += "\n";
|
||||||
|
textOffset++;
|
||||||
|
toolCall = false;
|
||||||
|
}
|
||||||
|
if (partStatus == "finish") textChunkLength = text.length;
|
||||||
|
thinkingText += text.substring(thinkingText.length, text.length);
|
||||||
|
return innerStr + text;
|
||||||
} else if (
|
} else if (
|
||||||
type == "quote_result" &&
|
type == "quote_result" &&
|
||||||
status == "finish" &&
|
status == "finish" &&
|
||||||
meta_data &&
|
meta_data &&
|
||||||
_.isArray(meta_data.metadata_list)
|
_.isArray(meta_data.metadata_list) &&
|
||||||
|
!isSilentModel
|
||||||
) {
|
) {
|
||||||
const searchText =
|
const searchText =
|
||||||
meta_data.metadata_list.reduce(
|
meta_data.metadata_list.reduce(
|
||||||
@ -947,7 +1219,7 @@ function createTransStream(stream: any, endCallback?: Function) {
|
|||||||
textOffset += imageText.length;
|
textOffset += imageText.length;
|
||||||
toolCall = true;
|
toolCall = true;
|
||||||
return innerStr + imageText;
|
return innerStr + imageText;
|
||||||
} else if (type == "code" && partStatus == "init") {
|
} else if (type == "code" && status == "init") {
|
||||||
let codeHead = "";
|
let codeHead = "";
|
||||||
if (!codeGenerating) {
|
if (!codeGenerating) {
|
||||||
codeGenerating = true;
|
codeGenerating = true;
|
||||||
@ -959,7 +1231,7 @@ function createTransStream(stream: any, endCallback?: Function) {
|
|||||||
return innerStr + codeHead + chunk;
|
return innerStr + codeHead + chunk;
|
||||||
} else if (
|
} else if (
|
||||||
type == "code" &&
|
type == "code" &&
|
||||||
partStatus == "finish" &&
|
status == "finish" &&
|
||||||
codeGenerating
|
codeGenerating
|
||||||
) {
|
) {
|
||||||
const codeFooter = "\n```\n";
|
const codeFooter = "\n```\n";
|
||||||
@ -970,7 +1242,7 @@ function createTransStream(stream: any, endCallback?: Function) {
|
|||||||
} else if (
|
} else if (
|
||||||
type == "execution_output" &&
|
type == "execution_output" &&
|
||||||
_.isString(content) &&
|
_.isString(content) &&
|
||||||
partStatus == "done" &&
|
status == "finish" &&
|
||||||
lastExecutionOutput != content
|
lastExecutionOutput != content
|
||||||
) {
|
) {
|
||||||
lastExecutionOutput = content;
|
lastExecutionOutput = content;
|
||||||
@ -1005,8 +1277,8 @@ function createTransStream(stream: any, endCallback?: Function) {
|
|||||||
index: 0,
|
index: 0,
|
||||||
delta:
|
delta:
|
||||||
result.status == "intervene" &&
|
result.status == "intervene" &&
|
||||||
result.last_error &&
|
result.last_error &&
|
||||||
result.last_error.intervene_text
|
result.last_error.intervene_text
|
||||||
? { content: `\n\n${result.last_error.intervene_text}` }
|
? { content: `\n\n${result.last_error.intervene_text}` }
|
||||||
: {},
|
: {},
|
||||||
finish_reason: "stop",
|
finish_reason: "stop",
|
||||||
@ -1061,10 +1333,10 @@ async function receiveImages(
|
|||||||
throw new APIException(EX.API_CONTENT_FILTERED);
|
throw new APIException(EX.API_CONTENT_FILTERED);
|
||||||
if (result.status != "finish") {
|
if (result.status != "finish") {
|
||||||
result.parts.forEach((part) => {
|
result.parts.forEach((part) => {
|
||||||
const { content } = part;
|
const { status: partStatus, content } = part;
|
||||||
if (!_.isArray(content)) return;
|
if (!_.isArray(content)) return;
|
||||||
content.forEach((value) => {
|
content.forEach((value) => {
|
||||||
const { status: partStatus, type, image, text } = value;
|
const { type, image, text } = value;
|
||||||
if (
|
if (
|
||||||
type == "image" &&
|
type == "image" &&
|
||||||
_.isArray(image) &&
|
_.isArray(image) &&
|
||||||
@ -1079,16 +1351,12 @@ async function receiveImages(
|
|||||||
imageUrls.push(value.image_url);
|
imageUrls.push(value.image_url);
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
if (
|
if (type == "text" && partStatus == "finish") {
|
||||||
type == "text" &&
|
|
||||||
partStatus == "finish"
|
|
||||||
) {
|
|
||||||
const urlPattern = /\((https?:\/\/\S+)\)/g;
|
const urlPattern = /\((https?:\/\/\S+)\)/g;
|
||||||
let match;
|
let match;
|
||||||
while ((match = urlPattern.exec(text)) !== null) {
|
while ((match = urlPattern.exec(text)) !== null) {
|
||||||
const url = match[1];
|
const url = match[1];
|
||||||
if (imageUrls.indexOf(url) == -1)
|
if (imageUrls.indexOf(url) == -1) imageUrls.push(url);
|
||||||
imageUrls.push(url);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
@ -1165,8 +1433,7 @@ async function getTokenLiveStatus(refreshToken: string) {
|
|||||||
const { result: _result } = checkResult(result, refreshToken);
|
const { result: _result } = checkResult(result, refreshToken);
|
||||||
const { accessToken } = _result;
|
const { accessToken } = _result;
|
||||||
return !!accessToken;
|
return !!accessToken;
|
||||||
}
|
} catch (err) {
|
||||||
catch (err) {
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -1175,6 +1442,7 @@ export default {
|
|||||||
createCompletion,
|
createCompletion,
|
||||||
createCompletionStream,
|
createCompletionStream,
|
||||||
generateImages,
|
generateImages,
|
||||||
|
generateVideos,
|
||||||
getTokenLiveStatus,
|
getTokenLiveStatus,
|
||||||
tokenSplit,
|
tokenSplit,
|
||||||
};
|
};
|
||||||
|
@ -5,6 +5,9 @@ import Response from '@/lib/response/Response.ts';
|
|||||||
import chat from '@/api/controllers/chat.ts';
|
import chat from '@/api/controllers/chat.ts';
|
||||||
import logger from '@/lib/logger.ts';
|
import logger from '@/lib/logger.ts';
|
||||||
|
|
||||||
|
// zero推理模型智能体ID
|
||||||
|
const ZERO_ASSISTANT_ID = "676411c38945bbc58a905d31";
|
||||||
|
|
||||||
export default {
|
export default {
|
||||||
|
|
||||||
prefix: '/v1/chat',
|
prefix: '/v1/chat',
|
||||||
@ -21,15 +24,15 @@ export default {
|
|||||||
// 随机挑选一个refresh_token
|
// 随机挑选一个refresh_token
|
||||||
const token = _.sample(tokens);
|
const token = _.sample(tokens);
|
||||||
const { model, conversation_id: convId, messages, stream } = request.body;
|
const { model, conversation_id: convId, messages, stream } = request.body;
|
||||||
const assistantId = /^[a-z0-9]{24,}$/.test(model) ? model : undefined
|
|
||||||
if (stream) {
|
if (stream) {
|
||||||
const stream = await chat.createCompletionStream(messages, token, assistantId, convId);
|
const stream = await chat.createCompletionStream(messages, token, model, convId);
|
||||||
return new Response(stream, {
|
return new Response(stream, {
|
||||||
type: "text/event-stream"
|
type: "text/event-stream"
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
return await chat.createCompletion(messages, token, assistantId, convId);
|
return await chat.createCompletion(messages, token, model, convId);
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -3,6 +3,7 @@ import fs from 'fs-extra';
|
|||||||
import Response from '@/lib/response/Response.ts';
|
import Response from '@/lib/response/Response.ts';
|
||||||
import chat from "./chat.ts";
|
import chat from "./chat.ts";
|
||||||
import images from "./images.ts";
|
import images from "./images.ts";
|
||||||
|
import videos from './videos.ts';
|
||||||
import ping from "./ping.ts";
|
import ping from "./ping.ts";
|
||||||
import token from './token.js';
|
import token from './token.js';
|
||||||
import models from './models.ts';
|
import models from './models.ts';
|
||||||
@ -23,6 +24,7 @@ export default [
|
|||||||
},
|
},
|
||||||
chat,
|
chat,
|
||||||
images,
|
images,
|
||||||
|
videos,
|
||||||
ping,
|
ping,
|
||||||
token,
|
token,
|
||||||
models
|
models
|
||||||
|
@ -18,6 +18,11 @@ export default {
|
|||||||
"object": "model",
|
"object": "model",
|
||||||
"owned_by": "glm-free-api"
|
"owned_by": "glm-free-api"
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
"id": "glm-4-plus",
|
||||||
|
"object": "model",
|
||||||
|
"owned_by": "glm-free-api"
|
||||||
|
},
|
||||||
{
|
{
|
||||||
"id": "glm-4v",
|
"id": "glm-4v",
|
||||||
"object": "model",
|
"object": "model",
|
||||||
|
78
src/api/routes/videos.ts
Normal file
78
src/api/routes/videos.ts
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
import _ from "lodash";
|
||||||
|
|
||||||
|
import Request from "@/lib/request/Request.ts";
|
||||||
|
import chat from "@/api/controllers/chat.ts";
|
||||||
|
import util from "@/lib/util.ts";
|
||||||
|
|
||||||
|
export default {
|
||||||
|
|
||||||
|
prefix: "/v1/videos",
|
||||||
|
|
||||||
|
post: {
|
||||||
|
|
||||||
|
"/generations": async (request: Request) => {
|
||||||
|
request
|
||||||
|
.validate(
|
||||||
|
"body.conversation_id",
|
||||||
|
(v) => _.isUndefined(v) || _.isString(v)
|
||||||
|
)
|
||||||
|
.validate("body.model", (v) => _.isUndefined(v) || _.isString(v))
|
||||||
|
.validate("body.prompt", _.isString)
|
||||||
|
.validate("body.audio_id", (v) => _.isUndefined(v) || _.isString(v))
|
||||||
|
.validate("body.image_url", (v) => _.isUndefined(v) || _.isString(v))
|
||||||
|
.validate(
|
||||||
|
"body.video_style",
|
||||||
|
(v) =>
|
||||||
|
_.isUndefined(v) ||
|
||||||
|
["卡通3D", "黑白老照片", "油画", "电影感"].includes(v),
|
||||||
|
"video_style must be one of 卡通3D/黑白老照片/油画/电影感"
|
||||||
|
)
|
||||||
|
.validate(
|
||||||
|
"body.emotional_atmosphere",
|
||||||
|
(v) =>
|
||||||
|
_.isUndefined(v) ||
|
||||||
|
["温馨和谐", "生动活泼", "紧张刺激", "凄凉寂寞"].includes(v),
|
||||||
|
"emotional_atmosphere must be one of 温馨和谐/生动活泼/紧张刺激/凄凉寂寞"
|
||||||
|
)
|
||||||
|
.validate(
|
||||||
|
"body.mirror_mode",
|
||||||
|
(v) =>
|
||||||
|
_.isUndefined(v) || ["水平", "垂直", "推近", "拉远"].includes(v),
|
||||||
|
"mirror_mode must be one of 水平/垂直/推近/拉远"
|
||||||
|
)
|
||||||
|
.validate("headers.authorization", _.isString);
|
||||||
|
// refresh_token切分
|
||||||
|
const tokens = chat.tokenSplit(request.headers.authorization);
|
||||||
|
// 随机挑选一个refresh_token
|
||||||
|
const token = _.sample(tokens);
|
||||||
|
const {
|
||||||
|
model,
|
||||||
|
conversation_id: convId,
|
||||||
|
prompt,
|
||||||
|
image_url: imageUrl,
|
||||||
|
video_style: videoStyle = "",
|
||||||
|
emotional_atmosphere: emotionalAtmosphere = "",
|
||||||
|
mirror_mode: mirrorMode = "",
|
||||||
|
audio_id: audioId,
|
||||||
|
} = request.body;
|
||||||
|
const data = await chat.generateVideos(
|
||||||
|
model,
|
||||||
|
prompt,
|
||||||
|
token,
|
||||||
|
{
|
||||||
|
imageUrl,
|
||||||
|
videoStyle,
|
||||||
|
emotionalAtmosphere,
|
||||||
|
mirrorMode,
|
||||||
|
audioId,
|
||||||
|
},
|
||||||
|
convId
|
||||||
|
);
|
||||||
|
return {
|
||||||
|
created: util.unixTimestamp(),
|
||||||
|
data,
|
||||||
|
};
|
||||||
|
},
|
||||||
|
},
|
||||||
|
|
||||||
|
};
|
@ -52,7 +52,7 @@ export default class Request {
|
|||||||
this.time = Number(_.defaultTo(time, util.timestamp()));
|
this.time = Number(_.defaultTo(time, util.timestamp()));
|
||||||
}
|
}
|
||||||
|
|
||||||
validate(key: string, fn?: Function) {
|
validate(key: string, fn?: Function, message?: string) {
|
||||||
try {
|
try {
|
||||||
const value = _.get(this, key);
|
const value = _.get(this, key);
|
||||||
if (fn) {
|
if (fn) {
|
||||||
@ -64,7 +64,7 @@ export default class Request {
|
|||||||
}
|
}
|
||||||
catch (err) {
|
catch (err) {
|
||||||
logger.warn(`Params ${key} invalid:`, err);
|
logger.warn(`Params ${key} invalid:`, err);
|
||||||
throw new APIException(EX.API_REQUEST_PARAMS_INVALID, `Params ${key} invalid`);
|
throw new APIException(EX.API_REQUEST_PARAMS_INVALID, message || `Params ${key} invalid`);
|
||||||
}
|
}
|
||||||
return this;
|
return this;
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user