agentnet 0.1.12 → 0.1.14

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentnet",
-  "version": "0.1.12",
+  "version": "0.1.14",
   "type": "module",
   "description": "Agent library used by Smartness",
   "main": "src/index.js",

package/src/agent/executor.js

@@ -267,7 +267,7 @@ export async function build(
 				logger.debug(`LLM response received for agent ${agentName}`);
 				
 				// Process the response
-				const finished = await api.onResponse(state, contents, toolsAndHandoffsMap, response, run);
+				const finished = await api.onResponse(state, contents, toolsAndHandoffsMap, response);
 				
 				// If not finished, continue with the next run
 				if (finished == null) {

package/src/llm/gemini.js

@@ -66,6 +66,7 @@ class GeminiLLM extends BaseLLM {
     
     try {
       const res = await client.models.generateContent(input);
+      console.log(JSON.stringify(res, null, 2))
       logger.debug('Gemini response received', {
         responseType: res.response?.candidates ? 'candidates' : 'unknown',
         hasContent: !!res.response?.candidates?.[0]?.content
@@ -149,17 +150,11 @@ class GeminiLLM extends BaseLLM {
    * @param {Object} response - The model response to process
    * @returns {Promise<string|null>} Text response or null if processing tool calls
    */
-  async onResponse(state, conversation, toolsAndHandoffsMap, response, run, maxRuns) {
+  async onResponse(state, conversation, toolsAndHandoffsMap, response) {
     // Handle simple text response
-    logger.info('Gemini onResponse', {text: response.text !== undefined, functionCalls: response.functionCalls?.length, run: run, maxRuns: maxRuns });
+    logger.info('Gemini onResponse', {text: response.text !== undefined, functionCalls: response.functionCalls?.length });
 
-    if (run > 4) {
-      console.log(JSON.stringify(response, null, 2))
-    }
-
-    if (response.text !== undefined && (response.functionCalls === undefined || response.functionCalls.length === 0 || run >= maxRuns - 1) ) {
-      logger.debug('Gemini response contains text, returning directly');
-      
+    if (response.text !== undefined) {
       if (conversation instanceof Conversation) {
         const modelResponse = {
           role: 'model',
@@ -167,7 +162,10 @@ class GeminiLLM extends BaseLLM {
         };
         conversation.addModelResponse(modelResponse);
       }
-      
+    }
+
+    if (response.text !== undefined && (response.functionCalls === undefined || response.functionCalls.length === 0)) {
+      logger.debug('Gemini response contains text, returning directly');
       return response.text;
     }
     

package/src/llm/gpt.js

@@ -156,7 +156,7 @@ class OpenAILLM extends BaseLLM {
 	 * @param {Object} response - The model response to process
 	 * @returns {Promise<string|null>} Text response or null if processing tool calls
 	 */
-	async onResponse(state, conversation, toolsAndHandoffsMap, response, run) { // TODO verify run and tool calls like gemini
+	async onResponse(state, conversation, toolsAndHandoffsMap, response) { // TODO verify run and tool calls like gemini
 		if (response.output_text !== undefined && response.output_text.length > 0 ) {
 			logger.debug('OpenAI response contains text, returning directly');