openclaw
Version:
Multi-channel AI gateway with extensible messaging integrations
129 lines (128 loc) • 5.59 kB
JavaScript
import { r as resolveGlobalMap } from "../../global-singleton-0m-JziJW.mjs";
import { o as getRuntimeConfigSnapshot } from "../../runtime-snapshot-DYptB0Cf.mjs";
import { t as jsonResult } from "../../tool-results-BCM3fdVS.mjs";
import { f as readStringArrayParam, m as readToolStringParam } from "../../common-Dyxcwsy8.mjs";
import { c as resolveTimeoutSeconds, i as readCache, l as writeCache, o as resolveCacheTtlMs } from "../../web-shared-D8fUK6Vo.mjs";
import { c as normalizeToIsoDate } from "../../web-search-provider-common-CA2avHBq.mjs";
import "../../runtime-config-snapshot-A-BTduWa.mjs";
import "../../global-singleton-yHEs_vfD.mjs";
import "../../provider-web-search-D50o08z_.mjs";
import { r as resolveXaiToolApiKeyWithAuth, t as isXaiToolEnabled } from "../../tool-auth-shared-DfHEiTF0.mjs";
import { t as resolveEffectiveXSearchConfig } from "../../x-search-config-wCiEf8cg.mjs";
import { n as buildMissingXSearchApiKeyPayload, r as createXSearchToolDefinition } from "../../x-search-tool-shared-nlZ40kdg.mjs";
import { a as resolveXaiXSearchInlineCitations, i as resolveXaiXSearchEndpoint, n as buildXaiXSearchPayload, o as resolveXaiXSearchMaxTurns, r as requestXaiXSearch, s as resolveXaiXSearchModel } from "../../x-search-shared-DlEwharA.mjs";
//#region extensions/xai/x-search.ts
var PluginToolInputError = class extends Error {
constructor(message) {
super(message);
this.name = "ToolInputError";
}
};
const X_SEARCH_CACHE = resolveGlobalMap(Symbol.for("openclaw.xai.x-search.cache"));
function normalizeOptionalIsoDate(value, label) {
if (!value) return;
const trimmed = value.trim();
if (!trimmed) return;
if (!/^\d{4}-\d{2}-\d{2}$/.test(trimmed)) throw new PluginToolInputError(`${label} must use YYYY-MM-DD`);
if (!normalizeToIsoDate(trimmed)) throw new PluginToolInputError(`${label} must be a valid calendar date`);
return trimmed;
}
function validateXSearchHandleFilters(params) {
if (params.allowedXHandles && params.excludedXHandles) throw new PluginToolInputError("allowed_x_handles and excluded_x_handles cannot be used together");
for (const [label, handles] of [["allowed_x_handles", params.allowedXHandles], ["excluded_x_handles", params.excludedXHandles]]) if (handles && handles.length > 20) throw new PluginToolInputError(`${label} cannot contain more than 20 handles`);
}
function buildXSearchCacheKey(params) {
return JSON.stringify([
"x_search",
params.model,
params.endpoint,
params.query,
params.inlineCitations,
params.maxTurns ?? null,
params.options.allowedXHandles ?? null,
params.options.excludedXHandles ?? null,
params.options.fromDate ?? null,
params.options.toDate ?? null,
params.options.enableImageUnderstanding ?? false,
params.options.enableVideoUnderstanding ?? false
]);
}
function createXSearchTool(options) {
const xSearchConfig = resolveEffectiveXSearchConfig(options?.config);
const runtimeConfig = options?.runtimeConfig ?? getRuntimeConfigSnapshot();
if (!isXaiToolEnabled({
enabled: typeof xSearchConfig?.enabled === "boolean" ? xSearchConfig.enabled : void 0,
runtimeConfig: runtimeConfig ?? void 0,
sourceConfig: options?.config,
auth: options?.auth
})) return null;
return createXSearchToolDefinition(async (_toolCallId, args, signal) => {
signal?.throwIfAborted();
const apiKey = await resolveXaiToolApiKeyWithAuth({
sourceConfig: options?.config,
runtimeConfig: runtimeConfig ?? void 0,
auth: options?.auth
});
if (!apiKey) return jsonResult(buildMissingXSearchApiKeyPayload());
const query = readToolStringParam(args, "query", { required: true });
const allowedXHandles = readStringArrayParam(args, "allowed_x_handles");
const excludedXHandles = readStringArrayParam(args, "excluded_x_handles");
validateXSearchHandleFilters({
allowedXHandles,
excludedXHandles
});
const fromDate = normalizeOptionalIsoDate(readToolStringParam(args, "from_date"), "from_date");
const toDate = normalizeOptionalIsoDate(readToolStringParam(args, "to_date"), "to_date");
if (fromDate && toDate && fromDate > toDate) throw new PluginToolInputError("from_date must be on or before to_date");
const xSearchOptions = {
query,
allowedXHandles,
excludedXHandles,
fromDate,
toDate,
enableImageUnderstanding: args.enable_image_understanding === true,
enableVideoUnderstanding: args.enable_video_understanding === true
};
const model = resolveXaiXSearchModel(xSearchConfig);
const endpoint = resolveXaiXSearchEndpoint(xSearchConfig);
const inlineCitations = resolveXaiXSearchInlineCitations(xSearchConfig);
const maxTurns = resolveXaiXSearchMaxTurns(xSearchConfig);
const cacheKey = buildXSearchCacheKey({
query,
model,
endpoint,
inlineCitations,
maxTurns,
options: xSearchOptions
});
const cacheTtlMs = resolveCacheTtlMs(xSearchConfig?.cacheTtlMinutes, 15);
const cached = readCache(X_SEARCH_CACHE, cacheKey, cacheTtlMs);
if (cached) return jsonResult({
...cached.value,
cached: true
});
const startedAt = Date.now();
const result = await requestXaiXSearch({
apiKey,
endpoint,
model,
timeoutSeconds: resolveTimeoutSeconds(xSearchConfig?.timeoutSeconds, 30),
inlineCitations,
maxTurns,
options: xSearchOptions,
...signal ? { signal } : {}
});
signal?.throwIfAborted();
const payload = buildXaiXSearchPayload({
query,
model,
tookMs: Date.now() - startedAt,
...result,
options: xSearchOptions
});
writeCache(X_SEARCH_CACHE, cacheKey, payload, cacheTtlMs);
return jsonResult(payload);
});
}
//#endregion
export { createXSearchTool };