File size: 3,257 Bytes
88c4c60 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 | import crypto from "crypto";
import { BaseExecutor } from "./base.js";
import { PROVIDERS } from "../config/providers.js";
/**
* IFlowExecutor - Executor for iFlow API with HMAC-SHA256 signature
*/
export class IFlowExecutor extends BaseExecutor {
constructor() {
super("iflow", PROVIDERS.iflow);
}
/**
* Generate UUID v4
* @returns {string} UUID v4 string
*/
generateUUID() {
return crypto.randomUUID();
}
/**
* Create iFlow signature using HMAC-SHA256
* @param {string} userAgent - User agent string
* @param {string} sessionID - Session ID
* @param {number} timestamp - Unix timestamp in milliseconds
* @param {string} apiKey - API key for signing
* @returns {string} Hex-encoded signature
*/
createIFlowSignature(userAgent, sessionID, timestamp, apiKey) {
if (!apiKey) return "";
const payload = `${userAgent}:${sessionID}:${timestamp}`;
const hmac = crypto.createHmac("sha256", apiKey);
hmac.update(payload);
return hmac.digest("hex");
}
/**
* Build headers with iFlow-specific signature
* @param {object} credentials - Provider credentials
* @param {boolean} stream - Whether streaming is enabled
* @returns {object} Headers object
*/
buildHeaders(credentials, stream = true) {
// Generate session ID and timestamp
const sessionID = `session-${this.generateUUID()}`;
const timestamp = Date.now();
// Get user agent from config
const userAgent = this.config.headers["User-Agent"] || "iFlow-Cli";
// Get API key (prefer apiKey, fallback to accessToken)
const apiKey = credentials.apiKey || credentials.accessToken || "";
// Create signature
const signature = this.createIFlowSignature(userAgent, sessionID, timestamp, apiKey);
// Build headers
const headers = {
"Content-Type": "application/json",
...this.config.headers,
"session-id": sessionID,
"x-iflow-timestamp": timestamp.toString(),
"x-iflow-signature": signature
};
// Add authorization
if (credentials.apiKey) {
headers["Authorization"] = `Bearer ${credentials.apiKey}`;
}
// Add streaming header
if (stream) {
headers["Accept"] = "text/event-stream";
}
return headers;
}
/**
* Build URL for iFlow API
* @param {string} model - Model name
* @param {boolean} stream - Whether streaming is enabled
* @param {number} urlIndex - URL index for fallback
* @param {object} credentials - Provider credentials
* @returns {string} API URL
*/
buildUrl(model, stream, urlIndex = 0, credentials = null) {
return this.config.baseUrl;
}
/**
* Transform request body - inject stream_options for usage data
* @param {string} model - Model name
* @param {object} body - Request body
* @param {boolean} stream - Whether streaming is enabled
* @param {object} credentials - Provider credentials
* @returns {object} Transformed body
*/
transformRequest(model, body, stream, credentials) {
// Inject stream_options for streaming requests to get usage data
if (stream && body.messages && !body.stream_options) {
body.stream_options = { include_usage: true };
}
return body;
}
}
export default IFlowExecutor;
|