mem-llm 1.0.0__py3-none-any.whl → 1.0.2__py3-none-any.whl

{memory_llm → 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.0"
+__version__ = "1.0.2"
 __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)
+

{memory_llm → 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,12 @@ 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 helpful AI assistant. Be concise and direct.
+- Give short, clear answers (2-3 sentences max)
+- Only use information from conversation history
+- If you don't know something, say so
+- Don't make assumptions or hallucinate"""
 
     def check_setup(self) -> Dict[str, Any]:
         """Check system setup"""
@@ -330,7 +337,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 +358,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.3) if hasattr(self, 'config') and self.config else 0.3,  # Lower = more focused
+                max_tokens=self.config.get("llm.max_tokens", 300) if hasattr(self, 'config') and self.config else 300  # Shorter responses
             )
         except Exception as e:
             self.logger.error(f"LLM response error: {e}")

{memory_llm → 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

{memory_llm → 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.0.dist-info → mem_llm-1.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mem-llm
-Version: 1.0.0
+Version: 1.0.2
 Summary: Memory-enabled AI assistant with local LLM support
 Home-page: https://github.com/emredeveloper/Mem-LLM
 Author: C. Emre Karataş
@@ -86,7 +86,7 @@ ollama pull granite4:tiny-h
 ### 3. Use Mem-Agent
 
 ```python
-from memory_llm import MemAgent
+from mem_llm import MemAgent
 
 # Create agent
 agent = MemAgent(model="granite4:tiny-h")
@@ -150,7 +150,7 @@ Memory LLM/
 ### MemAgent Class
 
 ```python
-from memory_llm import MemAgent
+from mem_llm import MemAgent
 
 agent = MemAgent(
     model="granite4:tiny-h",           # Ollama model name
@@ -197,7 +197,7 @@ agent.clear_user_memory("user_id", confirm=True)
 ### MemoryManager Class
 
 ```python
-from memory_llm import MemoryManager
+from mem_llm import MemoryManager
 
 memory = MemoryManager(memory_dir="memories")
 
@@ -222,7 +222,7 @@ results = memory.search_memory("user_id", "order")
 ### OllamaClient Class
 
 ```python
-from memory_llm import OllamaClient
+from mem_llm import OllamaClient
 
 client = OllamaClient(model="granite4:tiny-h")
 

mem_llm-1.0.2.dist-info/RECORD

@@ -0,0 +1,15 @@
+mem_llm/__init__.py,sha256=l9ynmAWNyC_CPZcb5q-pkJ_oVdJZpFN4hwVHRNqCkg8,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=tLNulVEV_tWdktvcQUokdhd0gTkIISUHipglRt17IWk,5255
+mem_llm/mem_agent.py,sha256=BIEMHpbss4QPstS-aEoZwmKBBc_fg87tf8Jj7MTIV8g,20357
+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.2.dist-info/METADATA,sha256=yJxAbApli62T27XkKE0SIHxgD50PPbhXldmBXfdyfhE,9347
+mem_llm-1.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mem_llm-1.0.2.dist-info/top_level.txt,sha256=_fU1ML-0JwkaxWdhqpwtmTNaJEOvDMQeJdA8d5WqDn8,8
+mem_llm-1.0.2.dist-info/RECORD,,

mem_llm-1.0.2.dist-info/top_level.txt

@@ -0,0 +1 @@
+mem_llm

mem_llm-1.0.0.dist-info/RECORD

@@ -1,14 +0,0 @@
-memory_llm/__init__.py,sha256=74hTFnqEMUtTnTLUtZllFo-8NM-JghqZgPH9SDgQj0g,827
-memory_llm/config.yaml.example,sha256=lgmfaU5pxnIm4zYxwgCcgLSohNx1Jw6oh3Qk0Xoe2DE,917
-memory_llm/config_manager.py,sha256=8PIHs21jZWlI-eG9DgekjOvNxU3-U4xH7SbT8Gr-Z6M,7075
-memory_llm/knowledge_loader.py,sha256=oSNhfYYcx7DlZLVogxnbSwaIydq_Q3__RDJFeZR2XVw,2699
-memory_llm/llm_client.py,sha256=tLNulVEV_tWdktvcQUokdhd0gTkIISUHipglRt17IWk,5255
-memory_llm/mem_agent.py,sha256=AMw8X5cFdHoyphyHf9B4eBXDFGTLEv9nkDBXnO_fGL4,19907
-memory_llm/memory_db.py,sha256=OGWTIHBHh1qETGvmrlZWfmv9szSaFuSCzJGMZg6HBww,12329
-memory_llm/memory_manager.py,sha256=-JM0Qb5dYm1Rj4jd3FQfDpZSaya-ly9rcgEjyvnyDzk,8052
-memory_llm/memory_tools.py,sha256=ARANFqu_bmL56SlV1RzTjfQsJj-Qe2QvqY0pF92hDxU,8678
-memory_llm/prompt_templates.py,sha256=tCiQJw3QQKIaH8NsxEKOIaIVxw4XT43PwdmyfCINzzM,6536
-mem_llm-1.0.0.dist-info/METADATA,sha256=Pdiho_vUo-vCZgKde5WYCqfabFtXoubMJA97u_qLjaY,9359
-mem_llm-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mem_llm-1.0.0.dist-info/top_level.txt,sha256=7I8wePWMtiZ-viJGXLYAiHpxiwpwPbFhNn1cyufySok,11
-mem_llm-1.0.0.dist-info/RECORD,,

mem_llm-1.0.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-memory_llm