aline-ai 0.6.2__py3-none-any.whl → 0.6.4__py3-none-any.whl

realign/events/event_summarizer.py

@@ -9,7 +9,7 @@ from ..db.sqlite_db import SQLiteDatabase
 from ..db.base import SessionRecord
 from ..db.locks import lease_lock, lock_key_for_event_summary, make_lock_owner
 from .debouncer import Debouncer
-from ..llm_client import call_llm, extract_json
+from ..llm_client import extract_json, call_llm_cloud
 
 logger = logging.getLogger(__name__)
 
@@ -201,41 +201,82 @@ def _generate_event_summary_llm(sessions: List[SessionRecord]) -> Tuple[str, str
             }
         )
 
-    system_prompt = _get_event_summary_prompt()
-
-    user_prompt = json.dumps(
-        {
-            "total_sessions": len(sessions),
-            "sessions": sessions_data,
-        },
-        ensure_ascii=False,
-        indent=2,
-    )
-
+    # Try cloud provider first if user is logged in
     try:
-        # Use unified LLM client
-        _, response = call_llm(
-            system_prompt=system_prompt,
-            user_prompt=user_prompt,
-            provider="auto",  # Try Claude first, fallback to OpenAI
-            max_tokens=500,
-            purpose="event_summary",
-        )
-
-        if not response:
-            logger.warning("LLM returned empty response, using fallback")
-            return _fallback_event_summary(sessions)
-
-        result = extract_json(response)
-
-        title = result.get("event_title", "Untitled Event")[:100]
-        description = result.get("event_description", "")
-
-        return title, description
-
-    except Exception as e:
-        logger.warning(f"LLM event summary failed, using fallback: {e}")
-        return _fallback_event_summary(sessions)
+        from ..auth import is_logged_in
+
+        if is_logged_in():
+            logger.debug("Attempting cloud LLM for event summary")
+            # Load user custom prompt if available
+            custom_prompt = None
+            user_prompt_path = Path.home() / ".aline" / "prompts" / "event_summary.md"
+            try:
+                if user_prompt_path.exists():
+                    custom_prompt = user_prompt_path.read_text(encoding="utf-8").strip()
+            except Exception:
+                pass
+
+            _, result = call_llm_cloud(
+                task="event_summary",
+                payload={"sessions": sessions_data},
+                custom_prompt=custom_prompt,
+                silent=True,
+            )
+
+            if result:
+                title = result.get("event_title", "Untitled Event")[:100]
+                description = result.get("event_description", "")
+                logger.info(f"Cloud LLM event summary success: title={title[:50]}...")
+                return title, description
+            else:
+                # Cloud LLM failed, use fallback (local fallback disabled)
+                logger.warning("Cloud LLM event summary failed, using fallback")
+                return _fallback_event_summary(sessions)
+    except ImportError:
+        logger.debug("Auth module not available, skipping cloud LLM")
+
+    # User not logged in, use fallback (local fallback disabled)
+    logger.warning("Not logged in, cannot use cloud LLM for event summary")
+    return _fallback_event_summary(sessions)
+
+    # =========================================================================
+    # LOCAL LLM FALLBACK DISABLED - Code kept for reference
+    # =========================================================================
+    # system_prompt = _get_event_summary_prompt()
+    #
+    # user_prompt = json.dumps(
+    #     {
+    #         "total_sessions": len(sessions),
+    #         "sessions": sessions_data,
+    #     },
+    #     ensure_ascii=False,
+    #     indent=2,
+    # )
+    #
+    # try:
+    #     # Use unified LLM client
+    #     _, response = call_llm(
+    #         system_prompt=system_prompt,
+    #         user_prompt=user_prompt,
+    #         provider="auto",  # Try Claude first, fallback to OpenAI
+    #         max_tokens=500,
+    #         purpose="event_summary",
+    #     )
+    #
+    #     if not response:
+    #         logger.warning("LLM returned empty response, using fallback")
+    #         return _fallback_event_summary(sessions)
+    #
+    #     result = extract_json(response)
+    #
+    #     title = result.get("event_title", "Untitled Event")[:100]
+    #     description = result.get("event_description", "")
+    #
+    #     return title, description
+    #
+    # except Exception as e:
+    #     logger.warning(f"LLM event summary failed, using fallback: {e}")
+    #     return _fallback_event_summary(sessions)
 
 
 def _fallback_event_summary(sessions: List[SessionRecord]) -> Tuple[str, str]:

realign/events/session_summarizer.py

@@ -10,7 +10,7 @@ from typing import List, Tuple, Optional
 from ..db.sqlite_db import SQLiteDatabase
 from ..db.base import TurnRecord
 from ..db.locks import lease_lock, lock_key_for_session_summary, make_lock_owner
-from ..llm_client import call_llm, extract_json
+from ..llm_client import extract_json, call_llm_cloud
 
 logger = logging.getLogger(__name__)
 
@@ -267,41 +267,82 @@ def _generate_session_summary_llm(turns: List[TurnRecord]) -> Tuple[str, str]:
             }
         )
 
-    system_prompt = _get_session_summary_prompt()
-
-    user_prompt = json.dumps(
-        {
-            "total_turns": len(turns),
-            "turns": turns_data,
-        },
-        ensure_ascii=False,
-        indent=2,
-    )
-
+    # Try cloud provider first if user is logged in
     try:
-        # Use unified LLM client
-        _, response = call_llm(
-            system_prompt=system_prompt,
-            user_prompt=user_prompt,
-            provider="auto",  # Try Claude first, fallback to OpenAI
-            max_tokens=500,
-            purpose="session_summary",
-        )
-
-        if not response:
-            logger.warning("LLM returned empty response, using fallback")
-            return _fallback_summary(turns)
-
-        result = extract_json(response)
+        from ..auth import is_logged_in
 
-        title = result.get("session_title", "Untitled Session")[:80]
-        summary = result.get("session_summary", "")
+        if is_logged_in():
+            logger.debug("Attempting cloud LLM for session summary")
+            # Load user custom prompt if available
+            custom_prompt = None
+            user_prompt_path = Path.home() / ".aline" / "prompts" / "session_summary.md"
+            try:
+                if user_prompt_path.exists():
+                    custom_prompt = user_prompt_path.read_text(encoding="utf-8").strip()
+            except Exception:
+                pass
 
-        return title, summary
+            _, result = call_llm_cloud(
+                task="session_summary",
+                payload={"turns": turns_data},
+                custom_prompt=custom_prompt,
+                silent=True,
+            )
 
-    except Exception as e:
-        logger.warning(f"LLM session summary failed, using fallback: {e}")
-        return _fallback_summary(turns)
+            if result:
+                title = result.get("session_title", "Untitled Session")[:80]
+                summary = result.get("session_summary", "")
+                logger.info(f"Cloud LLM session summary success: title={title[:50]}...")
+                return title, summary
+            else:
+                # Cloud LLM failed, use fallback (local fallback disabled)
+                logger.warning("Cloud LLM session summary failed, using fallback")
+                return _fallback_summary(turns)
+    except ImportError:
+        logger.debug("Auth module not available, skipping cloud LLM")
+
+    # User not logged in, use fallback (local fallback disabled)
+    logger.warning("Not logged in, cannot use cloud LLM for session summary")
+    return _fallback_summary(turns)
+
+    # =========================================================================
+    # LOCAL LLM FALLBACK DISABLED - Code kept for reference
+    # =========================================================================
+    # system_prompt = _get_session_summary_prompt()
+    #
+    # user_prompt = json.dumps(
+    #     {
+    #         "total_turns": len(turns),
+    #         "turns": turns_data,
+    #     },
+    #     ensure_ascii=False,
+    #     indent=2,
+    # )
+    #
+    # try:
+    #     # Use unified LLM client
+    #     _, response = call_llm(
+    #         system_prompt=system_prompt,
+    #         user_prompt=user_prompt,
+    #         provider="auto",  # Try Claude first, fallback to OpenAI
+    #         max_tokens=500,
+    #         purpose="session_summary",
+    #     )
+    #
+    #     if not response:
+    #         logger.warning("LLM returned empty response, using fallback")
+    #         return _fallback_summary(turns)
+    #
+    #     result = extract_json(response)
+    #
+    #     title = result.get("session_title", "Untitled Session")[:80]
+    #     summary = result.get("session_summary", "")
+    #
+    #     return title, summary
+    #
+    # except Exception as e:
+    #     logger.warning(f"LLM session summary failed, using fallback: {e}")
+    #     return _fallback_summary(turns)
 
 
 def _fallback_summary(turns: List[TurnRecord]) -> Tuple[str, str]: