praisonaiagents 0.0.37__py3-none-any.whl → 0.0.39__py3-none-any.whl

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.dist-info → praisonaiagents-0.0.39.dist-info}/METADATA

@@ -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.dist-info → praisonaiagents-0.0.39.dist-info}/RECORD

@@ -7,7 +7,7 @@ praisonaiagents/agents/agents.py,sha256=M-nR53A7Qcz_pJ-gyNc4xgM13Nhof7oM-5hXWzr8
 praisonaiagents/agents/autoagents.py,sha256=bjC2O5oZmoJItJXIMPTWc2lsp_AJC9tMiTQOal2hwPA,13532
 praisonaiagents/knowledge/__init__.py,sha256=xL1Eh-a3xsHyIcU4foOWF-JdWYIYBALJH9bge0Ujuto,246
 praisonaiagents/knowledge/chunking.py,sha256=FzoNY0q8MkvG4gADqk4JcRhmH3lcEHbRdonDgitQa30,6624
-praisonaiagents/knowledge/knowledge.py,sha256=fnZMGm8AtlKlaIIbHT5xeQz-dfIPsii8koTdrTPl7H8,10985
+praisonaiagents/knowledge/knowledge.py,sha256=dJd8WuFV3w-zWaSAcT21PcXGRIZZc6YxBsJZDfmZNrk,12358
 praisonaiagents/memory/memory.py,sha256=ZxqSpOUxk9jeTKGW0ZiTifC0uZtym-EZILP3kuOOKkU,35626
 praisonaiagents/process/__init__.py,sha256=lkYbL7Hn5a0ldvJtkdH23vfIIZLIcanK-65C0MwaorY,52
 praisonaiagents/process/process.py,sha256=uSudOFI1ZlGM_nbT8qHD4iIP3y5Ygu8V-izLot2te70,26316
@@ -33,7 +33,7 @@ praisonaiagents/tools/wikipedia_tools.py,sha256=pGko-f33wqXgxJTv8db7TbizY5XnzBQR
 praisonaiagents/tools/xml_tools.py,sha256=iYTMBEk5l3L3ryQ1fkUnNVYK-Nnua2Kx2S0dxNMMs1A,17122
 praisonaiagents/tools/yaml_tools.py,sha256=uogAZrhXV9O7xvspAtcTfpKSQYL2nlOTvCQXN94-G9A,14215
 praisonaiagents/tools/yfinance_tools.py,sha256=s2PBj_1v7oQnOobo2fDbQBACEHl61ftG4beG6Z979ZE,8529
-praisonaiagents-0.0.37.dist-info/METADATA,sha256=PVUCq8U4iPXxXah6DsUdfak6fXW5HEmQUEOA229MbTk,664
-praisonaiagents-0.0.37.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-praisonaiagents-0.0.37.dist-info/top_level.txt,sha256=_HsRddrJ23iDx5TTqVUVvXG2HeHBL5voshncAMDGjtA,16
-praisonaiagents-0.0.37.dist-info/RECORD,,
+praisonaiagents-0.0.39.dist-info/METADATA,sha256=nfUgMS-EXpJn1Aee7bNdYcDq_Oxgkcbe-W3DrcCEWLw,664
+praisonaiagents-0.0.39.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+praisonaiagents-0.0.39.dist-info/top_level.txt,sha256=_HsRddrJ23iDx5TTqVUVvXG2HeHBL5voshncAMDGjtA,16
+praisonaiagents-0.0.39.dist-info/RECORD,,