cite-agent 1.0.4__py3-none-any.whl → 1.2.3__py3-none-any.whl

src/services/simple_enhanced_main.py

@@ -0,0 +1,287 @@
+"""
+Simple Enhanced Main Application - Working version with core capabilities
+"""
+
+import asyncio
+import logging
+import os
+import time
+from datetime import datetime, timezone
+from fastapi import FastAPI, Request, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
+from pydantic import BaseModel
+from typing import Optional, Dict, Any
+
+# Import our enhanced services
+from services.reasoning_engine.reasoning_engine import ReasoningEngine
+from services.tool_framework.tool_manager import ToolManager
+from services.context_manager.advanced_context import AdvancedContextManager
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Initialize FastAPI app
+app = FastAPI(
+    title="Nocturnal Archive API - Enhanced",
+    description="Production-grade AI-powered research platform with advanced reasoning capabilities",
+    version="3.0.0",
+    docs_url="/docs",
+    redoc_url="/redoc"
+)
+
+
+def _utc_timestamp() -> str:
+    return datetime.now(timezone.utc).isoformat()
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Configure for production
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Initialize enhanced services
+tool_manager = ToolManager()
+context_manager = AdvancedContextManager()
+reasoning_engine = ReasoningEngine()
+
+# Pydantic models for request/response
+class ChatRequest(BaseModel):
+    message: str
+    session_id: Optional[str] = None
+    use_advanced_reasoning: Optional[bool] = True
+
+class ResearchRequest(BaseModel):
+    topic: str
+    max_results: Optional[int] = 10
+    use_advanced_reasoning: Optional[bool] = True
+
+class ReasoningRequest(BaseModel):
+    problem_description: str
+    context: Optional[Dict[str, Any]] = None
+    user_id: Optional[str] = None
+
+class ToolExecutionRequest(BaseModel):
+    tool_name: Optional[str] = None
+    task_description: str
+    context: Optional[Dict[str, Any]] = None
+    auto_select_tool: Optional[bool] = True
+
+# Health check endpoints
+@app.get("/")
+async def root():
+    """Root endpoint with service information."""
+    return {
+        "service": "Nocturnal Archive API - Enhanced",
+        "version": "3.0.0",
+        "status": "healthy",
+        "environment": os.getenv("ENVIRONMENT", "development"),
+        "capabilities": [
+            "Advanced Reasoning Engine",
+            "Dynamic Tool Framework", 
+            "Code Execution Environment",
+            "Advanced Context Management",
+            "Academic Research & Synthesis",
+            "Multi-LLM Integration"
+        ]
+    }
+
+@app.get("/health")
+async def health_check():
+    """Health check endpoint."""
+    return {
+        "status": "healthy", 
+        "timestamp": _utc_timestamp(),
+        "services": {
+            "reasoning_engine": "operational",
+            "tool_framework": "operational",
+            "context_manager": "operational"
+        }
+    }
+
+@app.get("/api/status")
+async def api_status():
+    """API status endpoint."""
+    return {
+        "status": "operational",
+        "services": {
+            "reasoning": "operational",
+            "tools": "operational",
+            "context": "operational"
+        }
+    }
+
+# Enhanced Reasoning Endpoints
+@app.post("/api/reasoning/solve")
+async def solve_problem(request: ReasoningRequest):
+    """Solve complex problems using advanced reasoning."""
+    try:
+        # Solve the problem using reasoning engine
+        result = await reasoning_engine.solve_problem(
+            problem_description=request.problem_description,
+            context=request.context,
+            user_id=request.user_id
+        )
+        
+        return {
+            "status": "success",
+            "result": result
+        }
+        
+    except Exception as e:
+        logger.error(f"Reasoning error: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Reasoning failed: {str(e)}")
+
+# Enhanced Tool Framework Endpoints
+@app.post("/api/tools/execute")
+async def execute_tool(request: ToolExecutionRequest):
+    """Execute a tool with dynamic selection."""
+    try:
+        # Execute tool
+        if request.auto_select_tool:
+            result = await tool_manager.execute_with_auto_selection(
+                task_description=request.task_description,
+                context=request.context
+            )
+        else:
+            if not request.tool_name:
+                raise HTTPException(status_code=400, detail="Tool name required when auto_select_tool is False")
+            
+            result = await tool_manager.execute_tool(
+                tool_name=request.tool_name,
+                task_description=request.task_description,
+                context=request.context
+            )
+        
+        return {
+            "status": "success",
+            "result": result
+        }
+        
+    except Exception as e:
+        logger.error(f"Tool execution error: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Tool execution failed: {str(e)}")
+
+@app.get("/api/tools/available")
+async def get_available_tools():
+    """Get list of available tools."""
+    try:
+        tools = tool_manager.get_available_tools()
+        tool_capabilities = {}
+        
+        for tool in tools:
+            tool_capabilities[tool] = tool_manager.get_tool_capabilities(tool)
+        
+        return {
+            "status": "success",
+            "tools": tools,
+            "capabilities": tool_capabilities
+        }
+    except Exception as e:
+        logger.error(f"Failed to get available tools: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Failed to get available tools: {str(e)}")
+
+# Enhanced Context Management Endpoints
+@app.post("/api/context/process")
+async def process_context(request: ChatRequest):
+    """Process interaction and update context."""
+    try:
+        session_id = request.session_id or "default_session"
+        
+        # Generate response (this would integrate with existing chat logic)
+        response = f"Enhanced response to: {request.message}"
+        
+        # Process interaction in context manager
+        result = await context_manager.process_interaction(
+            user_input=request.message,
+            response=response,
+            session_id=session_id,
+            user_id="anonymous"
+        )
+        
+        return {
+            "status": "success",
+            "response": response,
+            "context_result": result
+        }
+        
+    except Exception as e:
+        logger.error(f"Context processing error: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Context processing failed: {str(e)}")
+
+@app.get("/api/context/retrieve")
+async def retrieve_context(query: str, session_id: Optional[str] = None):
+    """Retrieve relevant context for a query."""
+    try:
+        result = await context_manager.retrieve_relevant_context(
+            query=query,
+            session_id=session_id,
+            user_id="anonymous"
+        )
+        
+        return {
+            "status": "success",
+            "result": result
+        }
+        
+    except Exception as e:
+        logger.error(f"Context retrieval error: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Context retrieval failed: {str(e)}")
+
+# Enhanced Chat Endpoint with Advanced Reasoning
+@app.post("/api/enhanced-chat")
+async def enhanced_chat_endpoint(request: ChatRequest):
+    """Enhanced chat endpoint with advanced reasoning capabilities."""
+    try:
+        session_id = request.session_id or "enhanced_session"
+        
+        # Use advanced reasoning if requested
+        if request.use_advanced_reasoning:
+            # Solve as a reasoning problem
+            reasoning_result = await reasoning_engine.solve_problem(
+                problem_description=request.message,
+                context={"session_id": session_id, "user_id": "anonymous"},
+                user_id="anonymous"
+            )
+            
+            response = reasoning_result.get("solution", "No solution generated")
+        else:
+            # Use simple response
+            response = f"Standard response to: {request.message}"
+        
+        # Process interaction in context manager
+        await context_manager.process_interaction(
+            user_input=request.message,
+            response=str(response),
+            session_id=session_id,
+            user_id="anonymous"
+        )
+        
+        return {
+            "response": response,
+            "session_id": session_id,
+            "timestamp": _utc_timestamp(),
+            "mode": "enhanced_reasoning" if request.use_advanced_reasoning else "standard"
+        }
+        
+    except Exception as e:
+        logger.error(f"Enhanced chat error: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Chat failed: {str(e)}")
+
+# Error handling
+@app.exception_handler(Exception)
+async def global_exception_handler(request: Request, exc: Exception):
+    """Global exception handler."""
+    logger.error(f"Unhandled exception: {str(exc)}")
+    return JSONResponse(
+        status_code=500,
+        content={"detail": "Internal server error"}
+    )
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="127.0.0.1", port=8003)

cite_agent/__distribution__.py

@@ -1,7 +0,0 @@
-"""
-This is a DISTRIBUTION build.
-Local LLM calling code has been removed.
-All queries must go through the centralized backend.
-"""
-DISTRIBUTION_BUILD = True
-BACKEND_ONLY = True

cite_agent-1.0.4.dist-info/METADATA

@@ -1,234 +0,0 @@
-Metadata-Version: 2.4
-Name: cite-agent
-Version: 1.0.4
-Summary: AI Research Assistant - Backend-Only Distribution
-Home-page: https://github.com/Spectating101/cite-agent
-Author: Cite-Agent Team
-Author-email: contact@citeagent.dev
-Classifier: Development Status :: 4 - Beta
-Classifier: Intended Audience :: Science/Research
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Requires-Python: >=3.9
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: requests>=2.31.0
-Requires-Dist: aiohttp>=3.9.0
-Requires-Dist: python-dotenv>=1.0.0
-Requires-Dist: pydantic>=2.5.0
-Requires-Dist: rich>=13.7.0
-Requires-Dist: keyring>=24.3.0
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: license-file
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
-
-# Cite-Agent
-
-> A fast, affordable AI research assistant with built-in citation capabilities
-
-**Cite-Agent** is a terminal-based AI assistant designed for research and finance, offering 70B model access at half the cost of comparable services.
-
----
-
-## 🎯 What is Cite-Agent?
-
-Cite-Agent is an AI-powered research tool that:
-- Answers complex questions using state-of-the-art 70B language models
-- Provides accurate, source-grounded responses (hence "cite")
-- Executes code (Python, R, SQL) for data analysis
-- Operates through a beautiful terminal interface
-- Costs $10/month (vs $20+ for Claude/ChatGPT)
-
----
-
-## ✨ Key Features
-
-### 🧠 **Powerful AI**
-- Access to Llama 3.3 70B via Cerebras & Groq
-- Multi-provider fallback for 99.9% uptime
-- 50 queries/day (50,000 tokens)
-- 2-5 second response times
-
-### 🎓 **Truth-Seeking**
-- Explicitly designed to correct user errors
-- Admits uncertainty instead of guessing
-- Cites sources and reasoning
-- Anti-appeasement prompts
-
-### 💻 **Code Execution**
-- Python, R, and SQL support
-- Data analysis and visualization
-- Financial calculations
-- Research automation
-
-### 🔒 **Secure & Private**
-- API keys never leave the backend
-- JWT-based authentication
-- Rate limiting per user
-- HTTPS-only communication
-
-### 📊 **Analytics**
-- Track your usage
-- Monitor token consumption
-- View query history
-- Download statistics
-
----
-
-## 💰 Pricing
-
-| Plan | Price | Queries/Day | Features |
-|------|-------|-------------|----------|
-| **Student** | $10/month (300 NTD) | 50 | Full access |
-| **Public** | $10/month (400 NTD) | 50 | Full access |
-
-**Beta**: First 50 users get 3 months free
-
----
-
-## 🚀 Quick Start
-
-### For Users
-
-1. **Download** for your OS:
-   - [Windows](https://cite-agent-api.herokuapp.com/api/downloads/windows)
-   - [macOS](https://cite-agent-api.herokuapp.com/api/downloads/macos)
-   - [Linux](https://cite-agent-api.herokuapp.com/api/downloads/linux)
-
-2. **Install** the downloaded package
-
-3. **Run** `cite-agent` in your terminal
-
-4. **Register** with email + password
-
-5. **Start asking questions!**
-
-### For Developers
-
-See [DEVELOPMENT.md](./docs/DEVELOPMENT.md) for setup instructions.
-
----
-
-## 🏗️ Architecture
-
-```
-User's Machine                Backend (Heroku)              AI Providers
-┌─────────────┐              ┌──────────────┐              ┌──────────┐
-│             │              │              │              │          │
-│   Terminal  │─────────────▶│   FastAPI    │─────────────▶│ Cerebras │
-│     UI      │              │     API      │              │  (70B)   │
-│             │◀─────────────│              │◀─────────────│          │
-│   (cite_    │              │  - Auth      │              └──────────┘
-│   agent/)   │              │  - Rate      │              
-│             │              │    Limiting  │              ┌──────────┐
-└─────────────┘              │  - Analytics │              │          │
-                             │  - Proxy     │─────────────▶│   Groq   │
-                             │              │              │  (70B)   │
-                             └──────────────┘              │          │
-                                    │                      └──────────┘
-                             ┌──────────────┐
-                             │  PostgreSQL  │
-                             │   Database   │
-                             └──────────────┘
-```
-
-### Components
-
-**Frontend (cite_agent/)**
-- Terminal UI using `rich` library
-- JWT authentication
-- Query management
-- Local session storage
-
-**Backend (cite-agent-api/)**
-- FastAPI REST API
-- User authentication & rate limiting
-- Multi-provider LLM routing
-- Analytics & tracking
-- PostgreSQL database
-
-**AI Providers**
-- **Primary**: Cerebras (14,400 RPD × 3 keys = 43,200/day)
-- **Backup**: Groq (1,000 RPD × 3 keys = 3,000/day)
-- **Total capacity**: ~46,000 queries/day
-
----
-
-## 📖 Documentation
-
-- **[DEVELOPMENT.md](./docs/DEVELOPMENT.md)** - Developer setup guide
-- **[DEPLOYMENT.md](./DEPLOYMENT.md)** - How to deploy (Heroku)
-- **[API_REFERENCE.md](./docs/API_REFERENCE.md)** - API endpoints
-- **[ARCHITECTURE.md](./ARCHITECTURE.md)** - System design details
-- **[ROADMAP.md](./ROADMAP.md)** - Future plans
-
----
-
-## 🛠️ Tech Stack
-
-**Frontend**
-- Python 3.13+
-- `rich` for terminal UI
-- `httpx` for async HTTP
-- `pydantic` for data validation
-
-**Backend**
-- FastAPI
-- PostgreSQL
-- JWT authentication
-- Multi-provider LLM integration
-
-**AI Providers**
-- Cerebras Inference API
-- Groq Cloud
-- Cloudflare Workers AI (fallback)
-- OpenRouter (fallback)
-
-**Deployment**
-- Heroku (backend + database)
-- GitHub Releases (installers)
-
----
-
-## 🤝 Contributing
-
-We're not accepting contributions yet (private beta), but stay tuned!
-
----
-
-## 📜 License
-
-Proprietary - All rights reserved
-
----
-
-## 📧 Support
-
-- **Email**: s1133958@mail.yzu.edu.tw
-- **Issues**: Coming soon
-- **Documentation**: See `/docs` directory
-
----
-
-## 🙏 Acknowledgments
-
-- **Cerebras** for generous free tier (14,400 RPD)
-- **Groq** for fast 70B inference
-- **GitHub Student Pack** for free hosting credits
-- **FastAPI** for excellent async framework
-
----
-
-**Built with ❤️ for researchers and analysts**
-
-*Cite-Agent - Because accuracy matters more than agreeableness*

cite_agent-1.0.4.dist-info/RECORD

@@ -1,23 +0,0 @@
-cite_agent/__distribution__.py,sha256=U7-p-qBMX7WrQD6WWjRC5b-PswXnlrqAox7EYnLogqI,178
-cite_agent/__init__.py,sha256=n12qGnmZ0th2OJTn5kBK-kzIEn_1EZgZ1qujT4-ZMeU,1643
-cite_agent/account_client.py,sha256=kvILYFKGm6ktCdi7iKSP5dkOp7iOAcBEQ7giHOh711w,6723
-cite_agent/agent_backend_only.py,sha256=Rmi3cUCcTMSHRxZu6MK2rZwme5SCfilxqn0BsoIds_U,5375
-cite_agent/ascii_plotting.py,sha256=lk8BaECs6fmjtp4iH12G09-frlRehAN7HLhHt2crers,8570
-cite_agent/auth.py,sha256=CYBNv8r1_wfdhsx-YcWOiXCiKvPBymaMca6w7JV__FQ,9809
-cite_agent/backend_only_client.py,sha256=WqLF8x7aXTro2Q3ehqKMsdCg53s6fNk9Hy86bGxqmmw,2561
-cite_agent/cli.py,sha256=T2zSv2ET3IKh1_ENfFIkW292KWC1lt-yUSdYn51Urw4,18564
-cite_agent/cli_enhanced.py,sha256=EAaSw9qtiYRWUXF6_05T19GCXlz9cCSz6n41ASnXIPc,7407
-cite_agent/dashboard.py,sha256=VGV5XQU1PnqvTsxfKMcue3j2ri_nvm9Be6O5aVays_w,10502
-cite_agent/enhanced_ai_agent.py,sha256=Rmi3cUCcTMSHRxZu6MK2rZwme5SCfilxqn0BsoIds_U,5375
-cite_agent/rate_limiter.py,sha256=-0fXx8Tl4zVB4O28n9ojU2weRo-FBF1cJo9Z5jC2LxQ,10908
-cite_agent/setup_config.py,sha256=fUVBxJ0CK7e5cGG4jBDpHbB2seC1RqEteneq-WslAw8,17064
-cite_agent/telemetry.py,sha256=55kXdHvI24ZsEkbFtihcjIfJt2oiSXcEpLzTxQ3KCdQ,2916
-cite_agent/ui.py,sha256=r1OAeY3NSeqhAjJYmEBH9CaennBuibFAz1Mur6YF80E,6134
-cite_agent/updater.py,sha256=kL2GYL1AKoZ9JoTXxFT5_AkvYvObcCrO2sIVyBw9JgU,7057
-cite_agent/web_search.py,sha256=j-BRhT8EBC6BEPgACQPeVwB1SVGKDz4XLM7sowacvSc,6587
-cite_agent-1.0.4.dist-info/licenses/LICENSE,sha256=XJkyO4IymhSUniN1ENY6lLrL2729gn_rbRlFK6_Hi9M,1074
-cite_agent-1.0.4.dist-info/METADATA,sha256=QklMsGqAD-NCLM1S4LHzCUSaeH58ld-IXG06NO24Bsw,6856
-cite_agent-1.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-cite_agent-1.0.4.dist-info/entry_points.txt,sha256=bJ0u28nFIxQKH1PWQ2ak4PV-FAjhoxTC7YADEdDenFw,83
-cite_agent-1.0.4.dist-info/top_level.txt,sha256=NNfD8pxDZzBK8tjDIpCs2BW9Va-OQ5qUFbEx0SgmyIE,11
-cite_agent-1.0.4.dist-info/RECORD,,