viho-llm 1.0.0 → 1.0.5

package/index.js

@@ -514,6 +514,76 @@ const OpenAIAPI = (options) => {
   return openai;
 };
 
+/**
+ * callLLM
+ * @param {*} options
+ * @returns
+ */
+const callLLM = async (options) => {
+  const response = await options.llm.chat({
+    model: options.modelName,
+    messages: options.messages,
+  });
+  const fullContent = response.content || '';
+
+  // r
+  return options.isJson ? extractJSON(fullContent) : fullContent;
+};
+
+/**
+ * extractJSON
+ * @param {*} text
+ * @returns
+ */
+const extractJSON = (text) => {
+  const cleaned = text.replace(/\{\{/g, '{').replace(/\}\}/g, '}');
+  try {
+    return JSON.parse(cleaned);
+  } catch (e) {
+    // try next
+  }
+  const codeBlock = cleaned.match(/```(?:json)?\s*([\s\S]*?)```/);
+  if (codeBlock) {
+    try {
+      return JSON.parse(codeBlock[1].trim());
+    } catch (e) {
+      // try next
+    }
+  }
+  const match = cleaned.match(/(\{[\s\S]*\}|\[[\s\S]*\])/);
+  if (match) {
+    try {
+      return JSON.parse(match[1]);
+    } catch (e) {
+      // try next
+    }
+  }
+  throw new Error(`Cannot extract JSON from: ${text.slice(0, 200)}`);
+};
+
+// util
+/**
+ * runAgents
+ * @param {*} agents
+ */
+const runAgents = async (agents) => {
+  // for
+  for (let i = 0; i < agents.length; i++) {
+    const agent = agents[i];
+
+    // const
+    const agentStartCallback = agent.agentStartCallback;
+    const agentRequestOptions = agent.agentRequestOptions;
+    const agentEndCallback = agent.agentEndCallback;
+
+    // go
+    agentStartCallback();
+    const agentResponse = await callLLM(agentRequestOptions);
+    agentEndCallback(agentResponse);
+  }
+};
+
 exports.GeminiAPI = GeminiAPI;
 exports.GeminiVertex = GeminiVertex;
 exports.OpenAIAPI = OpenAIAPI;
+exports.runAgents = runAgents;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "viho-llm",
-  "version": "1.0.0",
+  "version": "1.0.5",
   "description": "Utility library for working with multiple LLM providers (Google Gemini and OpenAI), providing common tools and helpers for AI interactions",
   "keywords": [
     "llm",
@@ -46,9 +46,9 @@
     "build": "qpro rollup ./rollup.config.js"
   },
   "dependencies": {
-    "@google/genai": "^1.34.0",
-    "mime-types": "^2.1.35",
-    "openai": "^5.23.2",
+    "@google/genai": "^1.43.0",
+    "mime-types": "^3.0.2",
+    "openai": "^6.25.0",
     "qiao.log.js": "^3.7.5"
   },
   "nx": {
@@ -68,5 +68,5 @@
       }
     }
   },
-  "gitHead": "fecda99c334deba09750fc67b42ecc172b36a239"
+  "gitHead": "7b7a9c1c3c14b967a21a13a72728eebb6a339335"
 }

package/src/agents/agent-util.js

@@ -0,0 +1,46 @@
+/**
+ * callLLM
+ * @param {*} options
+ * @returns
+ */
+export const callLLM = async (options) => {
+  const response = await options.llm.chat({
+    model: options.modelName,
+    messages: options.messages,
+  });
+  const fullContent = response.content || '';
+
+  // r
+  return options.isJson ? extractJSON(fullContent) : fullContent;
+};
+
+/**
+ * extractJSON
+ * @param {*} text
+ * @returns
+ */
+export const extractJSON = (text) => {
+  const cleaned = text.replace(/\{\{/g, '{').replace(/\}\}/g, '}');
+  try {
+    return JSON.parse(cleaned);
+  } catch (e) {
+    // try next
+  }
+  const codeBlock = cleaned.match(/```(?:json)?\s*([\s\S]*?)```/);
+  if (codeBlock) {
+    try {
+      return JSON.parse(codeBlock[1].trim());
+    } catch (e) {
+      // try next
+    }
+  }
+  const match = cleaned.match(/(\{[\s\S]*\}|\[[\s\S]*\])/);
+  if (match) {
+    try {
+      return JSON.parse(match[1]);
+    } catch (e) {
+      // try next
+    }
+  }
+  throw new Error(`Cannot extract JSON from: ${text.slice(0, 200)}`);
+};

package/src/agents/agent.js

@@ -0,0 +1,22 @@
+// util
+import { callLLM } from './agent-util.js';
+/**
+ * runAgents
+ * @param {*} agents
+ */
+export const runAgents = async (agents) => {
+  // for
+  for (let i = 0; i < agents.length; i++) {
+    const agent = agents[i];
+
+    // const
+    const agentStartCallback = agent.agentStartCallback;
+    const agentRequestOptions = agent.agentRequestOptions;
+    const agentEndCallback = agent.agentEndCallback;
+
+    // go
+    agentStartCallback();
+    const agentResponse = await callLLM(agentRequestOptions);
+    agentEndCallback(agentResponse);
+  }
+};

package/src/index.js

@@ -1,3 +1,4 @@
 export * from './models/gemini-api.js';
 export * from './models/gemini-vertex.js';
 export * from './models/openai.js';
+export * from './agents/agent.js';