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

mem_llm/__init__.py

@@ -24,7 +24,7 @@ try:
 except ImportError:
     __all_pro__ = []
 
-__version__ = "1.0.2"
+__version__ = "1.0.4"
 __author__ = "C. Emre Karataş"
 
 __all__ = [

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

@@ -220,11 +220,30 @@ class MemAgent:
             except Exception as e:
                 self.logger.error(f"Prompt template loading error: {e}")
                 # 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"""
+                self.current_system_prompt = """You are a helpful AI assistant with access to a knowledge base.
+
+CRITICAL RULES (FOLLOW EXACTLY):
+1. If KNOWLEDGE BASE information is provided below, USE IT FIRST - it's the correct answer!
+2. Knowledge base answers are marked with "📚 RELEVANT KNOWLEDGE BASE"
+3. Keep responses SHORT (1-3 sentences maximum)
+4. When user shares personal info: Just acknowledge briefly ("Got it!" or "Noted!")
+5. Answer from knowledge base EXACTLY as written, don't make up information
+6. If knowledge base has no info, use conversation history or say "I don't have that information"
+
+RESPONSE PRIORITY:
+1st Priority: Knowledge Base (if available) ← USE THIS!
+2nd Priority: Conversation History
+3rd Priority: General knowledge (be brief)
+
+EXAMPLES:
+User: "What's the shipping cost?"
+Knowledge Base: "Shipping is free over $150"
+You: "Shipping is free for orders over $150!"
+
+User: "My name is Alice" → You: "Nice to meet you, Alice!"
+User: "What's my name?" → You: "Your name is Alice."
+
+REMEMBER: Knowledge base = truth. Always use it when provided!"""
 
     def check_setup(self) -> Dict[str, Any]:
         """Check system setup"""
@@ -311,18 +330,24 @@ class MemAgent:
 
         # Knowledge base search (if using SQL)
         kb_context = ""
-        if ADVANCED_AVAILABLE and isinstance(self.memory, SQLMemoryManager) and hasattr(self, 'config') and self.config:
-            if self.config.get("response.use_knowledge_base", True):
+        if ADVANCED_AVAILABLE and isinstance(self.memory, SQLMemoryManager):
+            # Check config only if it exists, otherwise always use KB
+            use_kb = True
+            kb_limit = 5
+            
+            if hasattr(self, 'config') and self.config:
+                use_kb = self.config.get("response.use_knowledge_base", True)
+                kb_limit = self.config.get("knowledge_base.search_limit", 5)
+            
+            if use_kb:
                 try:
-                    kb_results = self.memory.search_knowledge(
-                        query=message,
-                        limit=self.config.get("knowledge_base.search_limit", 5)
-                    )
+                    kb_results = self.memory.search_knowledge(query=message, limit=kb_limit)
 
                     if kb_results:
-                        kb_context = "\n\nRelevant Information:\n"
+                        kb_context = "\n\n📚 RELEVANT KNOWLEDGE BASE:\n"
                         for i, result in enumerate(kb_results, 1):
-                            kb_context += f"{i}. S: {result['question']}\n   C: {result['answer']}\n"
+                            kb_context += f"{i}. Q: {result['question']}\n   A: {result['answer']}\n"
+                        kb_context += "\n⚠️ USE THIS INFORMATION TO ANSWER! Be brief but accurate.\n"
                 except Exception as e:
                     self.logger.error(f"Knowledge base search error: {e}")
 
@@ -344,22 +369,20 @@ class MemAgent:
         except Exception as e:
             self.logger.error(f"Memory history loading error: {e}")
 
-        # Add knowledge base context
+        # Add current message WITH knowledge base context (if available)
+        final_message = message
         if kb_context:
-            messages.append({
-                "role": "system",
-                "content": f"You can use this information when answering the user's question:{kb_context}"
-            })
-
-        # Add current message
-        messages.append({"role": "user", "content": message})
+            # Inject KB directly into user message for maximum visibility
+            final_message = f"{kb_context}\n\nUser Question: {message}"
+        
+        messages.append({"role": "user", "content": final_message})
 
         # Get response from LLM
         try:
             response = self.llm.chat(
                 messages=messages,
-                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
+                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}")
@@ -374,11 +397,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-1.0.4.dist-info/METADATA

@@ -0,0 +1,304 @@
+Metadata-Version: 2.4
+Name: mem-llm
+Version: 1.0.4
+Summary: Memory-enabled AI assistant with local LLM support
+Home-page: https://github.com/emredeveloper/Mem-LLM
+Author: C. Emre Karataş
+Author-email: karatasqemre@gmail.com
+Project-URL: Bug Reports, https://github.com/emredeveloper/Mem-LLM/issues
+Project-URL: Source, https://github.com/emredeveloper/Mem-LLM
+Keywords: llm ai memory agent chatbot ollama local
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+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.8
+Description-Content-Type: text/markdown
+Requires-Dist: requests>=2.31.0
+Requires-Dist: pyyaml>=6.0.1
+Provides-Extra: dev
+Requires-Dist: pytest>=7.4.0; extra == "dev"
+Requires-Dist: black>=23.7.0; extra == "dev"
+Requires-Dist: flake8>=6.1.0; extra == "dev"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# 🧠 mem-llm
+
+**Memory-enabled AI assistant that remembers conversations using local LLMs**
+
+[![Python](https://img.shields.io/badge/Python-3.8%2B-blue.svg)](https://www.python.org/downloads/)
+[![PyPI](https://img.shields.io/pypi/v/mem-llm.svg)](https://pypi.org/project/mem-llm/)
+[![License](https://img.shields.io/badge/License-MIT-green.svg)](LICENSE)
+
+---
+
+## 🎯 What is it?
+
+A lightweight Python library that adds **persistent memory** to local LLM chatbots. Each user gets their own conversation history that the AI remembers across sessions.
+
+**Perfect for:**
+- 💬 Customer service chatbots
+- 🤖 Personal AI assistants  
+- 📝 Context-aware applications
+- 🏢 Business automation
+
+---
+
+## ⚡ Quick Start
+
+### 1. Install
+
+```bash
+pip install mem-llm
+```
+
+### 2. Setup Ollama (one-time)
+
+```bash
+# Install: https://ollama.ai/download
+ollama serve
+
+# Download model (only 2.5GB)
+ollama pull granite4:tiny-h
+```
+
+### 3. Use
+
+```python
+from mem_llm import MemAgent
+
+# Create agent (one line!)
+agent = MemAgent()
+
+# Set user
+agent.set_user("john")
+
+# Chat - it remembers!
+agent.chat("My name is John")
+agent.chat("What's my name?")  # → "Your name is John"
+```
+
+---
+
+## 💡 Features
+
+| Feature | Description |
+|---------|-------------|
+| 🧠 **Memory** | Remembers each user's conversation history |
+| 👥 **Multi-user** | Separate memory for each user |
+| 🔒 **Privacy** | 100% local, no cloud/API needed |
+| ⚡ **Fast** | Lightweight SQLite/JSON storage |
+| 🎯 **Simple** | 3 lines of code to get started |
+
+---
+
+## 📖 Usage Examples
+
+### Basic Chat
+
+```python
+from mem_llm import MemAgent
+
+agent = MemAgent()
+agent.set_user("alice")
+
+# First conversation
+agent.chat("I love pizza")
+
+# Later...
+agent.chat("What's my favorite food?")
+# → "Your favorite food is pizza"
+```
+
+### Customer Service Bot
+
+```python
+agent = MemAgent()
+
+# Customer 1
+agent.set_user("customer_001")
+agent.chat("My order #12345 is delayed")
+
+# Customer 2 (different memory!)
+agent.set_user("customer_002")
+agent.chat("I want to return item #67890")
+```
+
+### Check User Profile
+
+```python
+# Get automatically extracted user info
+profile = agent.get_user_profile()
+# {'name': 'Alice', 'favorite_food': 'pizza', 'location': 'NYC'}
+```
+
+---
+
+## 🔧 Configuration
+
+### JSON Memory (default - simple)
+
+```python
+agent = MemAgent(
+    model="granite4:tiny-h",
+    use_sql=False,  # Use JSON files
+    memory_dir="memories"
+)
+```
+
+### SQL Memory (advanced - faster)
+
+```python
+agent = MemAgent(
+    model="granite4:tiny-h",
+    use_sql=True,  # Use SQLite
+    memory_dir="memories.db"
+)
+```
+
+### Custom Settings
+
+```python
+agent = MemAgent(
+    model="llama2",  # Any Ollama model
+    ollama_url="http://localhost:11434"
+)
+```
+
+---
+
+## 📚 API Reference
+
+### MemAgent
+
+```python
+# Initialize
+agent = MemAgent(model="granite4:tiny-h", use_sql=False)
+
+# Set active user
+agent.set_user(user_id: str, name: Optional[str] = None)
+
+# Chat
+response = agent.chat(message: str, metadata: Optional[Dict] = None) -> str
+
+# Get profile
+profile = agent.get_user_profile(user_id: Optional[str] = None) -> Dict
+
+# System check
+status = agent.check_setup() -> Dict
+```
+
+---
+
+## 🎨 Advanced: PDF/DOCX Config
+
+Generate config from business documents:
+
+```python
+from mem_llm import create_config_from_document
+
+# Create config.yaml from PDF
+create_config_from_document(
+    doc_path="company_info.pdf",
+    output_path="config.yaml",
+    company_name="Acme Corp"
+)
+
+# Use config
+agent = MemAgent(config_file="config.yaml")
+```
+
+---
+
+## 🔥 Models
+
+Works with any [Ollama](https://ollama.ai/) model:
+
+| Model | Size | Speed | Quality |
+|-------|------|-------|---------|
+| `granite4:tiny-h` | 2.5GB | ⚡⚡⚡ | ⭐⭐ |
+| `llama2` | 4GB | ⚡⚡ | ⭐⭐⭐ |
+| `mistral` | 4GB | ⚡⚡ | ⭐⭐⭐⭐ |
+| `llama3` | 5GB | ⚡ | ⭐⭐⭐⭐⭐ |
+
+```bash
+ollama pull <model-name>
+```
+
+---
+
+## 📦 Requirements
+
+- Python 3.8+
+- Ollama (for LLM)
+- 4GB RAM minimum
+- 5GB disk space
+
+**Dependencies** (auto-installed):
+- `requests >= 2.31.0`
+- `pyyaml >= 6.0.1`
+
+---
+
+## 🐛 Troubleshooting
+
+### Ollama not running?
+
+```bash
+ollama serve
+```
+
+### Model not found?
+
+```bash
+ollama pull granite4:tiny-h
+```
+
+### Import error?
+
+```bash
+pip install mem-llm --upgrade
+```
+
+---
+
+## 📄 License
+
+MIT License - feel free to use in personal and commercial projects!
+
+---
+
+## 🔗 Links
+
+- **PyPI:** https://pypi.org/project/mem-llm/
+- **GitHub:** https://github.com/emredeveloper/Mem-LLM
+- **Ollama:** https://ollama.ai/
+
+---
+
+## 🌟 Star us on GitHub!
+
+If you find this useful, give us a ⭐ on [GitHub](https://github.com/emredeveloper/Mem-LLM)!
+
+---
+
+<div align="center">
+Made with ❤️ by <a href="https://github.com/emredeveloper">C. Emre Karataş</a>
+</div>

{mem_llm-1.0.2.dist-info → mem_llm-1.0.4.dist-info}/RECORD

@@ -1,15 +1,15 @@
-mem_llm/__init__.py,sha256=l9ynmAWNyC_CPZcb5q-pkJ_oVdJZpFN4hwVHRNqCkg8,920
+mem_llm/__init__.py,sha256=8aY47mzeaxhyimSQ3vU4FeiHTvyRnderMhX0qkQ5zKw,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/llm_client.py,sha256=XW-LALYV4C-Nj2R8XYT2iT2YnTeH6_tUIldMKooY2uY,5461
+mem_llm/mem_agent.py,sha256=FYiZyOIadURC0RdsY-aRdaZ53oxtOj0U5Crt2yWEbz8,24058
 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.4.dist-info/METADATA,sha256=H4njHFVhNXq0msFIpiQdoE7sCcGB3YrR_LP6vzZH_tc,6505
+mem_llm-1.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mem_llm-1.0.4.dist-info/top_level.txt,sha256=_fU1ML-0JwkaxWdhqpwtmTNaJEOvDMQeJdA8d5WqDn8,8
+mem_llm-1.0.4.dist-info/RECORD,,

mem_llm-1.0.2.dist-info/METADATA

@@ -1,382 +0,0 @@
-Metadata-Version: 2.4
-Name: mem-llm
-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ş
-Author-email: karatasqemre@gmail.com
-Project-URL: Bug Reports, https://github.com/emredeveloper/Mem-LLM/issues
-Project-URL: Source, https://github.com/emredeveloper/Mem-LLM
-Keywords: llm ai memory agent chatbot ollama local
-Classifier: Development Status :: 4 - Beta
-Classifier: Intended Audience :: Developers
-Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
-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.8
-Description-Content-Type: text/markdown
-Requires-Dist: requests>=2.31.0
-Requires-Dist: pyyaml>=6.0.1
-Provides-Extra: dev
-Requires-Dist: pytest>=7.4.0; extra == "dev"
-Requires-Dist: black>=23.7.0; extra == "dev"
-Requires-Dist: flake8>=6.1.0; extra == "dev"
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: keywords
-Dynamic: project-url
-Dynamic: provides-extra
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
-
-# 🧠 Mem-Agent: Memory-Enabled Mini Assistant
-
-<div align="center">
-
-[![Python](https://img.shields.io/badge/Python-3.8%2B-blue.svg)](https://www.python.org/downloads/)
-[![License](https://img.shields.io/badge/License-MIT-green.svg)](LICENSE)
-[![Ollama](https://img.shields.io/badge/Ollama-Compatible-orange.svg)](https://ollama.ai/)
-
-**A local AI assistant that remembers user interactions and responds with context awareness using a lightweight 4-billion parameter LLM.**
-
-[Quick Start](#-quick-start) • [Features](#-features) • [Documentation](#-documentation) • [Examples](#-usage-examples)
-
-</div>
-
----
-
-## 🎯 Why Mem-Agent?
-
-Most Large Language Models (LLMs) treat every conversation as "new" and don't remember past interactions. **Mem-Agent** uses a small locally-running model to:
-
-- ✅ **Remember user history** - Separate memory for each customer/user
-- ✅ **Context awareness** - Responds based on previous conversations
-- ✅ **Fully local** - No internet connection required
-- ✅ **Lightweight & fast** - Only 2.5 GB model size
-- ✅ **Easy integration** - Get started with 3 lines of code
-
-## 🚀 Quick Start
-
-### 1. Install Ollama
-
-```bash
-# Windows/Mac/Linux: https://ollama.ai/download
-curl https://ollama.ai/install.sh | sh
-
-# Start the service
-ollama serve
-```
-
-### 2. Download Model
-
-```bash
-ollama pull granite4:tiny-h
-```
-
-### 3. Use Mem-Agent
-
-```python
-from mem_llm import MemAgent
-
-# Create agent
-agent = MemAgent(model="granite4:tiny-h")
-
-# System check
-status = agent.check_setup()
-if status['status'] == 'ready':
-    print("✅ System ready!")
-else:
-    print("❌ Error:", status)
-
-# Set user
-agent.set_user("user123")
-
-# First conversation
-response = agent.chat("Hello, my name is Ali")
-print(response)
-
-# Second conversation - It remembers me!
-response = agent.chat("Do you remember my name?")
-print(response)
-```
-
-## 📚 Example Scripts
-
-### 1. Simple Test
-
-```bash
-python examples/example_simple.py
-```
-
-### 2. Customer Service Simulation
-
-```bash
-python examples/example_customer_service.py
-```
-
-## 🏗️ Project Structure
-
-```
-Memory LLM/
-├── memory_llm/              # Main package
-│   ├── __init__.py          # Package initialization
-│   ├── mem_agent.py         # Main assistant class
-│   ├── memory_manager.py    # Memory management
-│   ├── memory_db.py         # SQL database support
-│   ├── llm_client.py        # Ollama integration
-│   ├── memory_tools.py      # User tools
-│   ├── knowledge_loader.py  # Knowledge base loader
-│   ├── prompt_templates.py  # Prompt templates
-│   └── config_manager.py    # Configuration manager
-├── examples/                # Example scripts
-├── tests/                   # Test files
-├── setup.py                 # Installation script
-├── requirements.txt         # Dependencies
-└── README.md               # This file
-```
-
-## 🔧 API Usage
-
-### MemAgent Class
-
-```python
-from mem_llm import MemAgent
-
-agent = MemAgent(
-    model="granite4:tiny-h",           # Ollama model name
-    memory_dir="memories",             # Memory directory
-    ollama_url="http://localhost:11434" # Ollama API URL
-)
-```
-
-#### Basic Methods
-
-```python
-# Set user
-agent.set_user("user_id")
-
-# Chat
-response = agent.chat(
-    message="Hello",
-    user_id="optional_user_id",  # If set_user not used
-    metadata={"key": "value"}     # Additional information
-)
-
-# Get memory summary
-summary = agent.memory_manager.get_summary("user_id")
-
-# Search in history
-results = agent.search_user_history("keyword", "user_id")
-
-# Update profile
-agent.update_user_info({
-    "name": "Ali",
-    "preferences": {"language": "en"}
-})
-
-# Get statistics
-stats = agent.get_statistics()
-
-# Export memory
-json_data = agent.export_memory("user_id")
-
-# Clear memory (WARNING!)
-agent.clear_user_memory("user_id", confirm=True)
-```
-
-### MemoryManager Class
-
-```python
-from mem_llm import MemoryManager
-
-memory = MemoryManager(memory_dir="memories")
-
-# Load memory
-data = memory.load_memory("user_id")
-
-# Add interaction
-memory.add_interaction(
-    user_id="user_id",
-    user_message="Hello",
-    bot_response="Hello! How can I help you?",
-    metadata={"timestamp": "2025-01-13"}
-)
-
-# Get recent conversations
-recent = memory.get_recent_conversations("user_id", limit=5)
-
-# Search
-results = memory.search_memory("user_id", "order")
-```
-
-### OllamaClient Class
-
-```python
-from mem_llm import OllamaClient
-
-client = OllamaClient(model="granite4:tiny-h")
-
-# Simple generation
-response = client.generate("Hello world!")
-
-# Chat format
-response = client.chat([
-    {"role": "system", "content": "You are a helpful assistant"},
-    {"role": "user", "content": "Hello"}
-])
-
-# Connection check
-is_ready = client.check_connection()
-
-# Model list
-models = client.list_models()
-```
-
-## 💡 Usage Scenarios
-
-### 1. Customer Service Bot
-- Remembers customer history
-- Knows previous issues
-- Makes personalized recommendations
-
-### 2. Personal Assistant
-- Tracks daily activities
-- Learns preferences
-- Makes reminders
-
-### 3. Education Assistant
-- Tracks student progress
-- Adjusts difficulty level
-- Remembers past mistakes
-
-### 4. Support Ticket System
-- Stores ticket history
-- Finds related old tickets
-- Provides solution suggestions
-
-## 📊 Memory Format
-
-Memories are stored in JSON format:
-
-```json
-{
-  "conversations": [
-    {
-      "timestamp": "2025-01-13T10:30:00",
-      "user_message": "Hello",
-      "bot_response": "Hello! How can I help you?",
-      "metadata": {
-        "topic": "greeting"
-      }
-    }
-  ],
-  "profile": {
-    "user_id": "user123",
-    "first_seen": "2025-01-13T10:30:00",
-    "preferences": {},
-    "summary": {}
-  },
-  "last_updated": "2025-01-13T10:35:00"
-}
-```
-
-## 🔒 Privacy and Security
-
-- ✅ Works completely locally (no internet connection required)
-- ✅ Data stored on your computer
-- ✅ No data sent to third-party services
-- ✅ Memories in JSON format, easily deletable
-
-## 🛠️ Development
-
-### Test Mode
-
-```python
-# Simple chat without memory (for testing)
-response = agent.simple_chat("Test message")
-```
-
-### Using Your Own Model
-
-```python
-# Different Ollama model
-agent = MemAgent(model="llama2:7b")
-
-# Or another LLM API
-# Customize llm_client.py file
-```
-
-## 🐛 Troubleshooting
-
-### Ollama Connection Error
-
-```bash
-# Start Ollama service
-ollama serve
-
-# Port check
-netstat -an | findstr "11434"
-```
-
-### Model Not Found
-
-```bash
-# Check model list
-ollama list
-
-# Download model
-ollama pull granite4:tiny-h
-```
-
-### Memory Issues
-
-```python
-# Check memory directory
-import os
-os.path.exists("memories")
-
-# List memory files
-os.listdir("memories")
-```
-
-## 📈 Performance
-
-- **Model Size**: ~2.5 GB
-- **Response Time**: ~1-3 seconds (depends on CPU)
-- **Memory Usage**: ~4-6 GB RAM
-- **Disk Usage**: ~10-50 KB per user
-
-## 🤝 Contributing
-
-1. Fork the repository
-2. Create feature branch (`git checkout -b feature/amazing-feature`)
-3. Commit changes (`git commit -m 'feat: Add amazing feature'`)
-4. Push to branch (`git push origin feature/amazing-feature`)
-5. Open Pull Request
-
-## 📝 License
-
-MIT License - See LICENSE file for details.
-
-## 🙏 Acknowledgments
-
-- [Ollama](https://ollama.ai/) - Local LLM server
-- [Granite](https://www.ibm.com/granite) - IBM Granite models
-
-## 📞 Contact
-
-You can open an issue for your questions.
-
----
-
-**Note**: This project is for educational and research purposes. Please perform comprehensive testing before using in production environment.