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

mem_llm/__init__.py

@@ -24,7 +24,7 @@ try:
 except ImportError:
     __all_pro__ = []
 
-__version__ = "1.0.2"
+__version__ = "1.0.3"
 __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,22 @@ 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 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"""
@@ -358,8 +369,8 @@ class MemAgent:
         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 +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-1.0.2.dist-info → mem_llm-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mem-llm
-Version: 1.0.2
+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.2.dist-info → mem_llm-1.0.3.dist-info}/RECORD

@@ -1,15 +1,15 @@
-mem_llm/__init__.py,sha256=l9ynmAWNyC_CPZcb5q-pkJ_oVdJZpFN4hwVHRNqCkg8,920
+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=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=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.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.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,,