praisonaiagents 0.0.37__tar.gz → 0.0.39__tar.gz

{praisonaiagents-0.0.37 → praisonaiagents-0.0.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: praisonaiagents
-Version: 0.0.37
+Version: 0.0.39
 Summary: Praison AI agents for completing complex tasks with Self Reflection Agents
 Author: Mervin Praison
 Requires-Dist: pydantic

{praisonaiagents-0.0.37 → praisonaiagents-0.0.39}/praisonaiagents/knowledge/knowledge.py

@@ -4,6 +4,7 @@ import uuid
 import time
 from .chunking import Chunking
 from functools import cached_property
+from rich.progress import Progress, SpinnerColumn, TextColumn, BarColumn, TaskProgressColumn
 
 logger = logging.getLogger(__name__)
 
@@ -43,9 +44,28 @@ class CustomMemory:
         }]
 
 class Knowledge:
-    def __init__(self, config=None):
+    def __init__(self, config=None, verbose=None):
         self._config = config
+        self._verbose = verbose or 0
         os.environ['ANONYMIZED_TELEMETRY'] = 'False'  # Chromadb
+        
+        # Configure logging levels based on verbose setting
+        if not self._verbose:
+            # Suppress logs from all relevant dependencies
+            for logger_name in [
+                'mem0', 
+                'chromadb', 
+                'local_persistent_hnsw',
+                '_client',
+                'main'
+            ]:
+                logging.getLogger(logger_name).setLevel(logging.WARNING)
+            
+            # Disable OpenAI API request logging
+            logging.getLogger('openai').setLevel(logging.WARNING)
+            
+            # Set root logger to warning to catch any uncategorized logs
+            logging.getLogger().setLevel(logging.WARNING)
 
     @cached_property
     def _deps(self):
@@ -121,14 +141,17 @@ class Knowledge:
             chunk_overlap=50
         )
 
+    def _log(self, message, level=2):
+        """Internal logging helper"""
+        if self._verbose and self._verbose >= level:
+            logger.info(message)
+
     def store(self, content, user_id=None, agent_id=None, run_id=None, metadata=None):
         """Store a memory."""
         try:
-            # Process content to match expected format
             if isinstance(content, str):
-                # Check if content is actually a file path
                 if any(content.lower().endswith(ext) for ext in ['.pdf', '.doc', '.docx', '.txt']):
-                    logger.info(f"Content appears to be a file path, processing file: {content}")
+                    self._log(f"Content appears to be a file path, processing file: {content}")
                     return self.add(content, user_id=user_id, agent_id=agent_id, run_id=run_id, metadata=metadata)
                 
                 content = content.strip()
@@ -136,7 +159,7 @@ class Knowledge:
                     return []
                 
             result = self.memory.add(content, user_id=user_id, agent_id=agent_id, run_id=run_id, metadata=metadata)
-            logger.info(f"Store operation result: {result}")
+            self._log(f"Store operation result: {result}")
             return result
         except Exception as e:
             logger.error(f"Error storing content: {str(e)}")
@@ -210,8 +233,7 @@ class Knowledge:
 
             # Check if input is URL
             if isinstance(input_path, str) and (input_path.startswith('http://') or input_path.startswith('https://')):
-                logger.info(f"Processing URL: {input_path}")
-                # TODO: Implement URL handling
+                self._log(f"Processing URL: {input_path}")
                 raise NotImplementedError("URL processing not yet implemented")
 
             # Check if input ends with any supported extension
@@ -220,7 +242,7 @@ class Knowledge:
                                   for ext in (exts if isinstance(exts, tuple) else (exts,)))
             
             if is_supported_file:
-                logger.info(f"Processing as file path: {input_path}")
+                self._log(f"Processing as file path: {input_path}")
                 if not os.path.exists(input_path):
                     logger.error(f"File not found: {input_path}")
                     raise FileNotFoundError(f"File not found: {input_path}")
@@ -253,14 +275,26 @@ class Knowledge:
                 # Treat as raw text content only if no file extension
                 memories = [self.normalize_content(input_path)]
 
-            # Store memories
+            # Create progress display
+            progress = Progress(
+                SpinnerColumn(),
+                TextColumn("[progress.description]{task.description}"),
+                BarColumn(),
+                TaskProgressColumn(),
+                transient=True
+            )
+
+            # Store memories with progress bar
             all_results = []
-            for memory in memories:
-                if memory:
-                    memory_result = self.store(memory, user_id=user_id, agent_id=agent_id, 
-                                             run_id=run_id, metadata=metadata)
-                    if memory_result:
-                        all_results.extend(memory_result.get('results', []))
+            with progress:
+                store_task = progress.add_task(f"Adding to Knowledge from {os.path.basename(input_path)}", total=len(memories))
+                for memory in memories:
+                    if memory:
+                        memory_result = self.store(memory, user_id=user_id, agent_id=agent_id, 
+                                                 run_id=run_id, metadata=metadata)
+                        if memory_result:
+                            all_results.extend(memory_result.get('results', []))
+                        progress.advance(store_task)
 
             return {'results': all_results, 'relations': []}
 

{praisonaiagents-0.0.37 → praisonaiagents-0.0.39}/praisonaiagents.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: praisonaiagents
-Version: 0.0.37
+Version: 0.0.39
 Summary: Praison AI agents for completing complex tasks with Self Reflection Agents
 Author: Mervin Praison
 Requires-Dist: pydantic

{praisonaiagents-0.0.37 → praisonaiagents-0.0.39}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "praisonaiagents"
-version = "0.0.37"
+version = "0.0.39"
 description = "Praison AI agents for completing complex tasks with Self Reflection Agents"
 authors = [
     { name="Mervin Praison" }