openxgen 1.1.0 → 1.1.1

package/dist/index.js

@@ -1243,29 +1243,37 @@ var init_tools = __esm({
 var document_exports = {};
 __export(document_exports, {
   getDocumentInfo: () => getDocumentInfo,
+  listCollections: () => listCollections,
   listDocuments: () => listDocuments,
   uploadDocument: () => uploadDocument
 });
-import { createReadStream, statSync } from "fs";
-import { basename } from "path";
+async function listCollections() {
+  const client2 = getClient();
+  const res = await client2.get("/api/retrieval/collections");
+  return Array.isArray(res.data) ? res.data : res.data.collections ?? [];
+}
 async function listDocuments(collectionId) {
   const client2 = getClient();
-  const params = {};
-  if (collectionId) params.collection_id = collectionId;
-  const res = await client2.get("/api/documents/list", { params });
-  return res.data.documents ?? res.data ?? [];
+  try {
+    const params = {};
+    if (collectionId) params.collection_id = collectionId;
+    const res = await client2.get("/api/retrieval/documents/list", { params });
+    return res.data.documents ?? res.data ?? [];
+  } catch {
+    return [];
+  }
 }
 async function uploadDocument(filePath, collectionId, name) {
   const client2 = getClient();
+  const { createReadStream, statSync } = await import("fs");
+  const { basename } = await import("path");
   const stat = statSync(filePath);
   const fileName = name || basename(filePath);
-  const FormData = (await import("buffer")).Blob ? globalThis.FormData : null;
-  if (!FormData) throw new Error("FormData not available");
   const form = new FormData();
   const fileBlob = new Blob([createReadStream(filePath)]);
   form.append("file", fileBlob, fileName);
   if (collectionId) form.append("collection_id", collectionId);
-  const res = await client2.post("/api/documents/upload", form, {
+  const res = await client2.post("/api/retrieval/documents/upload", form, {
     headers: { "Content-Type": "multipart/form-data" },
     maxBodyLength: stat.size + 1024 * 1024
   });
@@ -1273,7 +1281,7 @@ async function uploadDocument(filePath, collectionId, name) {
 }
 async function getDocumentInfo(docId) {
   const client2 = getClient();
-  const res = await client2.get(`/api/documents/${docId}`);
+  const res = await client2.get(`/api/retrieval/documents/${docId}`);
   return res.data;
 }
 var init_document = __esm({
@@ -1283,50 +1291,6 @@ var init_document = __esm({
   }
 });
 
-// src/api/ontology.ts
-var ontology_exports = {};
-__export(ontology_exports, {
-  getGraphStats: () => getGraphStats,
-  listGraphs: () => listGraphs,
-  queryGraphRAG: () => queryGraphRAG,
-  queryGraphRAGMultiTurn: () => queryGraphRAGMultiTurn
-});
-async function queryGraphRAG(query, graphId, opts) {
-  const client2 = getClient();
-  const res = await client2.post("/api/graph-rag", {
-    query,
-    graph_id: graphId,
-    use_scs: opts?.scs ?? true
-  });
-  return res.data;
-}
-async function queryGraphRAGMultiTurn(query, sessionId, graphId, opts) {
-  const client2 = getClient();
-  const res = await client2.post("/api/graph-rag/multi-turn", {
-    query,
-    session_id: sessionId,
-    graph_id: graphId,
-    max_turns: opts?.maxTurns ?? 5
-  });
-  return res.data;
-}
-async function getGraphStats(graphId) {
-  const client2 = getClient();
-  const res = await client2.get(`/api/graph/${graphId}/stats`);
-  return res.data;
-}
-async function listGraphs() {
-  const client2 = getClient();
-  const res = await client2.get("/api/graph/list");
-  return res.data.graphs ?? res.data ?? [];
-}
-var init_ontology = __esm({
-  "src/api/ontology.ts"() {
-    "use strict";
-    init_client();
-  }
-});
-
 // src/agent/tools/xgen-api.ts
 async function execute8(name, args) {
   const server = getServer();
@@ -1342,10 +1306,8 @@ async function execute8(name, args) {
         return await workflowRun2(args);
       case "xgen_workflow_info":
         return await workflowInfo2(args);
-      case "xgen_doc_list":
-        return await docList(args);
-      case "xgen_ontology_query":
-        return await ontologyQuery(args);
+      case "xgen_collection_list":
+        return await collectionList();
       case "xgen_server_status":
         return await serverStatus();
       case "xgen_execution_history":
@@ -1397,24 +1359,15 @@ ID: ${detail.id}
 \uB178\uB4DC: ${nodes}\uAC1C
 \uC5E3\uC9C0: ${edges}\uAC1C`;
 }
-async function docList(args) {
-  const { listDocuments: listDocuments2 } = await Promise.resolve().then(() => (init_document(), document_exports));
-  const docs = await listDocuments2(args.collection_id);
-  if (!docs.length) return "\uBB38\uC11C \uC5C6\uC74C.";
-  return docs.map(
-    (d, i) => `${i + 1}. ${d.file_name ?? d.name ?? "-"} (${d.file_type ?? "-"}) \u2014 ${d.status ?? "-"}`
-  ).join("\n");
-}
-async function ontologyQuery(args) {
-  const { queryGraphRAG: queryGraphRAG2 } = await Promise.resolve().then(() => (init_ontology(), ontology_exports));
-  const result = await queryGraphRAG2(args.query, args.graph_id);
-  let output = "";
-  if (result.answer) output += `\uB2F5\uBCC0: ${result.answer}
-`;
-  if (result.sources?.length) output += `\uCD9C\uCC98: ${result.sources.join(", ")}
-`;
-  if (result.triples_used?.length) output += `\uD2B8\uB9AC\uD50C: ${result.triples_used.join("; ")}`;
-  return output || "\uACB0\uACFC \uC5C6\uC74C.";
+async function collectionList() {
+  const { listCollections: listCollections2 } = await Promise.resolve().then(() => (init_document(), document_exports));
+  const cols = await listCollections2();
+  if (!cols.length) return "\uCEEC\uB809\uC158 \uC5C6\uC74C.";
+  return cols.map((c, i) => {
+    const shared = c.is_shared ? ` [\uACF5\uC720:${c.share_group}]` : "";
+    return `${i + 1}. ${c.collection_make_name}${shared}
+   \uBB38\uC11C: ${c.total_documents}\uAC1C \xB7 \uCCAD\uD06C: ${c.total_chunks}\uAC1C \xB7 \uBAA8\uB378: ${c.init_embedding_model ?? "-"}`;
+  }).join("\n");
 }
 async function serverStatus() {
   const server = getServer();
@@ -1482,29 +1435,9 @@ var init_xgen_api = __esm({
       {
         type: "function",
         function: {
-          name: "xgen_doc_list",
-          description: "XGEN \uC11C\uBC84\uC5D0\uC11C \uBB38\uC11C \uBAA9\uB85D\uC744 \uAC00\uC838\uC635\uB2C8\uB2E4.",
-          parameters: {
-            type: "object",
-            properties: {
-              collection_id: { type: "string", description: "\uCEEC\uB809\uC158 ID (\uC120\uD0DD)" }
-            }
-          }
-        }
-      },
-      {
-        type: "function",
-        function: {
-          name: "xgen_ontology_query",
-          description: "\uC628\uD1A8\uB85C\uC9C0(GraphRAG)\uC5D0 \uC9C8\uBB38\uD569\uB2C8\uB2E4. \uC9C0\uC2DD \uADF8\uB798\uD504 \uAE30\uBC18 \uAC80\uC0C9.",
-          parameters: {
-            type: "object",
-            properties: {
-              query: { type: "string", description: "\uC9C8\uC758 \uB0B4\uC6A9" },
-              graph_id: { type: "string", description: "\uADF8\uB798\uD504 ID (\uC120\uD0DD)" }
-            },
-            required: ["query"]
-          }
+          name: "xgen_collection_list",
+          description: "XGEN \uC11C\uBC84\uC758 \uBB38\uC11C \uCEEC\uB809\uC158(\uC9C0\uC2DD\uBCA0\uC774\uC2A4) \uBAA9\uB85D\uC744 \uAC00\uC838\uC635\uB2C8\uB2E4. \uBB38\uC11C \uC218, \uCCAD\uD06C \uC218, \uACF5\uC720 \uC0C1\uD0DC \uB4F1 \uD3EC\uD568.",
+          parameters: { type: "object", properties: {} }
         }
       },
       {
@@ -1703,31 +1636,29 @@ function buildSystemPrompt() {
   const server = getServer();
   const auth = getAuth();
   const env = getActiveEnvironment();
-  let prompt2 = `You are OPEN XGEN, an AI assistant in the user's terminal.
-You combine AI coding capabilities with the XGEN workflow platform.
+  let prompt2 = `You are OPEN XGEN. Terminal AI agent.
 
-## Capabilities
-1. **Coding**: Read/write files, execute commands, search code, run sandboxed code (JS/TS/Python)
-2. **XGEN Platform**: List/run workflows, manage documents, query ontology (GraphRAG)
+CRITICAL RULES:
+- Be extremely concise. No menus, no numbered lists of options, no "what would you like to do" questions.
+- Just DO things. If the user says "\uC6CC\uD06C\uD50C\uB85C\uC6B0 \uBAA9\uB85D" \u2192 call xgen_workflow_list immediately, show results.
+- If the user says a number after seeing a list, treat it as selection and act on it.
+- If the user says "\uC2E4\uD589" or "run" \u2192 call the tool immediately with the context you have.
+- Never ask "which option do you prefer" or show menus. Infer intent and act.
+- Respond in the user's language. Korean if they speak Korean.
+- Max 2-3 sentences per response unless showing data.
 
-## Rules
-- Respond in the same language as the user
-- Be concise. Show what you did, not how.
-- When using tools, briefly describe what you're doing
-- For XGEN operations, use the xgen_* tools`;
+TOOLS:
+- Coding: file_read, file_write, file_edit, bash, grep, list_files, sandbox_run
+- XGEN: xgen_workflow_list, xgen_workflow_run, xgen_workflow_info, xgen_doc_list, xgen_ontology_query, xgen_server_status, xgen_execution_history`;
   if (server && auth) {
     prompt2 += `
 
-## XGEN Server Connected
-- Server: ${server}
-- User: ${auth.username} (ID: ${auth.userId})
-- Environment: ${env?.name ?? "default"}
-You can use xgen_workflow_list, xgen_workflow_run, xgen_doc_list, xgen_ontology_query, etc.`;
+XGEN CONNECTED: ${server} as ${auth.username} (${env?.name ?? "default"})
+- Workflow execute uses deploy_key for deployed workflows.
+- If workflow execution returns 404, it means the Istio routing blocks direct stream access. Use deploy endpoint.`;
   } else {
     prompt2 += `
-
-## XGEN Server: Not connected
-Tell the user to run /connect to connect to an XGEN server.`;
+XGEN: Not connected. Tell user to run /connect.`;
   }
   return prompt2;
 }
@@ -3121,10 +3052,38 @@ function registerDocCommand(program2) {
 
 // src/commands/ontology.ts
 init_store();
-init_ontology();
-init_format();
 import chalk14 from "chalk";
 import { createInterface as createInterface6 } from "readline";
+
+// src/api/ontology.ts
+init_client();
+async function queryGraphRAG(query, graphId, opts) {
+  const client2 = getClient();
+  const res = await client2.post("/api/graph-rag", {
+    query,
+    graph_id: graphId,
+    use_scs: opts?.scs ?? true
+  });
+  return res.data;
+}
+async function queryGraphRAGMultiTurn(query, sessionId, graphId, opts) {
+  const client2 = getClient();
+  const res = await client2.post("/api/graph-rag/multi-turn", {
+    query,
+    session_id: sessionId,
+    graph_id: graphId,
+    max_turns: opts?.maxTurns ?? 5
+  });
+  return res.data;
+}
+async function getGraphStats(graphId) {
+  const client2 = getClient();
+  const res = await client2.get(`/api/graph/${graphId}/stats`);
+  return res.data;
+}
+
+// src/commands/ontology.ts
+init_format();
 import { randomUUID as randomUUID3 } from "crypto";
 function registerOntologyCommand(program2) {
   const ont = program2.command("ontology").alias("ont").description("\uC628\uD1A8\uB85C\uC9C0 GraphRAG \uC9C8\uC758");