poe-code 3.0.348 → 3.0.349

package/dist/index.js

@@ -79568,9 +79568,9 @@ function createPoeClient(options) {
       const data = await requestCompletion(httpClient, options.baseUrl, options.apiKey, request);
       return { content: extractTextContent(data) };
     },
-    async media(_type, request) {
+    async media(type, request) {
       const data = await requestCompletion(httpClient, options.baseUrl, options.apiKey, request);
-      return extractMediaFromCompletion(data);
+      return extractMediaFromCompletion(type, data);
     }
   };
 }
@@ -79634,7 +79634,7 @@ function extractTextContent(data) {
   if (!isRecord32(message2)) return void 0;
   return typeof message2.content === "string" ? message2.content : void 0;
 }
-function extractMediaFromCompletion(data) {
+function extractMediaFromCompletion(type, data) {
   const content = extractTextContent(data);
   if (!content) return {};
   try {
@@ -79657,20 +79657,58 @@ function extractMediaFromCompletion(data) {
   if (isValidUrl(content.trim())) {
     return { url: content.trim() };
   }
-  const markdownUrl = extractMarkdownUrl(content);
+  const markdownUrl = extractMarkdownMediaUrl(content, type);
   if (markdownUrl) {
     return { url: markdownUrl };
   }
   return { content };
 }
-function extractMarkdownUrl(content) {
-  const start = content.indexOf("](");
-  if (start === -1) return void 0;
-  const urlStart = start + 2;
-  const end = content.indexOf(")", urlStart);
-  if (end === -1) return void 0;
-  const url = content.slice(urlStart, end);
-  return isValidUrl(url) ? url : void 0;
+function extractMarkdownMediaUrl(content, type) {
+  let searchFrom = 0;
+  while (searchFrom < content.length) {
+    const openLabel = content.indexOf("[", searchFrom);
+    if (openLabel === -1) return void 0;
+    const closeLabel = content.indexOf("](", openLabel);
+    if (closeLabel === -1) return void 0;
+    const urlStart = closeLabel + 2;
+    const urlEnd = content.indexOf(")", urlStart);
+    if (urlEnd === -1) return void 0;
+    const label = content.slice(openLabel + 1, closeLabel);
+    const url = content.slice(urlStart, urlEnd);
+    const isImageMarkdown = openLabel > 0 && content[openLabel - 1] === "!";
+    if (isValidUrl(url) && isGeneratedMediaLabel(label, type, isImageMarkdown)) {
+      return url;
+    }
+    searchFrom = urlEnd + 1;
+  }
+  return void 0;
+}
+function isGeneratedMediaLabel(label, type, isImageMarkdown) {
+  if (type === "image" && isImageMarkdown) {
+    return true;
+  }
+  return labelTokens(label).includes(type);
+}
+function labelTokens(label) {
+  const tokens = [];
+  let current = "";
+  for (const char of label.toLowerCase()) {
+    const code = char.charCodeAt(0);
+    const isAsciiLetter7 = code >= 97 && code <= 122;
+    const isAsciiDigit2 = code >= 48 && code <= 57;
+    if (isAsciiLetter7 || isAsciiDigit2) {
+      current += char;
+      continue;
+    }
+    if (current.length > 0) {
+      tokens.push(current);
+      current = "";
+    }
+  }
+  if (current.length > 0) {
+    tokens.push(current);
+  }
+  return tokens;
 }
 function isValidUrl(value) {
   try {
@@ -138543,7 +138581,7 @@ var init_package2 = __esm({
   "package.json"() {
     package_default2 = {
       name: "poe-code",
-      version: "3.0.348",
+      version: "3.0.349",
       description: "CLI tool to configure Poe API for developer workflows.",
       type: "module",
       main: "./dist/index.js",