camo-gemini 0.1.0
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/LICENSE +21 -0
- package/README.md +242 -0
- package/dist/client/camofox-client.d.ts +21 -0
- package/dist/client/camofox-client.d.ts.map +1 -0
- package/dist/client/camofox-client.js +260 -0
- package/dist/client/camofox-client.js.map +1 -0
- package/dist/config.d.ts +3 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +94 -0
- package/dist/config.js.map +1 -0
- package/dist/core/browser-js.d.ts +27 -0
- package/dist/core/browser-js.d.ts.map +1 -0
- package/dist/core/browser-js.js +172 -0
- package/dist/core/browser-js.js.map +1 -0
- package/dist/core/constants.d.ts +78 -0
- package/dist/core/constants.d.ts.map +1 -0
- package/dist/core/constants.js +115 -0
- package/dist/core/constants.js.map +1 -0
- package/dist/core/error-messages.d.ts +8 -0
- package/dist/core/error-messages.d.ts.map +1 -0
- package/dist/core/error-messages.js +121 -0
- package/dist/core/error-messages.js.map +1 -0
- package/dist/core/failover.d.ts +21 -0
- package/dist/core/failover.d.ts.map +1 -0
- package/dist/core/failover.js +47 -0
- package/dist/core/failover.js.map +1 -0
- package/dist/core/logger.d.ts +13 -0
- package/dist/core/logger.d.ts.map +1 -0
- package/dist/core/logger.js +37 -0
- package/dist/core/logger.js.map +1 -0
- package/dist/core/request-builder.d.ts +20 -0
- package/dist/core/request-builder.d.ts.map +1 -0
- package/dist/core/request-builder.js +210 -0
- package/dist/core/request-builder.js.map +1 -0
- package/dist/core/response-parser.d.ts +12 -0
- package/dist/core/response-parser.d.ts.map +1 -0
- package/dist/core/response-parser.js +221 -0
- package/dist/core/response-parser.js.map +1 -0
- package/dist/core/retry.d.ts +10 -0
- package/dist/core/retry.d.ts.map +1 -0
- package/dist/core/retry.js +132 -0
- package/dist/core/retry.js.map +1 -0
- package/dist/core/stream-parser.d.ts +9 -0
- package/dist/core/stream-parser.d.ts.map +1 -0
- package/dist/core/stream-parser.js +69 -0
- package/dist/core/stream-parser.js.map +1 -0
- package/dist/core/token-manager.d.ts +9 -0
- package/dist/core/token-manager.d.ts.map +1 -0
- package/dist/core/token-manager.js +64 -0
- package/dist/core/token-manager.js.map +1 -0
- package/dist/core/utils.d.ts +2 -0
- package/dist/core/utils.d.ts.map +1 -0
- package/dist/core/utils.js +15 -0
- package/dist/core/utils.js.map +1 -0
- package/dist/dashboard/server.d.ts +31 -0
- package/dist/dashboard/server.d.ts.map +1 -0
- package/dist/dashboard/server.js +190 -0
- package/dist/dashboard/server.js.map +1 -0
- package/dist/errors.d.ts +32 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +59 -0
- package/dist/errors.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +36 -0
- package/dist/index.js.map +1 -0
- package/dist/server.d.ts +31 -0
- package/dist/server.d.ts.map +1 -0
- package/dist/server.js +54 -0
- package/dist/server.js.map +1 -0
- package/dist/services/account.d.ts +16 -0
- package/dist/services/account.d.ts.map +1 -0
- package/dist/services/account.js +55 -0
- package/dist/services/account.js.map +1 -0
- package/dist/services/auth.d.ts +39 -0
- package/dist/services/auth.d.ts.map +1 -0
- package/dist/services/auth.js +285 -0
- package/dist/services/auth.js.map +1 -0
- package/dist/services/chat.d.ts +18 -0
- package/dist/services/chat.d.ts.map +1 -0
- package/dist/services/chat.js +151 -0
- package/dist/services/chat.js.map +1 -0
- package/dist/services/cookie-rotation.d.ts +25 -0
- package/dist/services/cookie-rotation.d.ts.map +1 -0
- package/dist/services/cookie-rotation.js +138 -0
- package/dist/services/cookie-rotation.js.map +1 -0
- package/dist/services/gems.d.ts +33 -0
- package/dist/services/gems.d.ts.map +1 -0
- package/dist/services/gems.js +202 -0
- package/dist/services/gems.js.map +1 -0
- package/dist/services/generate.d.ts +27 -0
- package/dist/services/generate.d.ts.map +1 -0
- package/dist/services/generate.js +149 -0
- package/dist/services/generate.js.map +1 -0
- package/dist/services/health.d.ts +20 -0
- package/dist/services/health.d.ts.map +1 -0
- package/dist/services/health.js +77 -0
- package/dist/services/health.js.map +1 -0
- package/dist/services/upload.d.ts +17 -0
- package/dist/services/upload.d.ts.map +1 -0
- package/dist/services/upload.js +120 -0
- package/dist/services/upload.js.map +1 -0
- package/dist/setup.d.ts +7 -0
- package/dist/setup.d.ts.map +1 -0
- package/dist/setup.js +165 -0
- package/dist/setup.js.map +1 -0
- package/dist/state.d.ts +57 -0
- package/dist/state.d.ts.map +1 -0
- package/dist/state.js +260 -0
- package/dist/state.js.map +1 -0
- package/dist/tools/account.d.ts +4 -0
- package/dist/tools/account.d.ts.map +1 -0
- package/dist/tools/account.js +29 -0
- package/dist/tools/account.js.map +1 -0
- package/dist/tools/auth.d.ts +4 -0
- package/dist/tools/auth.d.ts.map +1 -0
- package/dist/tools/auth.js +56 -0
- package/dist/tools/auth.js.map +1 -0
- package/dist/tools/chat.d.ts +4 -0
- package/dist/tools/chat.d.ts.map +1 -0
- package/dist/tools/chat.js +101 -0
- package/dist/tools/chat.js.map +1 -0
- package/dist/tools/gems.d.ts +4 -0
- package/dist/tools/gems.d.ts.map +1 -0
- package/dist/tools/gems.js +93 -0
- package/dist/tools/gems.js.map +1 -0
- package/dist/tools/health.d.ts +4 -0
- package/dist/tools/health.d.ts.map +1 -0
- package/dist/tools/health.js +13 -0
- package/dist/tools/health.js.map +1 -0
- package/dist/tools/media.d.ts +4 -0
- package/dist/tools/media.d.ts.map +1 -0
- package/dist/tools/media.js +65 -0
- package/dist/tools/media.js.map +1 -0
- package/dist/types.d.ts +292 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/package.json +66 -0
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
export declare enum LogLevel {
|
|
2
|
+
DEBUG = 0,
|
|
3
|
+
INFO = 1,
|
|
4
|
+
WARN = 2,
|
|
5
|
+
ERROR = 3
|
|
6
|
+
}
|
|
7
|
+
export declare const logger: {
|
|
8
|
+
debug: (label: string, msg: string, data?: Record<string, unknown>) => void;
|
|
9
|
+
info: (label: string, msg: string, data?: Record<string, unknown>) => void;
|
|
10
|
+
warn: (label: string, msg: string, data?: Record<string, unknown>) => void;
|
|
11
|
+
error: (label: string, msg: string, data?: Record<string, unknown>) => void;
|
|
12
|
+
};
|
|
13
|
+
//# sourceMappingURL=logger.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../src/core/logger.ts"],"names":[],"mappings":"AAAA,oBAAY,QAAQ;IAClB,KAAK,IAAI;IACT,IAAI,IAAI;IACR,IAAI,IAAI;IACR,KAAK,IAAI;CACV;AA0BD,eAAO,MAAM,MAAM;mBACF,MAAM,OAAO,MAAM,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;kBACpD,MAAM,OAAO,MAAM,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;kBACnD,MAAM,OAAO,MAAM,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;mBAClD,MAAM,OAAO,MAAM,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;CACnE,CAAC"}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
export var LogLevel;
|
|
2
|
+
(function (LogLevel) {
|
|
3
|
+
LogLevel[LogLevel["DEBUG"] = 0] = "DEBUG";
|
|
4
|
+
LogLevel[LogLevel["INFO"] = 1] = "INFO";
|
|
5
|
+
LogLevel[LogLevel["WARN"] = 2] = "WARN";
|
|
6
|
+
LogLevel[LogLevel["ERROR"] = 3] = "ERROR";
|
|
7
|
+
})(LogLevel || (LogLevel = {}));
|
|
8
|
+
const LOG_LEVEL = (() => {
|
|
9
|
+
const level = process.env.CAMOGEMINI_LOG_LEVEL?.toUpperCase();
|
|
10
|
+
switch (level) {
|
|
11
|
+
case "DEBUG":
|
|
12
|
+
return LogLevel.DEBUG;
|
|
13
|
+
case "INFO":
|
|
14
|
+
return LogLevel.INFO;
|
|
15
|
+
case "WARN":
|
|
16
|
+
return LogLevel.WARN;
|
|
17
|
+
case "ERROR":
|
|
18
|
+
return LogLevel.ERROR;
|
|
19
|
+
default:
|
|
20
|
+
return LogLevel.INFO;
|
|
21
|
+
}
|
|
22
|
+
})();
|
|
23
|
+
function log(level, label, message, data) {
|
|
24
|
+
if (level < LOG_LEVEL)
|
|
25
|
+
return;
|
|
26
|
+
const timestamp = new Date().toISOString();
|
|
27
|
+
const levelName = LogLevel[level];
|
|
28
|
+
const dataStr = data ? ` ${JSON.stringify(data)}` : "";
|
|
29
|
+
process.stderr.write(`[${timestamp}] [${levelName}] [${label}] ${message}${dataStr}\n`);
|
|
30
|
+
}
|
|
31
|
+
export const logger = {
|
|
32
|
+
debug: (label, msg, data) => log(LogLevel.DEBUG, label, msg, data),
|
|
33
|
+
info: (label, msg, data) => log(LogLevel.INFO, label, msg, data),
|
|
34
|
+
warn: (label, msg, data) => log(LogLevel.WARN, label, msg, data),
|
|
35
|
+
error: (label, msg, data) => log(LogLevel.ERROR, label, msg, data),
|
|
36
|
+
};
|
|
37
|
+
//# sourceMappingURL=logger.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"logger.js","sourceRoot":"","sources":["../../src/core/logger.ts"],"names":[],"mappings":"AAAA,MAAM,CAAN,IAAY,QAKX;AALD,WAAY,QAAQ;IAClB,yCAAS,CAAA;IACT,uCAAQ,CAAA;IACR,uCAAQ,CAAA;IACR,yCAAS,CAAA;AACX,CAAC,EALW,QAAQ,KAAR,QAAQ,QAKnB;AAED,MAAM,SAAS,GAAG,CAAC,GAAG,EAAE;IACtB,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,WAAW,EAAE,CAAC;IAC9D,QAAQ,KAAK,EAAE,CAAC;QACd,KAAK,OAAO;YACV,OAAO,QAAQ,CAAC,KAAK,CAAC;QACxB,KAAK,MAAM;YACT,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,KAAK,MAAM;YACT,OAAO,QAAQ,CAAC,IAAI,CAAC;QACvB,KAAK,OAAO;YACV,OAAO,QAAQ,CAAC,KAAK,CAAC;QACxB;YACE,OAAO,QAAQ,CAAC,IAAI,CAAC;IACzB,CAAC;AACH,CAAC,CAAC,EAAE,CAAC;AAEL,SAAS,GAAG,CAAC,KAAe,EAAE,KAAa,EAAE,OAAe,EAAE,IAA8B;IAC1F,IAAI,KAAK,GAAG,SAAS;QAAE,OAAO;IAC9B,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC3C,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAClC,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACvD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,SAAS,MAAM,SAAS,MAAM,KAAK,KAAK,OAAO,GAAG,OAAO,IAAI,CAAC,CAAC;AAC1F,CAAC;AAED,MAAM,CAAC,MAAM,MAAM,GAAG;IACpB,KAAK,EAAE,CAAC,KAAa,EAAE,GAAW,EAAE,IAA8B,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,CAAC;IAC5G,IAAI,EAAE,CAAC,KAAa,EAAE,GAAW,EAAE,IAA8B,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,CAAC;IAC1G,IAAI,EAAE,CAAC,KAAa,EAAE,GAAW,EAAE,IAA8B,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,CAAC;IAC1G,KAAK,EAAE,CAAC,KAAa,EAAE,GAAW,EAAE,IAA8B,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,CAAC;CAC7G,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { ChatMetadata, GenerateOptions, GenerateRequest, GemCreateOptions, GemUpdateOptions, GeminiTokens, RPCPayload, RequestPayload, UploadResult } from "../types.js";
|
|
2
|
+
export declare class RequestBuilder {
|
|
3
|
+
private _reqid;
|
|
4
|
+
constructor();
|
|
5
|
+
resetReqId(): void;
|
|
6
|
+
get reqId(): number;
|
|
7
|
+
static randomReqId(): number;
|
|
8
|
+
buildGeneratePayload(request: GenerateRequest, tokens: GeminiTokens, accountIndex?: number, chatMetadata?: ChatMetadata): RequestPayload;
|
|
9
|
+
buildBatchPayload(payloads: RPCPayload[], tokens: GeminiTokens, accountIndex?: number): RequestPayload;
|
|
10
|
+
buildBardActivityPayload(tokens: GeminiTokens, accountIndex?: number): RequestPayload;
|
|
11
|
+
buildUploadedFileData(files: UploadResult[]): unknown[] | null;
|
|
12
|
+
buildImageGeneratePayload(prompt: string, options: Omit<GenerateOptions, "prompt">, tokens: GeminiTokens, accountIndex?: number): RequestPayload;
|
|
13
|
+
buildGemListPayload(type: "system" | "custom"): RPCPayload;
|
|
14
|
+
buildGemCreatePayload(options: GemCreateOptions): RPCPayload;
|
|
15
|
+
buildGemUpdatePayload(options: GemUpdateOptions): RPCPayload;
|
|
16
|
+
buildGemDeletePayload(gemId: string): RPCPayload;
|
|
17
|
+
resolveModel(nameOrAlias?: string): string;
|
|
18
|
+
private buildFileData;
|
|
19
|
+
}
|
|
20
|
+
//# sourceMappingURL=request-builder.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"request-builder.d.ts","sourceRoot":"","sources":["../../src/core/request-builder.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EACV,YAAY,EACZ,eAAe,EACf,eAAe,EACf,gBAAgB,EAChB,gBAAgB,EAChB,YAAY,EAEZ,UAAU,EACV,cAAc,EACd,YAAY,EACb,MAAM,aAAa,CAAC;AAErB,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAS;;IAMvB,UAAU,IAAI,IAAI;IAIlB,IAAI,KAAK,IAAI,MAAM,CAElB;IAED,MAAM,CAAC,WAAW,IAAI,MAAM;IAI5B,oBAAoB,CAClB,OAAO,EAAE,eAAe,EACxB,MAAM,EAAE,YAAY,EACpB,YAAY,SAAI,EAChB,YAAY,CAAC,EAAE,YAAY,GAC1B,cAAc;IAwCjB,iBAAiB,CACf,QAAQ,EAAE,UAAU,EAAE,EACtB,MAAM,EAAE,YAAY,EACpB,YAAY,SAAI,GACf,cAAc;IA2BjB,wBAAwB,CAAC,MAAM,EAAE,YAAY,EAAE,YAAY,SAAI,GAAG,cAAc;IAchF,qBAAqB,CAAC,KAAK,EAAE,YAAY,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI;IAK9D,yBAAyB,CACvB,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,IAAI,CAAC,eAAe,EAAE,QAAQ,CAAC,EACxC,MAAM,EAAE,YAAY,EACpB,YAAY,SAAI,GACf,cAAc;IAgDjB,mBAAmB,CAAC,IAAI,EAAE,QAAQ,GAAG,QAAQ,GAAG,UAAU;IAQ1D,qBAAqB,CAAC,OAAO,EAAE,gBAAgB,GAAG,UAAU;IA4B5D,qBAAqB,CAAC,OAAO,EAAE,gBAAgB,GAAG,UAAU;IA8B5D,qBAAqB,CAAC,KAAK,EAAE,MAAM,GAAG,UAAU;IAQhD,YAAY,CAAC,WAAW,CAAC,EAAE,MAAM,GAAG,MAAM;IAM1C,OAAO,CAAC,aAAa;CAItB"}
|
|
@@ -0,0 +1,210 @@
|
|
|
1
|
+
import { Endpoint, GEMINI_HEADERS, GrpcId, INNER_REQ_LIST_SIZE, MODEL_ALIASES, MODELS, NEW_CHAT_METADATA } from "./constants.js";
|
|
2
|
+
export class RequestBuilder {
|
|
3
|
+
_reqid;
|
|
4
|
+
constructor() {
|
|
5
|
+
this._reqid = RequestBuilder.randomReqId();
|
|
6
|
+
}
|
|
7
|
+
resetReqId() {
|
|
8
|
+
this._reqid = RequestBuilder.randomReqId();
|
|
9
|
+
}
|
|
10
|
+
get reqId() {
|
|
11
|
+
return this._reqid;
|
|
12
|
+
}
|
|
13
|
+
static randomReqId() {
|
|
14
|
+
return Math.floor(Math.random() * 90000) + 10000;
|
|
15
|
+
}
|
|
16
|
+
buildGeneratePayload(request, tokens, accountIndex = 0, chatMetadata) {
|
|
17
|
+
const reqid = this._reqid;
|
|
18
|
+
this._reqid += 100000;
|
|
19
|
+
const modelName = this.resolveModel(request.model);
|
|
20
|
+
const modelDef = MODELS[modelName] ?? MODELS.unspecified;
|
|
21
|
+
const fileData = request.images?.length ? this.buildFileData(request.images) : null;
|
|
22
|
+
const messageContent = [request.prompt, 0, null, fileData, null, null, 0];
|
|
23
|
+
const innerReqList = new Array(INNER_REQ_LIST_SIZE).fill(null);
|
|
24
|
+
innerReqList[0] = messageContent;
|
|
25
|
+
innerReqList[2] = chatMetadata ? [...chatMetadata] : [...NEW_CHAT_METADATA];
|
|
26
|
+
innerReqList[7] = 1;
|
|
27
|
+
if (request.gemId) {
|
|
28
|
+
innerReqList[19] = request.gemId;
|
|
29
|
+
}
|
|
30
|
+
const fReq = JSON.stringify([null, JSON.stringify(innerReqList)]);
|
|
31
|
+
const params = new URLSearchParams();
|
|
32
|
+
params.set("_reqid", String(reqid));
|
|
33
|
+
params.set("rt", "c");
|
|
34
|
+
if (tokens.cfb2h)
|
|
35
|
+
params.set("bl", tokens.cfb2h);
|
|
36
|
+
if (tokens.fdrfje)
|
|
37
|
+
params.set("f.sid", tokens.fdrfje);
|
|
38
|
+
const url = `${Endpoint.GENERATE(accountIndex)}?${params.toString()}`;
|
|
39
|
+
const headers = {
|
|
40
|
+
"Content-Type": GEMINI_HEADERS["Content-Type"],
|
|
41
|
+
"X-Same-Domain": GEMINI_HEADERS["X-Same-Domain"],
|
|
42
|
+
...modelDef.header
|
|
43
|
+
};
|
|
44
|
+
const body = new URLSearchParams({
|
|
45
|
+
at: tokens.snlm0e,
|
|
46
|
+
"f.req": fReq
|
|
47
|
+
}).toString();
|
|
48
|
+
return { url, headers, body };
|
|
49
|
+
}
|
|
50
|
+
buildBatchPayload(payloads, tokens, accountIndex = 0) {
|
|
51
|
+
const reqid = this._reqid;
|
|
52
|
+
this._reqid += 100000;
|
|
53
|
+
const serialized = payloads.map((payload) => [payload.rpcId, payload.payload, null, payload.identifier ?? "generic"]);
|
|
54
|
+
const params = new URLSearchParams();
|
|
55
|
+
params.set("rpcids", payloads.map((payload) => payload.rpcId).join(","));
|
|
56
|
+
params.set("_reqid", String(reqid));
|
|
57
|
+
params.set("rt", "c");
|
|
58
|
+
params.set("source-path", Endpoint.SOURCE_PATH(accountIndex));
|
|
59
|
+
if (tokens.cfb2h)
|
|
60
|
+
params.set("bl", tokens.cfb2h);
|
|
61
|
+
if (tokens.fdrfje)
|
|
62
|
+
params.set("f.sid", tokens.fdrfje);
|
|
63
|
+
const url = `${Endpoint.BATCH_EXEC(accountIndex)}?${params.toString()}`;
|
|
64
|
+
const headers = {
|
|
65
|
+
"Content-Type": GEMINI_HEADERS["Content-Type"],
|
|
66
|
+
"X-Same-Domain": GEMINI_HEADERS["X-Same-Domain"]
|
|
67
|
+
};
|
|
68
|
+
const body = new URLSearchParams({
|
|
69
|
+
at: tokens.snlm0e,
|
|
70
|
+
"f.req": JSON.stringify([serialized])
|
|
71
|
+
}).toString();
|
|
72
|
+
return { url, headers, body };
|
|
73
|
+
}
|
|
74
|
+
buildBardActivityPayload(tokens, accountIndex = 0) {
|
|
75
|
+
return this.buildBatchPayload([
|
|
76
|
+
{
|
|
77
|
+
rpcId: GrpcId.BARD_ACTIVITY,
|
|
78
|
+
payload: JSON.stringify([["bard_activity_enabled"]]),
|
|
79
|
+
identifier: "generic"
|
|
80
|
+
}
|
|
81
|
+
], tokens, accountIndex);
|
|
82
|
+
}
|
|
83
|
+
buildUploadedFileData(files) {
|
|
84
|
+
if (!files.length)
|
|
85
|
+
return null;
|
|
86
|
+
return files.map((file) => [[[file.fileUri], file.filename]]);
|
|
87
|
+
}
|
|
88
|
+
buildImageGeneratePayload(prompt, options, tokens, accountIndex = 0) {
|
|
89
|
+
const reqid = this._reqid;
|
|
90
|
+
this._reqid += 100000;
|
|
91
|
+
const modelName = this.resolveModel(options.model);
|
|
92
|
+
const modelDef = MODELS[modelName] ?? MODELS.unspecified;
|
|
93
|
+
const uploadedFileData = options.files?.length ? this.buildUploadedFileData(options.files) : null;
|
|
94
|
+
const imageFileData = options.images?.length ? this.buildFileData(options.images) : null;
|
|
95
|
+
const fileData = uploadedFileData ?? imageFileData;
|
|
96
|
+
const messageContent = [prompt, 0, null, fileData, null, null, 0];
|
|
97
|
+
if (options.usePro) {
|
|
98
|
+
messageContent.push(null, null, [null, null, null, null, null, null, [null, [1]]]);
|
|
99
|
+
}
|
|
100
|
+
const innerReqList = new Array(INNER_REQ_LIST_SIZE).fill(null);
|
|
101
|
+
innerReqList[0] = messageContent;
|
|
102
|
+
innerReqList[2] = options.chatMetadata ? [...options.chatMetadata] : [...NEW_CHAT_METADATA];
|
|
103
|
+
innerReqList[7] = 1;
|
|
104
|
+
if (options.gemId) {
|
|
105
|
+
innerReqList[19] = options.gemId;
|
|
106
|
+
}
|
|
107
|
+
if (options.usePro) {
|
|
108
|
+
innerReqList[32] = 1;
|
|
109
|
+
}
|
|
110
|
+
const fReq = JSON.stringify([null, JSON.stringify(innerReqList)]);
|
|
111
|
+
const params = new URLSearchParams();
|
|
112
|
+
params.set("_reqid", String(reqid));
|
|
113
|
+
params.set("rt", "c");
|
|
114
|
+
if (tokens.cfb2h)
|
|
115
|
+
params.set("bl", tokens.cfb2h);
|
|
116
|
+
if (tokens.fdrfje)
|
|
117
|
+
params.set("f.sid", tokens.fdrfje);
|
|
118
|
+
const url = `${Endpoint.GENERATE(accountIndex)}?${params.toString()}`;
|
|
119
|
+
const headers = {
|
|
120
|
+
"Content-Type": GEMINI_HEADERS["Content-Type"],
|
|
121
|
+
"X-Same-Domain": GEMINI_HEADERS["X-Same-Domain"],
|
|
122
|
+
...modelDef.header
|
|
123
|
+
};
|
|
124
|
+
const body = new URLSearchParams({
|
|
125
|
+
at: tokens.snlm0e,
|
|
126
|
+
"f.req": fReq
|
|
127
|
+
}).toString();
|
|
128
|
+
return { url, headers, body };
|
|
129
|
+
}
|
|
130
|
+
buildGemListPayload(type) {
|
|
131
|
+
return {
|
|
132
|
+
rpcId: GrpcId.LIST_GEMS,
|
|
133
|
+
payload: JSON.stringify([type === "system" ? 4 : 2, ["en"], 0]),
|
|
134
|
+
identifier: "generic"
|
|
135
|
+
};
|
|
136
|
+
}
|
|
137
|
+
buildGemCreatePayload(options) {
|
|
138
|
+
const payload = [
|
|
139
|
+
[
|
|
140
|
+
options.name,
|
|
141
|
+
options.description ?? "",
|
|
142
|
+
options.instructions,
|
|
143
|
+
null,
|
|
144
|
+
null,
|
|
145
|
+
null,
|
|
146
|
+
null,
|
|
147
|
+
null,
|
|
148
|
+
0,
|
|
149
|
+
null,
|
|
150
|
+
1,
|
|
151
|
+
null,
|
|
152
|
+
null,
|
|
153
|
+
null,
|
|
154
|
+
[]
|
|
155
|
+
]
|
|
156
|
+
];
|
|
157
|
+
return {
|
|
158
|
+
rpcId: GrpcId.CREATE_GEM,
|
|
159
|
+
payload: JSON.stringify(payload),
|
|
160
|
+
identifier: "generic"
|
|
161
|
+
};
|
|
162
|
+
}
|
|
163
|
+
buildGemUpdatePayload(options) {
|
|
164
|
+
const payload = [
|
|
165
|
+
options.gemId,
|
|
166
|
+
[
|
|
167
|
+
options.name ?? null,
|
|
168
|
+
options.description ?? "",
|
|
169
|
+
options.instructions ?? null,
|
|
170
|
+
null,
|
|
171
|
+
null,
|
|
172
|
+
null,
|
|
173
|
+
null,
|
|
174
|
+
null,
|
|
175
|
+
0,
|
|
176
|
+
null,
|
|
177
|
+
1,
|
|
178
|
+
null,
|
|
179
|
+
null,
|
|
180
|
+
null,
|
|
181
|
+
[],
|
|
182
|
+
0
|
|
183
|
+
]
|
|
184
|
+
];
|
|
185
|
+
return {
|
|
186
|
+
rpcId: GrpcId.UPDATE_GEM,
|
|
187
|
+
payload: JSON.stringify(payload),
|
|
188
|
+
identifier: "generic"
|
|
189
|
+
};
|
|
190
|
+
}
|
|
191
|
+
buildGemDeletePayload(gemId) {
|
|
192
|
+
return {
|
|
193
|
+
rpcId: GrpcId.DELETE_GEM,
|
|
194
|
+
payload: JSON.stringify([gemId]),
|
|
195
|
+
identifier: "generic"
|
|
196
|
+
};
|
|
197
|
+
}
|
|
198
|
+
resolveModel(nameOrAlias) {
|
|
199
|
+
if (!nameOrAlias)
|
|
200
|
+
return "unspecified";
|
|
201
|
+
const alias = MODEL_ALIASES[nameOrAlias];
|
|
202
|
+
return alias ?? nameOrAlias;
|
|
203
|
+
}
|
|
204
|
+
buildFileData(images) {
|
|
205
|
+
if (!images.length)
|
|
206
|
+
return null;
|
|
207
|
+
return images.map((image) => [[image.url ?? image.data, image.mimeType ?? "image/png"]]);
|
|
208
|
+
}
|
|
209
|
+
}
|
|
210
|
+
//# sourceMappingURL=request-builder.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"request-builder.js","sourceRoot":"","sources":["../../src/core/request-builder.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,cAAc,EACd,MAAM,EACN,mBAAmB,EACnB,aAAa,EACb,MAAM,EACN,iBAAiB,EAClB,MAAM,gBAAgB,CAAC;AAcxB,MAAM,OAAO,cAAc;IACjB,MAAM,CAAS;IAEvB;QACE,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,WAAW,EAAE,CAAC;IAC7C,CAAC;IAED,UAAU;QACR,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,WAAW,EAAE,CAAC;IAC7C,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,MAAM,CAAC,WAAW;QAChB,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,KAAK,CAAC,GAAG,KAAK,CAAC;IACnD,CAAC;IAED,oBAAoB,CAClB,OAAwB,EACxB,MAAoB,EACpB,YAAY,GAAG,CAAC,EAChB,YAA2B;QAE3B,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC;QAEtB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,QAAQ,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,MAAM,CAAC,WAAW,CAAC;QAEzD,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACpF,MAAM,cAAc,GAAc,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAErF,MAAM,YAAY,GAAc,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1E,YAAY,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC;QACjC,YAAY,CAAC,CAAC,CAAC,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC,CAAC;QAC5E,YAAY,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACpB,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,YAAY,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC;QACnC,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAElE,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;QACrC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QACtB,IAAI,MAAM,CAAC,KAAK;YAAE,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QACjD,IAAI,MAAM,CAAC,MAAM;YAAE,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAEtD,MAAM,GAAG,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC;QACtE,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,cAAc,CAAC,cAAc,CAAC;YAC9C,eAAe,EAAE,cAAc,CAAC,eAAe,CAAC;YAChD,GAAG,QAAQ,CAAC,MAAM;SACnB,CAAC;QACF,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC;YAC/B,EAAE,EAAE,MAAM,CAAC,MAAM;YACjB,OAAO,EAAE,IAAI;SACd,CAAC,CAAC,QAAQ,EAAE,CAAC;QAEd,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAChC,CAAC;IAED,iBAAiB,CACf,QAAsB,EACtB,MAAoB,EACpB,YAAY,GAAG,CAAC;QAEhB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC;QAEtB,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,UAAU,IAAI,SAAS,CAAC,CAAC,CAAC;QAEtH,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;QACrC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QACzE,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QACtB,MAAM,CAAC,GAAG,CAAC,aAAa,EAAE,QAAQ,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,CAAC;QAC9D,IAAI,MAAM,CAAC,KAAK;YAAE,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QACjD,IAAI,MAAM,CAAC,MAAM;YAAE,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAEtD,MAAM,GAAG,GAAG,GAAG,QAAQ,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC;QACxE,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,cAAc,CAAC,cAAc,CAAC;YAC9C,eAAe,EAAE,cAAc,CAAC,eAAe,CAAC;SACjD,CAAC;QACF,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC;YAC/B,EAAE,EAAE,MAAM,CAAC,MAAM;YACjB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,UAAU,CAAC,CAAC;SACtC,CAAC,CAAC,QAAQ,EAAE,CAAC;QAEd,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAChC,CAAC;IAED,wBAAwB,CAAC,MAAoB,EAAE,YAAY,GAAG,CAAC;QAC7D,OAAO,IAAI,CAAC,iBAAiB,CAC3B;YACE;gBACE,KAAK,EAAE,MAAM,CAAC,aAAa;gBAC3B,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC;gBACpD,UAAU,EAAE,SAAS;aACtB;SACF,EACD,MAAM,EACN,YAAY,CACb,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,KAAqB;QACzC,IAAI,CAAC,KAAK,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QAC/B,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,yBAAyB,CACvB,MAAc,EACd,OAAwC,EACxC,MAAoB,EACpB,YAAY,GAAG,CAAC;QAEhB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC;QAEtB,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,QAAQ,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,MAAM,CAAC,WAAW,CAAC;QACzD,MAAM,gBAAgB,GAAG,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAClG,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACzF,MAAM,QAAQ,GAAG,gBAAgB,IAAI,aAAa,CAAC;QAEnD,MAAM,cAAc,GAAc,CAAC,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC7E,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACrF,CAAC;QAED,MAAM,YAAY,GAAc,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1E,YAAY,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC;QACjC,YAAY,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC,CAAC;QAC5F,YAAY,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACpB,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,YAAY,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC;QACnC,CAAC;QACD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,YAAY,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;QACvB,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAElE,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;QACrC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QACtB,IAAI,MAAM,CAAC,KAAK;YAAE,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QACjD,IAAI,MAAM,CAAC,MAAM;YAAE,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAEtD,MAAM,GAAG,GAAG,GAAG,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC;QACtE,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,cAAc,CAAC,cAAc,CAAC;YAC9C,eAAe,EAAE,cAAc,CAAC,eAAe,CAAC;YAChD,GAAG,QAAQ,CAAC,MAAM;SACnB,CAAC;QACF,MAAM,IAAI,GAAG,IAAI,eAAe,CAAC;YAC/B,EAAE,EAAE,MAAM,CAAC,MAAM;YACjB,OAAO,EAAE,IAAI;SACd,CAAC,CAAC,QAAQ,EAAE,CAAC;QAEd,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAChC,CAAC;IAED,mBAAmB,CAAC,IAAyB;QAC3C,OAAO;YACL,KAAK,EAAE,MAAM,CAAC,SAAS;YACvB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/D,UAAU,EAAE,SAAS;SACtB,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,OAAyB;QAC7C,MAAM,OAAO,GAAG;YACd;gBACE,OAAO,CAAC,IAAI;gBACZ,OAAO,CAAC,WAAW,IAAI,EAAE;gBACzB,OAAO,CAAC,YAAY;gBACpB,IAAI;gBACJ,IAAI;gBACJ,IAAI;gBACJ,IAAI;gBACJ,IAAI;gBACJ,CAAC;gBACD,IAAI;gBACJ,CAAC;gBACD,IAAI;gBACJ,IAAI;gBACJ,IAAI;gBACJ,EAAE;aACH;SACF,CAAC;QAEF,OAAO;YACL,KAAK,EAAE,MAAM,CAAC,UAAU;YACxB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;YAChC,UAAU,EAAE,SAAS;SACtB,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,OAAyB;QAC7C,MAAM,OAAO,GAAG;YACd,OAAO,CAAC,KAAK;YACb;gBACE,OAAO,CAAC,IAAI,IAAI,IAAI;gBACpB,OAAO,CAAC,WAAW,IAAI,EAAE;gBACzB,OAAO,CAAC,YAAY,IAAI,IAAI;gBAC5B,IAAI;gBACJ,IAAI;gBACJ,IAAI;gBACJ,IAAI;gBACJ,IAAI;gBACJ,CAAC;gBACD,IAAI;gBACJ,CAAC;gBACD,IAAI;gBACJ,IAAI;gBACJ,IAAI;gBACJ,EAAE;gBACF,CAAC;aACF;SACF,CAAC;QAEF,OAAO;YACL,KAAK,EAAE,MAAM,CAAC,UAAU;YACxB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;YAChC,UAAU,EAAE,SAAS;SACtB,CAAC;IACJ,CAAC;IAED,qBAAqB,CAAC,KAAa;QACjC,OAAO;YACL,KAAK,EAAE,MAAM,CAAC,UAAU;YACxB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC;YAChC,UAAU,EAAE,SAAS;SACtB,CAAC;IACJ,CAAC;IAED,YAAY,CAAC,WAAoB;QAC/B,IAAI,CAAC,WAAW;YAAE,OAAO,aAAa,CAAC;QACvC,MAAM,KAAK,GAAG,aAAa,CAAC,WAAW,CAAC,CAAC;QACzC,OAAO,KAAK,IAAI,WAAW,CAAC;IAC9B,CAAC;IAEO,aAAa,CAAC,MAAyB;QAC7C,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QAChC,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,QAAQ,IAAI,WAAW,CAAC,CAAC,CAAC,CAAC;IAC3F,CAAC;CACF"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { ModelOutput, ParseResult } from "../types.js";
|
|
2
|
+
export declare class ResponseParser {
|
|
3
|
+
parseGenerateResponse(frames: unknown[]): ParseResult<ModelOutput>;
|
|
4
|
+
parseBatchResponse(frames: unknown[], rpcId: string): ParseResult<unknown>;
|
|
5
|
+
private mapErrorCode;
|
|
6
|
+
private extractCandidateText;
|
|
7
|
+
private parseWebImages;
|
|
8
|
+
private parseGeneratedImages;
|
|
9
|
+
private checkRateLimit;
|
|
10
|
+
private unescapeHtml;
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=response-parser.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"response-parser.d.ts","sourceRoot":"","sources":["../../src/core/response-parser.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAEV,WAAW,EAGX,WAAW,EACZ,MAAM,aAAa,CAAC;AAErB,qBAAa,cAAc;IACzB,qBAAqB,CAAC,MAAM,EAAE,OAAO,EAAE,GAAG,WAAW,CAAC,WAAW,CAAC;IAmGlE,kBAAkB,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,MAAM,GAAG,WAAW,CAAC,OAAO,CAAC;IAmC1E,OAAO,CAAC,YAAY;IAqBpB,OAAO,CAAC,oBAAoB;IAQ5B,OAAO,CAAC,cAAc;IAyBtB,OAAO,CAAC,oBAAoB;IAwB5B,OAAO,CAAC,cAAc;IAoBtB,OAAO,CAAC,YAAY;CAgBrB"}
|
|
@@ -0,0 +1,221 @@
|
|
|
1
|
+
import { API_ERROR_CODES, CARD_CONTENT_PATTERN, GOOGLE_USER_CONTENT_PATTERN, IMAGE_GEN_BLOCKED_PATTERNS, RATE_LIMIT_PATTERNS } from "./constants.js";
|
|
2
|
+
import { getNestedValue } from "./utils.js";
|
|
3
|
+
export class ResponseParser {
|
|
4
|
+
parseGenerateResponse(frames) {
|
|
5
|
+
let result = null;
|
|
6
|
+
let metadata = [];
|
|
7
|
+
let isCompleted = false;
|
|
8
|
+
for (const part of frames) {
|
|
9
|
+
const errorCode = getNestedValue(part, [5, 2, 0, 1, 0]);
|
|
10
|
+
if (typeof errorCode === "number") {
|
|
11
|
+
return { ok: false, error: this.mapErrorCode(errorCode) };
|
|
12
|
+
}
|
|
13
|
+
const innerStr = getNestedValue(part, [2]);
|
|
14
|
+
if (typeof innerStr !== "string" || !innerStr) {
|
|
15
|
+
continue;
|
|
16
|
+
}
|
|
17
|
+
let inner;
|
|
18
|
+
try {
|
|
19
|
+
inner = JSON.parse(innerStr);
|
|
20
|
+
}
|
|
21
|
+
catch {
|
|
22
|
+
continue;
|
|
23
|
+
}
|
|
24
|
+
const maybeMetadata = getNestedValue(inner, [1]);
|
|
25
|
+
if (Array.isArray(maybeMetadata)) {
|
|
26
|
+
metadata = maybeMetadata.map((value) => {
|
|
27
|
+
if (typeof value === "string" || value === null) {
|
|
28
|
+
return value;
|
|
29
|
+
}
|
|
30
|
+
return null;
|
|
31
|
+
});
|
|
32
|
+
}
|
|
33
|
+
const contextStr = getNestedValue(inner, [25]);
|
|
34
|
+
if (typeof contextStr === "string") {
|
|
35
|
+
isCompleted = true;
|
|
36
|
+
}
|
|
37
|
+
const candidatesList = getNestedValue(inner, [4], []);
|
|
38
|
+
if (!Array.isArray(candidatesList) || candidatesList.length === 0) {
|
|
39
|
+
continue;
|
|
40
|
+
}
|
|
41
|
+
const parsedCandidates = [];
|
|
42
|
+
for (const candidate of candidatesList) {
|
|
43
|
+
if (!Array.isArray(candidate)) {
|
|
44
|
+
continue;
|
|
45
|
+
}
|
|
46
|
+
const rcid = getNestedValue(candidate, [0]);
|
|
47
|
+
if (typeof rcid !== "string" || !rcid) {
|
|
48
|
+
continue;
|
|
49
|
+
}
|
|
50
|
+
const text = this.extractCandidateText(candidate);
|
|
51
|
+
const textError = this.checkRateLimit(text);
|
|
52
|
+
if (textError) {
|
|
53
|
+
return { ok: false, error: textError };
|
|
54
|
+
}
|
|
55
|
+
const thoughts = getNestedValue(candidate, [37, 0, 0]);
|
|
56
|
+
const completionObj = getNestedValue(candidate, [2]);
|
|
57
|
+
const status = getNestedValue(candidate, [8, 0], 1);
|
|
58
|
+
parsedCandidates.push({
|
|
59
|
+
rcid,
|
|
60
|
+
text,
|
|
61
|
+
thoughts: typeof thoughts === "string" ? thoughts : null,
|
|
62
|
+
webImages: this.parseWebImages(candidate),
|
|
63
|
+
generatedImages: this.parseGeneratedImages(candidate),
|
|
64
|
+
isFinal: (typeof completionObj === "object" && completionObj !== null) || status === 2
|
|
65
|
+
});
|
|
66
|
+
}
|
|
67
|
+
if (parsedCandidates.length === 0) {
|
|
68
|
+
continue;
|
|
69
|
+
}
|
|
70
|
+
result = {
|
|
71
|
+
metadata,
|
|
72
|
+
candidates: parsedCandidates,
|
|
73
|
+
chosenIndex: 0,
|
|
74
|
+
isCompleted
|
|
75
|
+
};
|
|
76
|
+
}
|
|
77
|
+
if (!result) {
|
|
78
|
+
return {
|
|
79
|
+
ok: false,
|
|
80
|
+
error: {
|
|
81
|
+
code: "PARSE_ERROR",
|
|
82
|
+
message: "No candidates found in response"
|
|
83
|
+
}
|
|
84
|
+
};
|
|
85
|
+
}
|
|
86
|
+
return { ok: true, data: result };
|
|
87
|
+
}
|
|
88
|
+
parseBatchResponse(frames, rpcId) {
|
|
89
|
+
for (const part of frames) {
|
|
90
|
+
const frameRpcId = getNestedValue(part, [1]);
|
|
91
|
+
if (frameRpcId !== rpcId) {
|
|
92
|
+
continue;
|
|
93
|
+
}
|
|
94
|
+
const innerStr = getNestedValue(part, [2]);
|
|
95
|
+
if (typeof innerStr !== "string") {
|
|
96
|
+
return { ok: true, data: null };
|
|
97
|
+
}
|
|
98
|
+
try {
|
|
99
|
+
return { ok: true, data: JSON.parse(innerStr) };
|
|
100
|
+
}
|
|
101
|
+
catch {
|
|
102
|
+
return {
|
|
103
|
+
ok: false,
|
|
104
|
+
error: {
|
|
105
|
+
code: "PARSE_ERROR",
|
|
106
|
+
message: `Failed to parse RPC ${rpcId} payload`,
|
|
107
|
+
raw: innerStr
|
|
108
|
+
}
|
|
109
|
+
};
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
return {
|
|
113
|
+
ok: false,
|
|
114
|
+
error: {
|
|
115
|
+
code: "PARSE_ERROR",
|
|
116
|
+
message: `RPC ${rpcId} not found`
|
|
117
|
+
}
|
|
118
|
+
};
|
|
119
|
+
}
|
|
120
|
+
mapErrorCode(code) {
|
|
121
|
+
switch (code) {
|
|
122
|
+
case API_ERROR_CODES.TEMPORARY_ERROR_1013:
|
|
123
|
+
return { code: "TEMPORARY_ERROR", message: "Temporary Gemini error", raw: code };
|
|
124
|
+
case API_ERROR_CODES.USAGE_LIMIT_EXCEEDED:
|
|
125
|
+
return { code: "USAGE_LIMIT_EXCEEDED", message: "Usage limit exceeded", raw: code };
|
|
126
|
+
case API_ERROR_CODES.MODEL_INCONSISTENT:
|
|
127
|
+
return { code: "MODEL_INCONSISTENT", message: "Model is inconsistent", raw: code };
|
|
128
|
+
case API_ERROR_CODES.MODEL_HEADER_INVALID:
|
|
129
|
+
return { code: "MODEL_HEADER_INVALID", message: "Model header is invalid", raw: code };
|
|
130
|
+
case API_ERROR_CODES.IP_TEMPORARILY_BLOCKED:
|
|
131
|
+
return { code: "IP_BLOCKED", message: "IP is temporarily blocked", raw: code };
|
|
132
|
+
default:
|
|
133
|
+
return {
|
|
134
|
+
code: "UNKNOWN_API_ERROR",
|
|
135
|
+
message: `Unknown Gemini API error code: ${code}`,
|
|
136
|
+
raw: code
|
|
137
|
+
};
|
|
138
|
+
}
|
|
139
|
+
}
|
|
140
|
+
extractCandidateText(candidateData) {
|
|
141
|
+
const primary = getNestedValue(candidateData, [1, 0], "") ?? "";
|
|
142
|
+
const fallback = getNestedValue(candidateData, [22, 0], "") ?? "";
|
|
143
|
+
const sourceText = CARD_CONTENT_PATTERN.test(primary) ? (fallback || primary) : primary;
|
|
144
|
+
const cleaned = sourceText.replace(GOOGLE_USER_CONTENT_PATTERN, "").trim();
|
|
145
|
+
return this.unescapeHtml(cleaned);
|
|
146
|
+
}
|
|
147
|
+
parseWebImages(candidateData) {
|
|
148
|
+
const imageItems = getNestedValue(candidateData, [12, 1], []);
|
|
149
|
+
if (!Array.isArray(imageItems)) {
|
|
150
|
+
return [];
|
|
151
|
+
}
|
|
152
|
+
const images = [];
|
|
153
|
+
for (const item of imageItems) {
|
|
154
|
+
const url = getNestedValue(item, [0, 0, 0]);
|
|
155
|
+
if (typeof url !== "string" || !url) {
|
|
156
|
+
continue;
|
|
157
|
+
}
|
|
158
|
+
const title = getNestedValue(item, [7, 0]);
|
|
159
|
+
const alt = getNestedValue(item, [0, 4]);
|
|
160
|
+
images.push({
|
|
161
|
+
url,
|
|
162
|
+
...(typeof title === "string" && title ? { title } : {}),
|
|
163
|
+
...(typeof alt === "string" && alt ? { alt } : {})
|
|
164
|
+
});
|
|
165
|
+
}
|
|
166
|
+
return images;
|
|
167
|
+
}
|
|
168
|
+
parseGeneratedImages(candidateData) {
|
|
169
|
+
const imageItems = getNestedValue(candidateData, [12, 7, 0], []);
|
|
170
|
+
if (!Array.isArray(imageItems)) {
|
|
171
|
+
return [];
|
|
172
|
+
}
|
|
173
|
+
const images = [];
|
|
174
|
+
for (const item of imageItems) {
|
|
175
|
+
const url = getNestedValue(item, [0, 3, 3]);
|
|
176
|
+
if (typeof url !== "string" || !url) {
|
|
177
|
+
continue;
|
|
178
|
+
}
|
|
179
|
+
const alt = getNestedValue(item, [3, 5, 0]);
|
|
180
|
+
images.push({
|
|
181
|
+
url,
|
|
182
|
+
title: "generated",
|
|
183
|
+
...(typeof alt === "string" && alt ? { alt } : {})
|
|
184
|
+
});
|
|
185
|
+
}
|
|
186
|
+
return images;
|
|
187
|
+
}
|
|
188
|
+
checkRateLimit(text) {
|
|
189
|
+
if (RATE_LIMIT_PATTERNS.some((pattern) => pattern.test(text))) {
|
|
190
|
+
return {
|
|
191
|
+
code: "RATE_LIMITED",
|
|
192
|
+
message: "Rate limited by Gemini",
|
|
193
|
+
raw: text
|
|
194
|
+
};
|
|
195
|
+
}
|
|
196
|
+
if (IMAGE_GEN_BLOCKED_PATTERNS.some((pattern) => pattern.test(text))) {
|
|
197
|
+
return {
|
|
198
|
+
code: "IMAGE_GEN_BLOCKED",
|
|
199
|
+
message: "Image generation is currently blocked",
|
|
200
|
+
raw: text
|
|
201
|
+
};
|
|
202
|
+
}
|
|
203
|
+
return null;
|
|
204
|
+
}
|
|
205
|
+
unescapeHtml(text) {
|
|
206
|
+
const unescaped = text
|
|
207
|
+
.replace(/&/g, "&")
|
|
208
|
+
.replace(/</g, "<")
|
|
209
|
+
.replace(/>/g, ">")
|
|
210
|
+
.replace(/'/g, "'")
|
|
211
|
+
.replace(/"/g, '"');
|
|
212
|
+
return unescaped.replace(/&#(\d+);/g, (full, decimal) => {
|
|
213
|
+
const code = Number.parseInt(decimal, 10);
|
|
214
|
+
if (Number.isNaN(code)) {
|
|
215
|
+
return full;
|
|
216
|
+
}
|
|
217
|
+
return String.fromCharCode(code);
|
|
218
|
+
});
|
|
219
|
+
}
|
|
220
|
+
}
|
|
221
|
+
//# sourceMappingURL=response-parser.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"response-parser.js","sourceRoot":"","sources":["../../src/core/response-parser.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,eAAe,EACf,oBAAoB,EACpB,2BAA2B,EAC3B,0BAA0B,EAC1B,mBAAmB,EACpB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAS5C,MAAM,OAAO,cAAc;IACzB,qBAAqB,CAAC,MAAiB;QACrC,IAAI,MAAM,GAAuB,IAAI,CAAC;QACtC,IAAI,QAAQ,GAAsB,EAAE,CAAC;QACrC,IAAI,WAAW,GAAG,KAAK,CAAC;QAExB,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;YAC1B,MAAM,SAAS,GAAG,cAAc,CAAS,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAChE,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAClC,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,EAAE,CAAC;YAC5D,CAAC;YAED,MAAM,QAAQ,GAAG,cAAc,CAAS,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACnD,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC9C,SAAS;YACX,CAAC;YAED,IAAI,KAAc,CAAC;YACnB,IAAI,CAAC;gBACH,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAY,CAAC;YAC1C,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;YAED,MAAM,aAAa,GAAG,cAAc,CAAY,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5D,IAAI,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;gBACjC,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;oBACrC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;wBAChD,OAAO,KAAK,CAAC;oBACf,CAAC;oBACD,OAAO,IAAI,CAAC;gBACd,CAAC,CAAC,CAAC;YACL,CAAC;YAED,MAAM,UAAU,GAAG,cAAc,CAAS,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YACvD,IAAI,OAAO,UAAU,KAAK,QAAQ,EAAE,CAAC;gBACnC,WAAW,GAAG,IAAI,CAAC;YACrB,CAAC;YAED,MAAM,cAAc,GAAG,cAAc,CAAY,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACjE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAClE,SAAS;YACX,CAAC;YAED,MAAM,gBAAgB,GAAsB,EAAE,CAAC;YAC/C,KAAK,MAAM,SAAS,IAAI,cAAc,EAAE,CAAC;gBACvC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC9B,SAAS;gBACX,CAAC;gBAED,MAAM,IAAI,GAAG,cAAc,CAAS,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBACpD,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;oBACtC,SAAS;gBACX,CAAC;gBAED,MAAM,IAAI,GAAG,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;gBAClD,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;gBAC5C,IAAI,SAAS,EAAE,CAAC;oBACd,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;gBACzC,CAAC;gBAED,MAAM,QAAQ,GAAG,cAAc,CAAS,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC/D,MAAM,aAAa,GAAG,cAAc,CAAU,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9D,MAAM,MAAM,GAAG,cAAc,CAAS,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBAE5D,gBAAgB,CAAC,IAAI,CAAC;oBACpB,IAAI;oBACJ,IAAI;oBACJ,QAAQ,EAAE,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI;oBACxD,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC;oBACzC,eAAe,EAAE,IAAI,CAAC,oBAAoB,CAAC,SAAS,CAAC;oBACrD,OAAO,EAAE,CAAC,OAAO,aAAa,KAAK,QAAQ,IAAI,aAAa,KAAK,IAAI,CAAC,IAAI,MAAM,KAAK,CAAC;iBACvF,CAAC,CAAC;YACL,CAAC;YAED,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAClC,SAAS;YACX,CAAC;YAED,MAAM,GAAG;gBACP,QAAQ;gBACR,UAAU,EAAE,gBAAgB;gBAC5B,WAAW,EAAE,CAAC;gBACd,WAAW;aACZ,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;gBACL,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE;oBACL,IAAI,EAAE,aAAa;oBACnB,OAAO,EAAE,iCAAiC;iBAC3C;aACF,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IACpC,CAAC;IAED,kBAAkB,CAAC,MAAiB,EAAE,KAAa;QACjD,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;YAC1B,MAAM,UAAU,GAAG,cAAc,CAAS,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACrD,IAAI,UAAU,KAAK,KAAK,EAAE,CAAC;gBACzB,SAAS;YACX,CAAC;YAED,MAAM,QAAQ,GAAG,cAAc,CAAS,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACnD,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACjC,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;YAClC,CAAC;YAED,IAAI,CAAC;gBACH,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAY,EAAE,CAAC;YAC7D,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO;oBACL,EAAE,EAAE,KAAK;oBACT,KAAK,EAAE;wBACL,IAAI,EAAE,aAAa;wBACnB,OAAO,EAAE,uBAAuB,KAAK,UAAU;wBAC/C,GAAG,EAAE,QAAQ;qBACd;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO;YACL,EAAE,EAAE,KAAK;YACT,KAAK,EAAE;gBACL,IAAI,EAAE,aAAa;gBACnB,OAAO,EAAE,OAAO,KAAK,YAAY;aAClC;SACF,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,IAAY;QAC/B,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,eAAe,CAAC,oBAAoB;gBACvC,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,wBAAwB,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;YACnF,KAAK,eAAe,CAAC,oBAAoB;gBACvC,OAAO,EAAE,IAAI,EAAE,sBAAsB,EAAE,OAAO,EAAE,sBAAsB,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;YACtF,KAAK,eAAe,CAAC,kBAAkB;gBACrC,OAAO,EAAE,IAAI,EAAE,oBAAoB,EAAE,OAAO,EAAE,uBAAuB,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;YACrF,KAAK,eAAe,CAAC,oBAAoB;gBACvC,OAAO,EAAE,IAAI,EAAE,sBAAsB,EAAE,OAAO,EAAE,yBAAyB,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;YACzF,KAAK,eAAe,CAAC,sBAAsB;gBACzC,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,2BAA2B,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;YACjF;gBACE,OAAO;oBACL,IAAI,EAAE,mBAAmB;oBACzB,OAAO,EAAE,kCAAkC,IAAI,EAAE;oBACjD,GAAG,EAAE,IAAI;iBACV,CAAC;QACN,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,aAAwB;QACnD,MAAM,OAAO,GAAG,cAAc,CAAS,aAAa,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;QACxE,MAAM,QAAQ,GAAG,cAAc,CAAS,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;QAC1E,MAAM,UAAU,GAAG,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QACxF,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,2BAA2B,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC3E,OAAO,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;IAEO,cAAc,CAAC,aAAwB;QAC7C,MAAM,UAAU,GAAG,cAAc,CAAY,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACzE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,MAAM,GAAgB,EAAE,CAAC;QAC/B,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,cAAc,CAAS,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACpD,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,CAAC,GAAG,EAAE,CAAC;gBACpC,SAAS;YACX,CAAC;YAED,MAAM,KAAK,GAAG,cAAc,CAAS,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,GAAG,GAAG,cAAc,CAAS,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACjD,MAAM,CAAC,IAAI,CAAC;gBACV,GAAG;gBACH,GAAG,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACxD,GAAG,CAAC,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACnD,CAAC,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,oBAAoB,CAAC,aAAwB;QACnD,MAAM,UAAU,GAAG,cAAc,CAAY,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5E,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,MAAM,GAAgB,EAAE,CAAC;QAC/B,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,cAAc,CAAS,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACpD,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,CAAC,GAAG,EAAE,CAAC;gBACpC,SAAS;YACX,CAAC;YAED,MAAM,GAAG,GAAG,cAAc,CAAS,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACpD,MAAM,CAAC,IAAI,CAAC;gBACV,GAAG;gBACH,KAAK,EAAE,WAAW;gBAClB,GAAG,CAAC,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACnD,CAAC,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,cAAc,CAAC,IAAY;QACjC,IAAI,mBAAmB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;YAC9D,OAAO;gBACL,IAAI,EAAE,cAAc;gBACpB,OAAO,EAAE,wBAAwB;gBACjC,GAAG,EAAE,IAAI;aACV,CAAC;QACJ,CAAC;QAED,IAAI,0BAA0B,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;YACrE,OAAO;gBACL,IAAI,EAAE,mBAAmB;gBACzB,OAAO,EAAE,uCAAuC;gBAChD,GAAG,EAAE,IAAI;aACV,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,YAAY,CAAC,IAAY;QAC/B,MAAM,SAAS,GAAG,IAAI;aACnB,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC;aACtB,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC;aACrB,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC;aACrB,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC;aACtB,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;QAE3B,OAAO,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE;YACtD,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAC1C,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvB,OAAO,IAAI,CAAC;YACd,CAAC;YACD,OAAO,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
export type ErrorClassification = "retryable" | "rate_limited" | "failover_only" | "fatal";
|
|
2
|
+
export interface RetryOptions {
|
|
3
|
+
maxRetries: number;
|
|
4
|
+
delayFactor: number;
|
|
5
|
+
onRetry?: (attempt: number, error: Error, delay: number) => void;
|
|
6
|
+
}
|
|
7
|
+
export declare function classifyError(error: unknown): ErrorClassification;
|
|
8
|
+
export declare function calculateDelay(attempt: number, maxRetries: number, delayFactor: number): number;
|
|
9
|
+
export declare function withRetry<T>(fn: () => Promise<T>, options?: Partial<RetryOptions>): Promise<T>;
|
|
10
|
+
//# sourceMappingURL=retry.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"retry.d.ts","sourceRoot":"","sources":["../../src/core/retry.ts"],"names":[],"mappings":"AAIA,MAAM,MAAM,mBAAmB,GAAG,WAAW,GAAG,cAAc,GAAG,eAAe,GAAG,OAAO,CAAC;AAE3F,MAAM,WAAW,YAAY;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CAClE;AAkFD,wBAAgB,aAAa,CAAC,KAAK,EAAE,OAAO,GAAG,mBAAmB,CA6CjE;AAED,wBAAgB,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,CAI/F;AAED,wBAAsB,SAAS,CAAC,CAAC,EAC/B,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EACpB,OAAO,GAAE,OAAO,CAAC,YAAY,CAAM,GAClC,OAAO,CAAC,CAAC,CAAC,CAkCZ"}
|