@caoguo/maplibre-ai 0.0.5 → 0.0.6

package/dist/index.cjs

@@ -1478,6 +1478,168 @@ function createDeepSeekClient(config) {
   return new DeepSeekClient(config);
 }
 
+// src/llm/openaiCompatible.ts
+function normalizeError2(body, provider) {
+  const err = new Error(body || `${provider} API \u8BF7\u6C42\u5931\u8D25`);
+  err.name = `${provider}Error`;
+  return err;
+}
+var OpenAICompatibleClient = class {
+  apiKey;
+  baseUrl;
+  model;
+  temperature;
+  maxTokens;
+  timeoutMs;
+  retries;
+  fetchImpl;
+  authScheme;
+  extraHeaders;
+  providerLabel;
+  constructor(config) {
+    if (!config.apiKey) throw new Error("OpenAICompatibleClient: apiKey \u4E0D\u80FD\u4E3A\u7A7A");
+    if (!config.baseUrl) throw new Error("OpenAICompatibleClient: baseUrl \u4E0D\u80FD\u4E3A\u7A7A");
+    if (!config.model) throw new Error("OpenAICompatibleClient: model \u4E0D\u80FD\u4E3A\u7A7A");
+    this.apiKey = config.apiKey;
+    this.baseUrl = config.baseUrl.replace(/\/+$/, "");
+    this.model = config.model;
+    this.temperature = config.temperature ?? 0.3;
+    this.maxTokens = config.maxTokens ?? 2048;
+    this.timeoutMs = config.timeoutMs ?? 3e4;
+    this.retries = config.retries ?? 2;
+    this.fetchImpl = config.fetchImpl ?? (globalThis.fetch ?? fetch);
+    this.authScheme = config.authScheme ?? "Bearer";
+    this.extraHeaders = config.extraHeaders ?? {};
+    this.providerLabel = (() => {
+      try {
+        return new URL(this.baseUrl).hostname || "LLM";
+      } catch {
+        return "LLM";
+      }
+    })();
+  }
+  async chat(messages, opts = {}) {
+    return this.requestWithRetry(messages, opts);
+  }
+  async chatJson(messages) {
+    const result = await this.chat(messages, { json: true });
+    const raw = result.content.trim();
+    const cleaned = raw.replace(/^```(?:json)?\s*/i, "").replace(/\s*```$/, "");
+    let data;
+    try {
+      data = JSON.parse(cleaned);
+    } catch {
+      const match = cleaned.match(/\{[\s\S]*\}|\[[\s\S]*\]/);
+      if (match) {
+        data = JSON.parse(match[0]);
+      } else {
+        throw new Error(`${this.providerLabel} \u8FD4\u56DE\u4E86\u975E JSON \u5185\u5BB9: ${raw.slice(0, 200)}`);
+      }
+    }
+    return { data, raw, model: result.model };
+  }
+  async requestWithRetry(messages, opts) {
+    let lastError = null;
+    for (let attempt = 0; attempt <= this.retries; attempt++) {
+      try {
+        return await this.requestOnce(messages, opts);
+      } catch (e) {
+        lastError = e;
+        if (attempt < this.retries) {
+          await this.sleep(Math.min(1e3 * 2 ** attempt, 8e3));
+        }
+      }
+    }
+    throw lastError ?? new Error(`${this.providerLabel} \u8BF7\u6C42\u5931\u8D25`);
+  }
+  async requestOnce(messages, opts) {
+    const controller = new AbortController();
+    const timer = setTimeout(() => controller.abort(), this.timeoutMs);
+    const body = {
+      model: this.model,
+      messages,
+      temperature: this.temperature,
+      max_tokens: this.maxTokens,
+      stream: Boolean(opts.onChunk)
+    };
+    if (opts.json) {
+      body.response_format = { type: "json_object" };
+    }
+    const headers = {
+      "Content-Type": "application/json",
+      ...this.extraHeaders
+    };
+    if (this.authScheme === "Bearer") {
+      headers.Authorization = `Bearer ${this.apiKey}`;
+    } else {
+      headers["X-Api-Key"] = this.apiKey;
+    }
+    try {
+      const res = await this.fetchImpl(`${this.baseUrl}/chat/completions`, {
+        method: "POST",
+        headers,
+        body: JSON.stringify(body),
+        signal: controller.signal
+      });
+      if (!res.ok) {
+        const text = await res.text().catch(() => "");
+        throw normalizeError2(`${this.providerLabel} HTTP ${res.status}: ${text}`, this.providerLabel);
+      }
+      if (opts.onChunk && res.body) {
+        return this.parseStream(res.body, opts.onChunk);
+      }
+      const json = await res.json();
+      const content = json.choices?.[0]?.message?.content ?? "";
+      return {
+        content,
+        model: json.model ?? this.model,
+        usage: json.usage ? {
+          promptTokens: json.usage.prompt_tokens,
+          completionTokens: json.usage.completion_tokens,
+          totalTokens: json.usage.prompt_tokens + json.usage.completion_tokens
+        } : void 0
+      };
+    } finally {
+      clearTimeout(timer);
+    }
+  }
+  async parseStream(body, onChunk) {
+    const reader = body.getReader();
+    const decoder = new TextDecoder();
+    let content = "";
+    let buffer = "";
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) break;
+      buffer += decoder.decode(value, { stream: true });
+      const lines = buffer.split("\n");
+      buffer = lines.pop() ?? "";
+      for (const line of lines) {
+        const trimmed = line.trim();
+        if (!trimmed.startsWith("data:")) continue;
+        const data = trimmed.slice(5).trim();
+        if (data === "[DONE]") continue;
+        try {
+          const json = JSON.parse(data);
+          const delta = json.choices?.[0]?.delta?.content ?? "";
+          if (delta) {
+            content += delta;
+            onChunk(delta);
+          }
+        } catch {
+        }
+      }
+    }
+    return { content, model: this.model };
+  }
+  sleep(ms) {
+    return new Promise((r) => setTimeout(r, ms));
+  }
+};
+function createOpenAICompatibleClient(config) {
+  return new OpenAICompatibleClient(config);
+}
+
 exports.CARRIER_STYLES = CARRIER_STYLES;
 exports.CHINA_BOUNDS = CHINA_BOUNDS;
 exports.COLOR_NAMES = COLOR_NAMES;
@@ -1489,6 +1651,7 @@ exports.LOCAL_GEO_DB = LOCAL_GEO_DB;
 exports.LlmMapCopilot = LlmMapCopilot;
 exports.LlmNlpg = LlmNlpg;
 exports.MapCopilot = MapCopilot;
+exports.OpenAICompatibleClient = OpenAICompatibleClient;
 exports.PLACE_COORDINATES = PLACE_COORDINATES;
 exports.adjustBrightness = adjustBrightness;
 exports.analyzePerformance = analyzePerformance;
@@ -1496,6 +1659,7 @@ exports.analyzeTiles = analyzeTiles;
 exports.batchGeocode = batchGeocode;
 exports.classifyIntent = classifyIntent;
 exports.createDeepSeekClient = createDeepSeekClient;
+exports.createOpenAICompatibleClient = createOpenAICompatibleClient;
 exports.detectCRS = detectCRS;
 exports.detectField = detectField;
 exports.detectHeaders = detectHeaders;