viho-llm 1.0.1 → 1.0.6

package/index.js

@@ -514,6 +514,83 @@ 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;
+    const agentBreakCallback = agent.agentBreakCallback;
+
+    // go
+    agentStartCallback();
+    const agentResponse = await callLLM(agentRequestOptions);
+
+    // check
+    const breakAgent = agentEndCallback(agentResponse);
+    if (breakAgent) {
+      if (agentBreakCallback) agentBreakCallback();
+      break;
+    }
+  }
+};
+
 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.1",
+  "version": "1.0.6",
   "description": "Utility library for working with multiple LLM providers (Google Gemini and OpenAI), providing common tools and helpers for AI interactions",
   "keywords": [
     "llm",
@@ -68,5 +68,5 @@
       }
     }
   },
-  "gitHead": "6162788575d9b2b7ed08844d848be9482e5f6132"
+  "gitHead": "1c8edc1b39d4b2617ec3b6c6ed555e3c8a27a91d"
 }

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,29 @@
+// 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;
+    const agentBreakCallback = agent.agentBreakCallback;
+
+    // go
+    agentStartCallback();
+    const agentResponse = await callLLM(agentRequestOptions);
+
+    // check
+    const breakAgent = agentEndCallback(agentResponse);
+    if (breakAgent) {
+      if (agentBreakCallback) agentBreakCallback();
+      break;
+    }
+  }
+};

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';