improved logging and cleanup

autonomys · Dec 25, 2024 · 0da26ab · 0da26ab
1 parent 9b3fabc
commit 0da26ab
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/auto-kol/agent/src/services/agents/nodes/autoApprovalNode.ts b/auto-kol/agent/src/services/agents/nodes/autoApprovalNode.ts
@@ -2,7 +2,7 @@ import { AIMessage } from '@langchain/core/messages';
 import { State, logger, parseMessageContent } from '../workflow.js';
 import * as prompts from '../prompts.js';
 import { WorkflowConfig } from '../workflow.js';
-import { getLastDsnCid, updateResponseStatusByTweetId } from '../../../database/index.js';
+import { updateResponseStatusByTweetId } from '../../../database/index.js';
 import { uploadToDsn } from '../../../utils/dsn.js';
 import { config as globalConfig } from '../../../config/index.js';
 import { ResponseStatus } from '../../../types/queue.js';

diff --git a/auto-kol/agent/src/services/agents/nodes/responseGenerationNode.ts b/auto-kol/agent/src/services/agents/nodes/responseGenerationNode.ts
@@ -20,6 +20,11 @@ export const createResponseGenerationNode = (config: WorkflowConfig) => {
       await Promise.all(
         batchToRespond.map(async (item: any) => {
           const { tweet, decision, toneAnalysis, workflowState } = item;
+          logger.info('Processing tweet:', {
+            id: tweet.id,
+            author: tweet.author_username,
+          });
+
 
           if (!workflowState) {
             item.workflowState = { autoFeedback: [] };
@@ -81,6 +86,12 @@ export const createResponseGenerationNode = (config: WorkflowConfig) => {
               rejectionInstructions,
             });
 
+          logger.info('Response strategy:', {
+            content: responseStrategy.content,
+            tone: responseStrategy.tone,
+            strategy: responseStrategy.strategy,
+          });
+
           const data = {
             type: ResponseStatus.PENDING,
             tweet,