cite-agent 1.2.8__tar.gz → 1.2.9__tar.gz

{cite_agent-1.2.8/cite_agent.egg-info → cite_agent-1.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cite-agent
-Version: 1.2.8
+Version: 1.2.9
 Summary: Terminal AI assistant for academic research with citation verification
 Home-page: https://github.com/Spectating101/cite-agent
 Author: Cite-Agent Team

cite_agent-1.2.9/cite_agent/__version__.py

@@ -0,0 +1 @@
+__version__ = "1.2.9"

{cite_agent-1.2.8 → cite_agent-1.2.9}/cite_agent/cli.py

@@ -91,7 +91,6 @@ class NocturnalCLI:
 
         if not config.check_setup():
             # Check if we have env vars or session file (non-interactive mode)
-            import os
             from pathlib import Path
             session_file = Path.home() / ".nocturnal_archive" / "session.json"
             has_env_creds = os.getenv("NOCTURNAL_ACCOUNT_EMAIL") and os.getenv("NOCTURNAL_ACCOUNT_PASSWORD")
@@ -110,7 +109,7 @@ class NocturnalCLI:
                     self.console.print("[error]❌ Not authenticated. Run 'cite-agent --setup' to configure.[/error]")
                     return False
                 
-                self.console.print("\n[warning]👋 Hey there, looks like this machine hasn't met Nocturnal yet.[/warning]")
+                self.console.print("\n[warning]👋 Hey there, looks like this machine hasn't been set up yet.[/warning]")
                 self.console.print("[banner]Let's get you signed in — this only takes a minute.[/banner]")
                 try:
                     if not config.interactive_setup():
@@ -748,7 +747,7 @@ Examples:
     
     # Handle version
     if args.version:
-        print("Cite Agent v1.2.8")
+        print("Cite Agent v1.2.9")
         print("AI Research Assistant with real data integration")
         return
 

{cite_agent-1.2.8 → cite_agent-1.2.9}/cite_agent/enhanced_ai_agent.py

@@ -134,27 +134,70 @@ class EnhancedNocturnalAgent:
 
     def _load_authentication(self):
         """Load authentication from session file"""
-        use_local_keys = os.getenv("USE_LOCAL_KEYS", "true").lower() == "true"
+        use_local_keys = os.getenv("USE_LOCAL_KEYS", "false").lower() == "true"
+        
+        debug_mode = os.getenv("NOCTURNAL_DEBUG", "").lower() == "1"
+        if debug_mode:
+            print(f"🔍 _load_authentication: USE_LOCAL_KEYS={os.getenv('USE_LOCAL_KEYS')}, use_local_keys={use_local_keys}")
         
         if not use_local_keys:
             # Backend mode - load auth token from session
             from pathlib import Path
             session_file = Path.home() / ".nocturnal_archive" / "session.json"
+            if debug_mode:
+                print(f"🔍 _load_authentication: session_file exists={session_file.exists()}")
             if session_file.exists():
                 try:
                     import json
                     with open(session_file, 'r') as f:
                         session_data = json.load(f)
-                        self.auth_token = session_data.get('access_token')
-                        self.user_id = session_data.get('user_id')
-                except Exception:
+                        self.auth_token = session_data.get('auth_token')
+                        self.user_id = session_data.get('account_id')
+                        if debug_mode:
+                            print(f"🔍 _load_authentication: loaded auth_token={self.auth_token}, user_id={self.user_id}")
+                except Exception as e:
+                    if debug_mode:
+                        print(f"🔍 _load_authentication: ERROR loading session: {e}")
                     self.auth_token = None
                     self.user_id = None
             else:
-                self.auth_token = None
-                self.user_id = None
+                # FALLBACK: Check if config.env has credentials but session.json is missing
+                # This handles cases where old setup didn't create session.json
+                import json
+                email = os.getenv("NOCTURNAL_ACCOUNT_EMAIL")
+                account_id = os.getenv("NOCTURNAL_ACCOUNT_ID")
+                auth_token = os.getenv("NOCTURNAL_AUTH_TOKEN")
+                
+                if email and account_id and auth_token:
+                    # Auto-create session.json from config.env
+                    try:
+                        session_data = {
+                            "email": email,
+                            "account_id": account_id,
+                            "auth_token": auth_token,
+                            "refresh_token": "auto_generated",
+                            "issued_at": datetime.now(timezone.utc).isoformat()
+                        }
+                        session_file.parent.mkdir(parents=True, exist_ok=True)
+                        session_file.write_text(json.dumps(session_data, indent=2))
+                        
+                        self.auth_token = auth_token
+                        self.user_id = account_id
+                        
+                        if debug_mode:
+                            print(f"🔍 _load_authentication: Auto-created session.json from config.env")
+                    except Exception as e:
+                        if debug_mode:
+                            print(f"🔍 _load_authentication: Failed to auto-create session: {e}")
+                        self.auth_token = None
+                        self.user_id = None
+                else:
+                    self.auth_token = None
+                    self.user_id = None
         else:
             # Local keys mode
+            if debug_mode:
+                print(f"🔍 _load_authentication: Local keys mode, not loading session")
             self.auth_token = None
             self.user_id = None
         self._session_topics: Dict[str, Dict[str, Any]] = {}
@@ -922,10 +965,11 @@ class EnhancedNocturnalAgent:
                 )
             else:
                 intro = (
-                    "You are Cite Agent, a truth-seeking research and finance AI. "
-                    "PRIMARY DIRECTIVE: Accuracy > Agreeableness. Ask clarifying questions when context is missing. "
-                    "You are a fact-checker and analyst, NOT a people-pleaser. "
-                    "You have access to research (Archive) and financial data (FinSight SEC filings)."
+                    "You are Cite Agent, a truth-seeking research and finance AI with CODE EXECUTION. "
+                    "PRIMARY DIRECTIVE: Accuracy > Agreeableness. Execute code for analysis, calculations, and file operations. "
+                    "You are a fact-checker and analyst with a persistent shell session. "
+                    "You have access to research (Archive), financial data (FinSight SEC filings), and can run Python/R/SQL/Bash. "
+                    "When user asks about files, directories, or data: EXECUTE commands to find answers."
                 )
         
         sections.append(intro)
@@ -1475,8 +1519,8 @@ class EnhancedNocturnalAgent:
                         import json
                         with open(session_file, 'r') as f:
                             session_data = json.load(f)
-                            self.auth_token = session_data.get('access_token')
-                            self.user_id = session_data.get('user_id')
+                            self.auth_token = session_data.get('auth_token')
+                            self.user_id = session_data.get('account_id')
                     except Exception:
                         self.auth_token = None
                         self.user_id = None
@@ -1538,6 +1582,8 @@ class EnhancedNocturnalAgent:
                     except Exception as e:
                         print(f"⚠️ Failed to initialize {self.llm_provider.upper()} client: {e}")
 
+            # Initialize shell session for BOTH production and dev mode
+            # Production users need code execution too (like Cursor/Aider)
             if self.shell_session and self.shell_session.poll() is not None:
                 self.shell_session = None
 
@@ -1596,6 +1642,11 @@ class EnhancedNocturnalAgent:
         This is the SECURE method - all API keys stay on server
         Includes API results (Archive, FinSight) in context for better responses
         """
+        # DEBUG: Print auth status
+        debug_mode = os.getenv("NOCTURNAL_DEBUG", "").lower() == "1"
+        if debug_mode:
+            print(f"🔍 call_backend_query: auth_token={self.auth_token}, user_id={self.user_id}")
+        
         if not self.auth_token:
             return ChatResponse(
                 response="❌ Not authenticated. Please log in first.",
@@ -2654,14 +2705,123 @@ class EnhancedNocturnalAgent:
                         if debug_mode:
                             print(f"🔍 Web search failed: {e}")
             
-            # PRODUCTION MODE: Send to backend LLM with API results
+            # PRODUCTION MODE: Check for shell/code execution needs FIRST
             if self.client is None:
-                return await self.call_backend_query(
+                # Check if query needs directory/file info or exploration
+                question_lower = request.question.lower()
+                
+                # Basic info queries
+                needs_shell_info = any(phrase in question_lower for phrase in [
+                    'directory', 'folder', 'where am i', 'pwd', 'current location',
+                    'list files', 'what files', 'ls', 'files in', 'show files',
+                    'data files', 'csv files', 'check if file', 'file exists'
+                ])
+                
+                # Fuzzy search queries (find similar directories/files)
+                needs_find = any(phrase in question_lower for phrase in [
+                    'looking for', 'find', 'search for', 'similar to',
+                    'go to', 'cd to', 'navigate to', 'or something', 'forgot the name',
+                    'look into', 'check what', 'what\'s in'
+                ])
+                
+                if (needs_shell_info or needs_find) and self.shell_session:
+                    # Execute exploration commands
+                    try:
+                        api_results["shell_info"] = {}
+                        
+                        # Always include current location
+                        pwd_output = self.execute_command("pwd")
+                        api_results["shell_info"]["current_directory"] = pwd_output.strip()
+                        
+                        if needs_shell_info and not needs_find:
+                            # Just list current directory
+                            ls_output = self.execute_command("ls -lah")
+                            api_results["shell_info"]["directory_contents"] = ls_output
+                        
+                        if needs_find:
+                            # Smart search: extract directory name and location hints
+                            import re
+                            
+                            # Check if user is referring to previous context ("it", "there")
+                            has_pronoun = any(word in question_lower for word in ['it', 'there', 'that folder', 'that directory'])
+                            
+                            if has_pronoun and len(self.conversation_history) > 0:
+                                # Look for directory path in last assistant message
+                                last_assistant = None
+                                for msg in reversed(self.conversation_history):
+                                    if msg.get('role') == 'assistant':
+                                        last_assistant = msg.get('content', '')
+                                        break
+                                
+                                if last_assistant:
+                                    # Extract paths like /home/user/Downloads/cm522-main
+                                    paths = re.findall(r'(/[\w/.-]+)', last_assistant)
+                                    if paths:
+                                        # List contents of the first path found
+                                        target_path = paths[0]
+                                        ls_output = self.execute_command(f"ls -lah {target_path}")
+                                        api_results["shell_info"]["directory_contents"] = ls_output
+                                        api_results["shell_info"]["target_path"] = target_path
+                                        tools_used.append("shell_execution")
+                                        # Skip generic search
+                                        break
+                            
+                            # Generic search if no pronoun or no path found
+                            # Common words to ignore
+                            ignore_words = {
+                                'looking', 'find', 'folder', 'directory', 'called', 'something',
+                                'forgot', 'name', 'think', 'can', 'you', 'look', 'for', 'somewhere',
+                                'computer', 'downloads', 'the', 'this', 'that', 'class', 'investment',
+                                'check', 'what', 'into'
+                            }
+                            
+                            # Extract potential target names (prefer short alphanumeric codes)
+                            all_words = re.findall(r'\b([a-zA-Z0-9_-]+)\b', request.question)
+                            potential_names = [w for w in all_words if len(w) >= 2 and w.lower() not in ignore_words]
+                            
+                            # Detect location hints
+                            search_path = "~"  # Default to home
+                            if 'downloads' in question_lower:
+                                search_path = "~/Downloads"
+                            elif 'documents' in question_lower:
+                                search_path = "~/Documents"
+                            
+                            search_results = []
+                            searched_terms = []
+                            
+                            for name in potential_names[:2]:  # Limit to 2 best candidates
+                                if name in searched_terms:
+                                    continue
+                                searched_terms.append(name)
+                                
+                                # Search with increasing depth if needed
+                                find_output = self.execute_command(f"find {search_path} -maxdepth 4 -type d -iname '*{name}*' 2>/dev/null | head -20")
+                                if find_output.strip():
+                                    search_results.append(f"Searched for '*{name}*' in {search_path}:\n{find_output}")
+                            
+                            if search_results:
+                                api_results["shell_info"]["search_results"] = "\n\n".join(search_results)
+                            elif potential_names:
+                                api_results["shell_info"]["search_results"] = f"No directories matching '{', '.join(searched_terms)}' found in {search_path}"
+                        
+                        tools_used.append("shell_execution")
+                    except Exception as e:
+                        if debug_mode:
+                            print(f"🔍 Shell execution failed: {e}")
+                
+                # Call backend and UPDATE CONVERSATION HISTORY
+                response = await self.call_backend_query(
                     query=request.question,
                     conversation_history=self.conversation_history[-10:],
                     api_results=api_results,  # Include the data!
                     tools_used=tools_used  # Pass tools list for history
                 )
+                
+                # CRITICAL: Save to conversation history for context
+                self.conversation_history.append({"role": "user", "content": request.question})
+                self.conversation_history.append({"role": "assistant", "content": response.response})
+                
+                return response
 
             # DEV MODE ONLY: Direct Groq calls (only works with local API keys)
             # This code path won't execute in production since self.client = None

{cite_agent-1.2.8 → cite_agent-1.2.9}/cite_agent/session_manager.py

@@ -179,36 +179,45 @@ class SessionManager:
     
     def setup_environment_variables(self):
         """Set up environment variables for backend mode"""
-        # Load .env.local FIRST if it exists (for dev mode)
-        try:
-            from dotenv import load_dotenv
-            from pathlib import Path
-            env_local = Path.home() / ".nocturnal_archive" / ".env.local"
-            if env_local.exists():
-                load_dotenv(env_local)
-                import os
-                debug = os.getenv("NOCTURNAL_DEBUG", "").lower() == "1"
-                if debug:
-                    print(f"🔍 Loaded .env.local: USE_LOCAL_KEYS={os.getenv('USE_LOCAL_KEYS')}")
-        except Exception as e:
-            import os
-            debug = os.getenv("NOCTURNAL_DEBUG", "").lower() == "1"
-            if debug:
-                print(f"⚠️ Failed to load .env.local: {e}")
-        
-        # Check if dev mode is enabled
-        dev_mode = os.getenv("CITE_AGENT_DEV_MODE", "").lower() == "true"
-        
-        if not dev_mode:
-            # PRODUCTION MODE: Force backend, ensure monetization
-            # Set backend URL if not already set
+        # PRIORITY 1: Check if user has production credentials
+        # If they do, force production mode (ignore .env.local)
+        from pathlib import Path
+        session_file = Path.home() / ".nocturnal_archive" / "session.json"
+        has_session = session_file.exists()
+        has_config_creds = (
+            os.getenv("NOCTURNAL_ACCOUNT_EMAIL") and 
+            os.getenv("NOCTURNAL_AUTH_TOKEN")
+        )
+        
+        if has_session or has_config_creds:
+            # User is logged in → FORCE production mode
+            os.environ["USE_LOCAL_KEYS"] = "false"
             if "NOCTURNAL_API_URL" not in os.environ:
                 os.environ["NOCTURNAL_API_URL"] = "https://cite-agent-api-720dfadd602c.herokuapp.com/api"
             
-            # SECURITY: Default to backend mode (USE_LOCAL_KEYS=false)
-            # This ensures users MUST authenticate and pay
+            debug = os.getenv("NOCTURNAL_DEBUG", "").lower() == "1"
+            if debug:
+                print(f"🔍 Production mode: User has credentials, ignoring .env.local")
+        else:
+            # No production credentials → Allow dev mode from .env.local
+            try:
+                from dotenv import load_dotenv
+                env_local = Path.home() / ".nocturnal_archive" / ".env.local"
+                if env_local.exists():
+                    load_dotenv(env_local)
+                    debug = os.getenv("NOCTURNAL_DEBUG", "").lower() == "1"
+                    if debug:
+                        print(f"🔍 Loaded .env.local: USE_LOCAL_KEYS={os.getenv('USE_LOCAL_KEYS')}")
+            except Exception as e:
+                debug = os.getenv("NOCTURNAL_DEBUG", "").lower() == "1"
+                if debug:
+                    print(f"⚠️ Failed to load .env.local: {e}")
+            
+            # Default to production if no explicit dev mode
             if "USE_LOCAL_KEYS" not in os.environ:
                 os.environ["USE_LOCAL_KEYS"] = "false"
+            if "NOCTURNAL_API_URL" not in os.environ:
+                os.environ["NOCTURNAL_API_URL"] = "https://cite-agent-api-720dfadd602c.herokuapp.com/api"
     
     def get_session_status(self) -> Dict[str, Any]:
         """Get current session status for debugging"""

{cite_agent-1.2.8 → cite_agent-1.2.9}/cite_agent/setup_config.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 """
-Automatic setup and configuration for Nocturnal Archive
+Automatic setup and configuration for Cite Agent
 """
 
 import os
@@ -11,7 +11,7 @@ from typing import Optional, Dict, Any, List, Tuple
 from .account_client import AccountClient, AccountCredentials, AccountProvisioningError
 
 KEY_PLACEHOLDER = "__KEYRING__"
-KEYRING_SERVICE = "Nocturnal Archive"
+KEYRING_SERVICE = "Cite Agent"
 DEFAULT_QUERY_LIMIT = 25
 
 # Production: Users don't need API keys - backend has them
@@ -50,15 +50,29 @@ class NocturnalConfig:
     
     def interactive_setup(self) -> bool:
         """Interactive setup for account authentication and configuration."""
-        print("🚀 Nocturnal Archive Beta Setup")
+        print("🚀 Cite Agent Beta Setup")
         print("=" * 40)
         print()
 
-        if self.config_file.exists():
+        # Check if session exists
+        from pathlib import Path
+        session_file = Path.home() / ".nocturnal_archive" / "session.json"
+        
+        # Check if we have actual credentials (not just auto-generated config)
+        existing_config = self.load_config()
+        has_credentials = (
+            existing_config.get("NOCTURNAL_ACCOUNT_EMAIL") or 
+            existing_config.get("NOCTURNAL_AUTH_TOKEN")
+        )
+        
+        if session_file.exists() and has_credentials:
             print("✅ Configuration already exists!")
             response = input("Do you want to reconfigure? (y/N): ").strip().lower()
             if response not in ["y", "yes"]:
                 return True
+        elif has_credentials and not session_file.exists():
+            print("⚠️  Found saved credentials but no active session.")
+            print("Let's get you authenticated...")
 
         print("You'll use your institution-issued account to sign in. No invite codes or manual API keys required.")
         print()
@@ -81,6 +95,18 @@ class NocturnalConfig:
             print(f"❌ Could not verify your account: {exc}")
             return False
 
+        # Save session.json for authentication
+        import json
+        session_data = {
+            "email": credentials.email,
+            "account_id": credentials.account_id,
+            "auth_token": credentials.auth_token,
+            "refresh_token": credentials.refresh_token,
+            "issued_at": credentials.issued_at
+        }
+        session_file.parent.mkdir(parents=True, exist_ok=True)
+        session_file.write_text(json.dumps(session_data, indent=2))
+
         print("\n🛡️  Recap of beta limitations:")
         for item in self._beta_limitations():
             print(f" • {item}")
@@ -103,27 +129,10 @@ class NocturnalConfig:
             "NOCTURNAL_CONFIG_VERSION": "2.0.0",
         }
 
-        optional_updates = self._configure_optional_secrets(existing_config=config)
-        config.update(optional_updates)
-
         self.save_config(config)
 
         print("\n✅ Configuration saved successfully!")
-        print(f"📁 Config location: {self.config_file}")
-        print("\n🎉 You're ready to use Nocturnal Archive!")
-        print("\nQuick start:")
-        print("```python")
-        print("from nocturnal_archive import EnhancedNocturnalAgent, ChatRequest")
-        print("import asyncio")
-        print()
-        print("async def main():")
-        print("    agent = EnhancedNocturnalAgent()")
-        print("    await agent.initialize()")
-        print("    # Your code here...")
-        print("    await agent.close()")
-        print()
-        print("asyncio.run(main())")
-        print("```")
+        print("🎉 You're ready to use Cite Agent!")
 
         return True
 
@@ -312,7 +321,7 @@ class NocturnalConfig:
     def save_config(self, config: Dict[str, Any]):
         """Save configuration to file"""
         with open(self.config_file, 'w') as f:
-            f.write("# Nocturnal Archive Configuration\n")
+            f.write("# Cite Agent Configuration\n")
             f.write("# Generated automatically - do not edit manually\n\n")
             
             for key, value in config.items():
@@ -406,7 +415,7 @@ def auto_setup():
         return True
     
     # If no config exists, run interactive setup
-    print("🔧 Nocturnal Archive needs initial setup")
+    print("🔧 Cite Agent needs initial setup")
     return config.interactive_setup()
 
 def get_config():

{cite_agent-1.2.8 → cite_agent-1.2.9}/cite_agent/updater.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 """
-Auto-updater for Nocturnal Archive
+Auto-updater for Cite Agent
 Checks for updates and handles installation
 """
 
@@ -28,7 +28,7 @@ except ImportError:
         get_version = None
 
 class NocturnalUpdater:
-    """Handles automatic updates for Nocturnal Archive"""
+    """Handles automatic updates for Cite Agent"""
     
     def __init__(self):
         self.current_version = self.get_current_version()
@@ -185,7 +185,7 @@ def main():
     """CLI for updater"""
     import argparse
     
-    parser = argparse.ArgumentParser(description="Nocturnal Archive Updater")
+    parser = argparse.ArgumentParser(description="Cite Agent Updater")
     parser.add_argument("--check", action="store_true", help="Check for updates")
     parser.add_argument("--update", action="store_true", help="Update to latest version")
     parser.add_argument("--force", action="store_true", help="Force update even if up to date")

{cite_agent-1.2.8 → cite_agent-1.2.9/cite_agent.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cite-agent
-Version: 1.2.8
+Version: 1.2.9
 Summary: Terminal AI assistant for academic research with citation verification
 Home-page: https://github.com/Spectating101/cite-agent
 Author: Cite-Agent Team

{cite_agent-1.2.8 → cite_agent-1.2.9}/setup.py

@@ -7,7 +7,7 @@ long_description = readme_path.read_text() if readme_path.exists() else "Termina
 
 setup(
     name="cite-agent",
-    version="1.2.8",
+    version="1.2.9",
     author="Cite-Agent Team",
     author_email="contact@citeagent.dev",
     description="Terminal AI assistant for academic research with citation verification",

cite_agent-1.2.8/cite_agent/__version__.py

@@ -1 +0,0 @@
-__version__ = "1.2.8"