mem-llm 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

mem_llm/__init__.py

@@ -19,11 +19,12 @@ try:
     from .memory_db import SQLMemoryManager
     from .prompt_templates import prompt_manager
     from .config_manager import get_config
-    __all_pro__ = ["SQLMemoryManager", "prompt_manager", "get_config"]
+    from .config_from_docs import create_config_from_document
+    __all_pro__ = ["SQLMemoryManager", "prompt_manager", "get_config", "create_config_from_document"]
 except ImportError:
     __all_pro__ = []
 
-__version__ = "1.0.1"
+__version__ = "1.0.3"
 __author__ = "C. Emre Karataş"
 
 __all__ = [

mem_llm/config_from_docs.py

@@ -0,0 +1,180 @@
+"""
+Config Generator from Documents (PDF, DOCX, TXT)
+Automatically creates config.yaml from business documents
+"""
+
+import os
+from typing import Optional, Dict, Any
+import yaml
+
+
+def extract_text_from_file(file_path: str) -> str:
+    """
+    Extract text from PDF, DOCX, or TXT files
+    
+    Args:
+        file_path: Path to document
+        
+    Returns:
+        Extracted text
+    """
+    file_ext = os.path.splitext(file_path)[1].lower()
+    
+    if file_ext == '.txt':
+        with open(file_path, 'r', encoding='utf-8') as f:
+            return f.read()
+    
+    elif file_ext == '.pdf':
+        try:
+            import PyPDF2
+            text = []
+            with open(file_path, 'rb') as f:
+                reader = PyPDF2.PdfReader(f)
+                for page in reader.pages:
+                    text.append(page.extract_text())
+            return '\n'.join(text)
+        except ImportError:
+            return "⚠️ PyPDF2 not installed. Run: pip install PyPDF2"
+    
+    elif file_ext in ['.docx', '.doc']:
+        try:
+            import docx
+            doc = docx.Document(file_path)
+            text = []
+            for paragraph in doc.paragraphs:
+                text.append(paragraph.text)
+            return '\n'.join(text)
+        except ImportError:
+            return "⚠️ python-docx not installed. Run: pip install python-docx"
+    
+    else:
+        return f"⚠️ Unsupported file format: {file_ext}"
+
+
+def generate_config_from_text(text: str, company_name: Optional[str] = None) -> Dict[str, Any]:
+    """
+    Generate config.yaml structure from text
+    
+    Args:
+        text: Extracted text from document
+        company_name: Company name (optional)
+        
+    Returns:
+        Config dictionary
+    """
+    # Simple config template
+    config = {
+        "usage_mode": "business",  # or "personal"
+        
+        "llm": {
+            "model": "granite4:tiny-h",
+            "temperature": 0.3,
+            "max_tokens": 300,
+            "ollama_url": "http://localhost:11434"
+        },
+        
+        "memory": {
+            "use_sql": True,
+            "db_path": "memories.db",
+            "json_dir": "memories"
+        },
+        
+        "response": {
+            "use_knowledge_base": True,
+            "recent_conversations_limit": 5
+        },
+        
+        "business": {
+            "company_name": company_name or "Your Company",
+            "industry": "Technology",
+            "founded_year": "2024"
+        },
+        
+        "knowledge_base": {
+            "auto_load": True,
+            "search_limit": 5
+        },
+        
+        "logging": {
+            "level": "INFO",
+            "file": "mem_agent.log"
+        }
+    }
+    
+    # Try to extract company name from text if not provided
+    if not company_name:
+        lines = text.split('\n')[:10]  # First 10 lines
+        for line in lines:
+            if any(keyword in line.lower() for keyword in ['company', 'corp', 'inc', 'ltd']):
+                config["business"]["company_name"] = line.strip()[:50]
+                break
+    
+    return config
+
+
+def create_config_from_document(
+    doc_path: str,
+    output_path: str = "config.yaml",
+    company_name: Optional[str] = None
+) -> str:
+    """
+    Create config.yaml from a business document
+    
+    Args:
+        doc_path: Path to PDF/DOCX/TXT document
+        output_path: Output config.yaml path
+        company_name: Company name (optional)
+        
+    Returns:
+        Success message
+    """
+    if not os.path.exists(doc_path):
+        return f"❌ File not found: {doc_path}"
+    
+    # Extract text
+    print(f"📄 Reading document: {doc_path}")
+    text = extract_text_from_file(doc_path)
+    
+    if text.startswith("⚠️"):
+        return text  # Error message
+    
+    print(f"✅ Extracted {len(text)} characters")
+    
+    # Generate config
+    config = generate_config_from_text(text, company_name)
+    
+    # Save to YAML
+    with open(output_path, 'w', encoding='utf-8') as f:
+        yaml.dump(config, f, default_flow_style=False, allow_unicode=True)
+    
+    print(f"✅ Config created: {output_path}")
+    print(f"📌 Company: {config['business']['company_name']}")
+    
+    return f"✅ Config successfully created at {output_path}"
+
+
+# Simple CLI
+if __name__ == "__main__":
+    import sys
+    
+    if len(sys.argv) < 2:
+        print("""
+🔧 Config Generator from Documents
+
+Usage:
+    python -m mem_llm.config_from_docs <document_path> [output_path] [company_name]
+
+Examples:
+    python -m mem_llm.config_from_docs company_info.pdf
+    python -m mem_llm.config_from_docs business.docx my_config.yaml "Acme Corp"
+    python -m mem_llm.config_from_docs info.txt
+        """)
+        sys.exit(1)
+    
+    doc_path = sys.argv[1]
+    output_path = sys.argv[2] if len(sys.argv) > 2 else "config.yaml"
+    company_name = sys.argv[3] if len(sys.argv) > 3 else None
+    
+    result = create_config_from_document(doc_path, output_path, company_name)
+    print(result)
+

mem_llm/llm_client.py

@@ -107,7 +107,11 @@ class OllamaClient:
             "stream": False,
             "options": {
                 "temperature": temperature,
-                "num_predict": max_tokens
+                "num_predict": max_tokens,
+                "num_ctx": 2048,  # Context window
+                "top_k": 40,  # Limit vocab
+                "top_p": 0.9,  # Nucleus sampling
+                "stop": ["\n\n\n", "---"]  # Stop sequences
             }
         }
         

mem_llm/mem_agent.py

@@ -111,6 +111,8 @@ class MemAgent:
             self.logger.info(f"JSON memory system active: {json_dir}")
 
         # LLM client
+        self.model = model  # Store model name
+        self.use_sql = use_sql  # Store SQL usage flag
         self.llm = OllamaClient(model, ollama_url)
         self.logger.info(f"LLM client ready: {model}")
 
@@ -217,7 +219,23 @@ class MemAgent:
                 self.logger.info(f"Prompt template loaded: {template_name} (Mode: {self.usage_mode})")
             except Exception as e:
                 self.logger.error(f"Prompt template loading error: {e}")
-                self.current_system_prompt = f"You are a helpful assistant in {self.usage_mode} mode."
+                # Simple, short and effective default prompt
+                self.current_system_prompt = """You are a concise AI assistant. Be EXTREMELY brief.
+
+RULES (MANDATORY):
+1. MAX 1-2 SHORT sentences per response
+2. When user shares info: Just say "Got it!" or "Noted!"
+3. Answer questions: ONE sentence, direct
+4. NO lists, NO explanations, NO examples
+5. Use conversation history when relevant
+
+EXAMPLES:
+User: "My name is Alice" → You: "Nice to meet you, Alice!"
+User: "My favorite food is pizza" → You: "Got it!"
+User: "What's my name?" → You: "Your name is Alice."
+User: "Tell me about Python" → You: "Python is a versatile programming language for web, data science, and AI."
+
+BE BRIEF OR USER WILL LEAVE!"""
 
     def check_setup(self) -> Dict[str, Any]:
         """Check system setup"""
@@ -330,7 +348,8 @@ class MemAgent:
                 recent_limit = self.config.get("response.recent_conversations_limit", 5) if hasattr(self, 'config') and self.config else 5
                 recent_convs = self.memory.get_recent_conversations(user_id, recent_limit)
 
-                for conv in reversed(recent_convs):
+                # Add conversations in chronological order (oldest first)
+                for conv in recent_convs:
                     messages.append({"role": "user", "content": conv.get('user_message', '')})
                     messages.append({"role": "assistant", "content": conv.get('bot_response', '')})
         except Exception as e:
@@ -350,8 +369,8 @@ class MemAgent:
         try:
             response = self.llm.chat(
                 messages=messages,
-                temperature=self.config.get("llm.temperature", 0.7) if hasattr(self, 'config') and self.config else 0.7,
-                max_tokens=self.config.get("llm.max_tokens", 500) if hasattr(self, 'config') and self.config else 500
+                temperature=self.config.get("llm.temperature", 0.2) if hasattr(self, 'config') and self.config else 0.2,  # Very focused
+                max_tokens=self.config.get("llm.max_tokens", 150) if hasattr(self, 'config') and self.config else 150  # Max 2-3 sentences
             )
         except Exception as e:
             self.logger.error(f"LLM response error: {e}")
@@ -366,11 +385,78 @@ class MemAgent:
                     bot_response=response,
                     metadata=metadata
                 )
+                
+                # Extract and save user info to profile
+                self._update_user_profile(user_id, message, response)
         except Exception as e:
             self.logger.error(f"Interaction saving error: {e}")
 
         return response
+    
+    def _update_user_profile(self, user_id: str, message: str, response: str):
+        """Extract user info from conversation and update profile"""
+        if not hasattr(self.memory, 'update_profile'):
+            return
+        
+        msg_lower = message.lower()
+        updates = {}
+        
+        # Extract name
+        if "my name is" in msg_lower or "i am" in msg_lower or "i'm" in msg_lower:
+            # Simple name extraction
+            for phrase in ["my name is ", "i am ", "i'm "]:
+                if phrase in msg_lower:
+                    name_part = message[msg_lower.index(phrase) + len(phrase):].strip()
+                    name = name_part.split()[0] if name_part else None
+                    if name and len(name) > 1:
+                        updates['name'] = name.strip('.,!?')
+                        break
+        
+        # Extract favorite food
+        if "favorite food" in msg_lower or "favourite food" in msg_lower:
+            if "is" in msg_lower:
+                food = msg_lower.split("is")[-1].strip().strip('.,!?')
+                if food and len(food) < 50:
+                    updates['favorite_food'] = food
+        
+        # Extract location
+        if "i live in" in msg_lower or "i'm from" in msg_lower or "from" in msg_lower:
+            for phrase in ["i live in ", "i'm from ", "from "]:
+                if phrase in msg_lower:
+                    loc = message[msg_lower.index(phrase) + len(phrase):].strip()
+                    location = loc.split()[0] if loc else None
+                    if location and len(location) > 2:
+                        updates['location'] = location.strip('.,!?')
+                        break
+        
+        # Save updates
+        if updates:
+            try:
+                self.memory.update_profile(user_id, updates)
+                self.logger.debug(f"Profile updated for {user_id}: {updates}")
+            except:
+                pass
 
+    def get_user_profile(self, user_id: Optional[str] = None) -> Dict:
+        """
+        Get user's profile info
+        
+        Args:
+            user_id: User ID (uses current_user if not specified)
+            
+        Returns:
+            User profile dictionary
+        """
+        uid = user_id or self.current_user
+        if not uid:
+            return {}
+        
+        try:
+            memory_data = self.memory.load_memory(uid)
+            return memory_data.get('profile', {})
+        except:
+            return {}
+    
     def add_knowledge(self, category: str, question: str, answer: str,
                      keywords: Optional[List[str]] = None, priority: int = 0) -> int:
         """Add new record to knowledge base"""

mem_llm/memory_db.py

@@ -164,6 +164,11 @@ class SQLMemoryManager:
         self.conn.commit()
         return interaction_id
     
+    # Alias for compatibility
+    def add_conversation(self, user_id: str, user_message: str, bot_response: str, metadata: Optional[Dict] = None) -> int:
+        """Alias for add_interaction"""
+        return self.add_interaction(user_id, user_message, bot_response, metadata)
+    
     def get_recent_conversations(self, user_id: str, limit: int = 10) -> List[Dict]:
         """
         Kullanıcının son konuşmalarını getirir

mem_llm/memory_manager.py

@@ -101,6 +101,11 @@ class MemoryManager:
         self.conversations[user_id].append(interaction)
         self.save_memory(user_id)
     
+    # Alias for compatibility
+    def add_conversation(self, user_id: str, user_message: str, bot_response: str, metadata: Optional[Dict] = None) -> None:
+        """Alias for add_interaction"""
+        return self.add_interaction(user_id, user_message, bot_response, metadata)
+    
     def update_profile(self, user_id: str, updates: Dict) -> None:
         """
         Update user profile

{mem_llm-1.0.1.dist-info → mem_llm-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mem-llm
-Version: 1.0.1
+Version: 1.0.3
 Summary: Memory-enabled AI assistant with local LLM support
 Home-page: https://github.com/emredeveloper/Mem-LLM
 Author: C. Emre Karataş

mem_llm-1.0.3.dist-info/RECORD

@@ -0,0 +1,15 @@
+mem_llm/__init__.py,sha256=C_7xanudK0nG04K2RkCf-6M5AwphjnkIqVfTmoCKkfc,920
+mem_llm/config.yaml.example,sha256=lgmfaU5pxnIm4zYxwgCcgLSohNx1Jw6oh3Qk0Xoe2DE,917
+mem_llm/config_from_docs.py,sha256=YFhq1SWyK63C-TNMS73ncNHg8sJ-XGOf2idWVCjxFco,4974
+mem_llm/config_manager.py,sha256=8PIHs21jZWlI-eG9DgekjOvNxU3-U4xH7SbT8Gr-Z6M,7075
+mem_llm/knowledge_loader.py,sha256=oSNhfYYcx7DlZLVogxnbSwaIydq_Q3__RDJFeZR2XVw,2699
+mem_llm/llm_client.py,sha256=XW-LALYV4C-Nj2R8XYT2iT2YnTeH6_tUIldMKooY2uY,5461
+mem_llm/mem_agent.py,sha256=R37BDniAdoCf7kc9i9zbPdzarSlftFlRoqGuYpyGp9Y,23352
+mem_llm/memory_db.py,sha256=KyNIcChYihSavd2ot5KMBlVB9lq8rexoBQ0lA5bCJNI,12611
+mem_llm/memory_manager.py,sha256=iXnf5YEJXmQ75jgJ2LEx9zCHxIpZTcLtHlp2eWgFjRg,8335
+mem_llm/memory_tools.py,sha256=ARANFqu_bmL56SlV1RzTjfQsJj-Qe2QvqY0pF92hDxU,8678
+mem_llm/prompt_templates.py,sha256=tCiQJw3QQKIaH8NsxEKOIaIVxw4XT43PwdmyfCINzzM,6536
+mem_llm-1.0.3.dist-info/METADATA,sha256=NPZlthtPNoMaD4hi1uTCMlSEJ_cSKlbCRW5D3b_BHn4,9347
+mem_llm-1.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mem_llm-1.0.3.dist-info/top_level.txt,sha256=_fU1ML-0JwkaxWdhqpwtmTNaJEOvDMQeJdA8d5WqDn8,8
+mem_llm-1.0.3.dist-info/RECORD,,

mem_llm-1.0.1.dist-info/RECORD

@@ -1,14 +0,0 @@
-mem_llm/__init__.py,sha256=Q_Ryd4aayMcfh9dNelr7TNupMc_AWWIuFmZUwO_Rn4I,827
-mem_llm/config.yaml.example,sha256=lgmfaU5pxnIm4zYxwgCcgLSohNx1Jw6oh3Qk0Xoe2DE,917
-mem_llm/config_manager.py,sha256=8PIHs21jZWlI-eG9DgekjOvNxU3-U4xH7SbT8Gr-Z6M,7075
-mem_llm/knowledge_loader.py,sha256=oSNhfYYcx7DlZLVogxnbSwaIydq_Q3__RDJFeZR2XVw,2699
-mem_llm/llm_client.py,sha256=tLNulVEV_tWdktvcQUokdhd0gTkIISUHipglRt17IWk,5255
-mem_llm/mem_agent.py,sha256=AMw8X5cFdHoyphyHf9B4eBXDFGTLEv9nkDBXnO_fGL4,19907
-mem_llm/memory_db.py,sha256=OGWTIHBHh1qETGvmrlZWfmv9szSaFuSCzJGMZg6HBww,12329
-mem_llm/memory_manager.py,sha256=-JM0Qb5dYm1Rj4jd3FQfDpZSaya-ly9rcgEjyvnyDzk,8052
-mem_llm/memory_tools.py,sha256=ARANFqu_bmL56SlV1RzTjfQsJj-Qe2QvqY0pF92hDxU,8678
-mem_llm/prompt_templates.py,sha256=tCiQJw3QQKIaH8NsxEKOIaIVxw4XT43PwdmyfCINzzM,6536
-mem_llm-1.0.1.dist-info/METADATA,sha256=QbeIJOQT5bhn7suA7Eo8AUe7RfDhK0jV8UZqxeF6o6g,9347
-mem_llm-1.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mem_llm-1.0.1.dist-info/top_level.txt,sha256=_fU1ML-0JwkaxWdhqpwtmTNaJEOvDMQeJdA8d5WqDn8,8
-mem_llm-1.0.1.dist-info/RECORD,,