custom-provider-pi 0.1.7 → 0.1.8
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/custom-provider.ts +4 -2
- package/package.json +1 -1
package/custom-provider.ts
CHANGED
|
@@ -1146,6 +1146,8 @@ export default function customProviderExtension(pi: ExtensionAPI) {
|
|
|
1146
1146
|
// ---- LB key 轮询 ----
|
|
1147
1147
|
const poolName = event.headers["X-LB-POOL"] as string | undefined;
|
|
1148
1148
|
if (!poolName) return;
|
|
1149
|
+
// 保留 pool name 到响应阶段（after_provider_response 需要它做冷却回调）
|
|
1150
|
+
event.headers["X-LB-POOL-ID"] = poolName;
|
|
1149
1151
|
delete event.headers["X-LB-POOL"];
|
|
1150
1152
|
|
|
1151
1153
|
const pool = lbPools.get(poolName);
|
|
@@ -1167,8 +1169,8 @@ export default function customProviderExtension(pi: ExtensionAPI) {
|
|
|
1167
1169
|
|
|
1168
1170
|
pi.on("after_provider_response", (event) => {
|
|
1169
1171
|
// 从 headers 中读取 key 索引（before_provider_headers 中设置）
|
|
1170
|
-
const poolName = event.headers["
|
|
1171
|
-
const keyIdxStr = event.headers["
|
|
1172
|
+
const poolName = event.headers["X-LB-POOL-ID"];
|
|
1173
|
+
const keyIdxStr = event.headers["X-LB-KEY-INDEX"];
|
|
1172
1174
|
if (!poolName || !keyIdxStr) return;
|
|
1173
1175
|
|
|
1174
1176
|
const pool = lbPools.get(poolName);
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "custom-provider-pi",
|
|
3
|
-
"version": "0.1.
|
|
3
|
+
"version": "0.1.8",
|
|
4
4
|
"description": "pi 扩展：统一管理第三方模型 Provider。子命令体系 /custom-provider add|remove|refresh|list|test|config|enable|disable|prune，支持交互向导、flags/JSON 非交互添加、双协议混用（OpenAI + Anthropic）、模型关键字过滤与修剪、per-provider 代理配置、负载均衡",
|
|
5
5
|
"keywords": [
|
|
6
6
|
"pi-package",
|