bulltrackers-module 1.0.76 → 1.0.78

package/functions/social-task-handler/helpers/handler_helpers.js

@@ -21,11 +21,20 @@ async function getSentimentFromGemini(dependencies, snippet) {
         return 'Neutral';
     }
 
-    const prompt = `You are a financial sentiment analyzer for social media posts.
-Classify the following post as 'Bullish', 'Bearish', or 'Neutral'.
-Respond with only one of those three words.
+    const prompt = `You are a financial sentiment analyzer specializing in social media posts about trading.
+Your task is to determine the sentiment of the author toward the mentioned asset(s).
+
+Possible labels:
+- Bullish: Positive outlook, confidence, or optimism about the asset.
+- Bearish: Negative outlook, loss, or pessimism about the asset.
+- Neutral: Informational or balanced, with no clear positive or negative tone.
+
+Important: If the author reports selling at a loss, expresses regret, or uses negative emojis or language about the asset, classify as Bearish.
+
+Return only one word: Bullish, Bearish, or Neutral.
+
+Post: "${snippet}"`
 
-Post: "${snippet}"`;
 
     try {
         // --- START FIX: Corrected Request Payload ---
@@ -51,15 +60,15 @@ Post: "${snippet}"`;
 
         // --- START FIX: Corrected Response Parsing ---
         // The text is located at `content.parts[0].text`,
-        // not `content[0].text` as in the original file.
         const text = response?.candidates?.[0]?.content?.parts?.[0]?.text?.trim() || '';
         // --- END FIX ---
 
         // Robust parsing: Find the keyword, case-insensitive.
-        const match = text.match(/(Bullish|Bearish|Neutral)/i);
+        const match = text.match(/\b(Bullish|Bearish|Neutral)\b/i);
         if (match && match[0]) {
             // Capitalize first letter, e.g., "bullish" -> "Bullish"
             const sentiment = match[0].charAt(0).toUpperCase() + match[0].slice(1).toLowerCase();
+            logger.log('TRACE', `[getSentimentFromGemini] Raw Gemini response: ${JSON.stringify(response)}`);
             logger.log('INFO', `[getSentimentFromGemini] Classified sentiment: ${sentiment}`);
             return sentiment;
         }
@@ -193,6 +202,25 @@ exports.handleSocialTask = async (message, context, config, dependencies) => {
                 processedInThisRun.add(post.id);
                 const text = post.message.text;
 
+                // --- START: Enhanced Data Extraction ---
+                const likeCount = discussion.emotionsData?.like?.paging?.totalCount || 0;
+                const commentCount = discussion.summary?.totalCommentsAndReplies || 0;
+                let pollData = null;
+
+                // Check for poll data and extract it
+                if (post.type === 'Poll' && post.metadata?.poll?.options) {
+                    pollData = {
+                        id: post.metadata.poll.id,
+                        options: post.metadata.poll.options.map(opt => ({
+                            id: opt.id,
+                            text: opt.text,
+                            votesCount: opt.votesCount || 0
+                        }))
+                    };
+                }
+                // --- END: Enhanced Data Extraction ---
+
+
                 // 1. Truncate for AI
                 const MAX_CHARS = 500; // ~125 tokens, very cheap
                 let snippet = text;
@@ -212,9 +240,13 @@ exports.handleSocialTask = async (message, context, config, dependencies) => {
                 const postData = {
                     sentiment: sentiment,
                     textSnippet: snippet,
+                    fullText: text, // Store the full original text
                     language: lang,
                     tickers: post.tags.map(t => t.market?.symbolName).filter(Boolean),
                     postOwnerId: post.owner?.id,
+                    likeCount: likeCount, // Store like count
+                    commentCount: commentCount, // Store comment count
+                    pollData: pollData, // Store poll data (will be null if not a poll)
                     createdAt: post.created,
                     fetchedAt: FieldValue.serverTimestamp()
                 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bulltrackers-module",
-  "version": "1.0.76",
+  "version": "1.0.78",
   "description": "Helper Functions for Bulltrackers.",
   "main": "index.js",
   "files": [