推荐个迁移 openai api 比较便捷的方式

10次阅读

共计 894 个字符,预计需要花费 3 分钟才能阅读完成。

用的一个开源 api 网关 higress,可以对接基本国内所有的模型厂商

以通义千问为例,先启动网关:

# 创建一个工作目录
mkdir higress; cd higress
# 启动 higress,配置文件会写到工作目录下
docker run -d --rm --name higress-ai -v ${PWD}:/data 
        -p 8001:8001 -p 8080:8080 
        -e CONFIG_TEMPLATE=ai-proxy -e DEFAULT_AI_SERVICE=qwen 
        -e DASHSCOPE_API_KEY= 这里填通义千问 APIKey 
        higress-registry.cn-hangzhou.cr.aliyuncs.com/higress/all-in-one:1.4.1

然后代码里调整下 base url 就行了:

from openai import OpenAI

client = OpenAI(
    api_key="your-api-key",
    base_url="http://127.0.0.1:8080/v1"
)

# 不需要修改模型,higress 做了等价映射
stream = client.chat.completions.create(
    model="gpt-4o",
    messages=[{"role": "user", "content": "你是谁"}],
    stream=True,
)

for chunk in stream:
    if chunk.choices[0].delta.content:
        print(chunk.choices[0].delta.content)

还能进这个 api 网关的控制台:http://127.0.0.1:8001

可以看到默认的模型映射规则,自己可以调整:

推荐个迁移 openai api 比较便捷的方式

插件配置还可以开启联网搜索:

provider:
  type: qwen
  apiTokens:
  - "通义千问 API Key"
  modelMapping:
  # 映射配置省略
  ...
  # 开启联网搜索
  qwenEnableSearch: true

用 lobechat 对接的效果:

推荐个迁移 openai api 比较便捷的方式

对接 lobechat 参考的这个:https://github.com/alibaba/higress/issues/1023

正文完
 0