mcli-framework 7.6.0__py3-none-any.whl → 7.6.2__py3-none-any.whl

mcli/self/self_cmd.py

@@ -29,8 +29,8 @@ try:
 except ImportError:
     process = None
 
-from mcli.lib.logger.logger import get_logger
 from mcli.lib.custom_commands import get_command_manager
+from mcli.lib.logger.logger import get_logger
 
 logger = get_logger()
 
@@ -345,29 +345,35 @@ def collect_commands() -> List[Dict[str, Any]]:
 
                 try:
                     # Suppress Streamlit warnings and logging during module import
-                    import warnings
                     import logging
-                    import sys
                     import os
+                    import sys
+                    import warnings
                     from contextlib import redirect_stderr
                     from io import StringIO
-                    
+
                     # Suppress Python warnings
                     with warnings.catch_warnings():
                         warnings.filterwarnings("ignore", message=".*missing ScriptRunContext.*")
                         warnings.filterwarnings("ignore", message=".*No runtime found.*")
-                        warnings.filterwarnings("ignore", message=".*Session state does not function.*")
+                        warnings.filterwarnings(
+                            "ignore", message=".*Session state does not function.*"
+                        )
                         warnings.filterwarnings("ignore", message=".*to view this Streamlit app.*")
-                        
+
                         # Suppress Streamlit logger warnings
                         streamlit_logger = logging.getLogger("streamlit")
                         original_level = streamlit_logger.level
                         streamlit_logger.setLevel(logging.CRITICAL)
-                        
+
                         # Also suppress specific Streamlit sub-loggers
-                        logging.getLogger("streamlit.runtime.scriptrunner_utils.script_run_context").setLevel(logging.CRITICAL)
-                        logging.getLogger("streamlit.runtime.caching.cache_data_api").setLevel(logging.CRITICAL)
-                        
+                        logging.getLogger(
+                            "streamlit.runtime.scriptrunner_utils.script_run_context"
+                        ).setLevel(logging.CRITICAL)
+                        logging.getLogger("streamlit.runtime.caching.cache_data_api").setLevel(
+                            logging.CRITICAL
+                        )
+
                         # Redirect stderr to suppress Streamlit warnings
                         with redirect_stderr(StringIO()):
                             try:
@@ -417,40 +423,44 @@ def collect_commands() -> List[Dict[str, Any]]:
     return commands
 
 
-def open_editor_for_command(command_name: str, command_group: str, description: str) -> Optional[str]:
+def open_editor_for_command(
+    command_name: str, command_group: str, description: str
+) -> Optional[str]:
     """
     Open the user's default editor to allow them to write command logic.
-    
+
     Args:
         command_name: Name of the command
         command_group: Group for the command
         description: Description of the command
-        
+
     Returns:
         The Python code written by the user, or None if cancelled
     """
-    import tempfile
-    import subprocess
     import os
+    import subprocess
     import sys
+    import tempfile
     from pathlib import Path
-    
+
     # Get the user's default editor
-    editor = os.environ.get('EDITOR')
+    editor = os.environ.get("EDITOR")
     if not editor:
         # Try common editors in order of preference
-        for common_editor in ['vim', 'nano', 'code', 'subl', 'atom', 'emacs']:
-            if subprocess.run(['which', common_editor], capture_output=True).returncode == 0:
+        for common_editor in ["vim", "nano", "code", "subl", "atom", "emacs"]:
+            if subprocess.run(["which", common_editor], capture_output=True).returncode == 0:
                 editor = common_editor
                 break
-    
+
     if not editor:
-        click.echo("❌ No editor found. Please set the EDITOR environment variable or install vim/nano.")
+        click.echo(
+            "❌ No editor found. Please set the EDITOR environment variable or install vim/nano."
+        )
         return None
-    
+
     # Create a temporary file with the template
     template = get_command_template(command_name, command_group)
-    
+
     # Add helpful comments to the template
     enhanced_template = f'''"""
 {command_name} command for mcli.{command_group}.
@@ -491,64 +501,66 @@ logger = get_logger()
 #     logger.info(f"Executing {command_name} command with name: {{name}}")
 #     click.echo(f"Hello, {{name}}! This is the {command_name} command.")
 '''
-    
+
     # Create temporary file
-    with tempfile.NamedTemporaryFile(mode='w', suffix='.py', delete=False) as temp_file:
+    with tempfile.NamedTemporaryFile(mode="w", suffix=".py", delete=False) as temp_file:
         temp_file.write(enhanced_template)
         temp_file_path = temp_file.name
-    
+
     try:
         # Check if we're in an interactive environment
         if not sys.stdin.isatty() or not sys.stdout.isatty():
-            click.echo("❌ Editor requires an interactive terminal. Use --template flag for non-interactive mode.")
+            click.echo(
+                "❌ Editor requires an interactive terminal. Use --template flag for non-interactive mode."
+            )
             return None
-            
+
         # Open editor
         click.echo(f"📝 Opening {editor} to edit command logic...")
         click.echo("💡 Write your Python command logic and save the file to continue.")
         click.echo("💡 Press Ctrl+C to cancel command creation.")
-        
+
         # Run the editor
         result = subprocess.run([editor, temp_file_path], check=False)
-        
+
         if result.returncode != 0:
             click.echo("❌ Editor exited with error. Command creation cancelled.")
             return None
-        
+
         # Read the edited content
-        with open(temp_file_path, 'r') as f:
+        with open(temp_file_path, "r") as f:
             edited_code = f.read()
-        
+
         # Check if the file was actually edited (not just the template)
         if edited_code.strip() == enhanced_template.strip():
             click.echo("⚠️  No changes detected. Command creation cancelled.")
             return None
-        
+
         # Extract the actual command code (remove the instructions)
-        lines = edited_code.split('\n')
+        lines = edited_code.split("\n")
         code_lines = []
         in_code_section = False
-        
+
         for line in lines:
-            if line.strip().startswith('# Your command implementation goes here:'):
+            if line.strip().startswith("# Your command implementation goes here:"):
                 in_code_section = True
                 continue
             if in_code_section:
                 code_lines.append(line)
-        
+
         if not code_lines or not any(line.strip() for line in code_lines):
             # Fallback: use the entire file content
             code_lines = lines
-        
-        final_code = '\n'.join(code_lines).strip()
-        
+
+        final_code = "\n".join(code_lines).strip()
+
         if not final_code:
             click.echo("❌ No command code found. Command creation cancelled.")
             return None
-        
+
         click.echo("✅ Command code captured successfully!")
         return final_code
-        
+
     except KeyboardInterrupt:
         click.echo("\n❌ Command creation cancelled by user.")
         return None
@@ -611,7 +623,9 @@ def extract_workflow_commands(output):
 
                     if isinstance(cmd_obj, click.Group):
                         # For groups, create a template
-                        command_info["code"] = f'''"""
+                        command_info[
+                            "code"
+                        ] = f'''"""
 {cmd_name} workflow command.
 """
 import click
@@ -625,7 +639,9 @@ def app():
 '''
                     else:
                         # For regular commands, create a template
-                        command_info["code"] = f'''"""
+                        command_info[
+                            "code"
+                        ] = f'''"""
 {cmd_name} workflow command.
 """
 import click
@@ -650,9 +666,7 @@ def app():
             click.echo(
                 f"\n💡 These are templates. Import with: mcli self import-commands {output_file}"
             )
-            click.echo(
-                "   Then customize the code in ~/.mcli/commands/<command>.json"
-            )
+            click.echo("   Then customize the code in ~/.mcli/commands/<command>.json")
             return 0
         else:
             click.echo("⚠️  No workflow commands found to extract")
@@ -1229,24 +1243,20 @@ def update(check: bool, pre: bool, yes: bool, skip_ci_check: bool):
 
         console.print(f"[dim]{traceback.format_exc()}[/dim]")
 
-
-
         # Validate syntax
         try:
-            compile(new_code, '<string>', 'exec')
+            compile(new_code, "<string>", "exec")
         except SyntaxError as e:
             click.echo(f"❌ Syntax error in edited code: {e}", err=True)
-            should_save = Prompt.ask(
-                "Save anyway?", choices=["y", "n"], default="n"
-            )
+            should_save = Prompt.ask("Save anyway?", choices=["y", "n"], default="n")
             if should_save.lower() != "y":
                 return 1
 
         # Update the command
-        command_data['code'] = new_code
-        command_data['updated_at'] = datetime.now().isoformat()
+        command_data["code"] = new_code
+        command_data["updated_at"] = datetime.now().isoformat()
 
-        with open(command_file, 'w') as f:
+        with open(command_file, "w") as f:
             json.dump(command_data, f, indent=2)
 
         # Update lockfile
@@ -1268,6 +1278,7 @@ self_app.add_command(plugin)
 # Import and register new commands that have been moved to self
 try:
     from mcli.self.completion_cmd import completion
+
     self_app.add_command(completion, name="completion")
     logger.debug("Added completion command to self group")
 except ImportError as e:
@@ -1275,6 +1286,7 @@ except ImportError as e:
 
 try:
     from mcli.self.logs_cmd import logs_group
+
     self_app.add_command(logs_group, name="logs")
     logger.debug("Added logs command to self group")
 except ImportError as e:
@@ -1282,6 +1294,7 @@ except ImportError as e:
 
 try:
     from mcli.self.redis_cmd import redis_group
+
     self_app.add_command(redis_group, name="redis")
     logger.debug("Added redis command to self group")
 except ImportError as e:
@@ -1289,6 +1302,7 @@ except ImportError as e:
 
 try:
     from mcli.self.visual_cmd import visual
+
     self_app.add_command(visual, name="visual")
     logger.debug("Added visual command to self group")
 except ImportError as e:

mcli/workflow/daemon/daemon.py

@@ -20,6 +20,7 @@ from sklearn.metrics.pairwise import cosine_similarity
 try:
     from watchdog.events import FileSystemEventHandler
     from watchdog.observers import Observer
+
     HAS_WATCHDOG = True
 except ImportError:
     # Watchdog not available, file watching will be disabled
@@ -40,6 +41,7 @@ class CommandDatabase:
     """Stub database for backward compatibility.
     Commands are now stored as JSON files and loaded via the custom commands system.
     """
+
     def __init__(self, db_path: Optional[str] = None):
         logger.debug("CommandDatabase stub initialized - commands now managed via JSON files")
         pass

mcli/workflow/model_service/openai_adapter.py

@@ -10,7 +10,7 @@ import uuid
 from datetime import datetime
 from typing import Any, AsyncGenerator, Dict, List, Optional
 
-from fastapi import APIRouter, Depends, HTTPException, Header, Request, status
+from fastapi import APIRouter, Depends, Header, HTTPException, Request, status
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel, Field
 
@@ -116,7 +116,11 @@ class APIKeyManager:
     def list_keys(self) -> List[Dict[str, Any]]:
         """List all API keys (without showing the actual key)"""
         return [
-            {"name": info["name"], "created_at": info["created_at"], "usage_count": info["usage_count"]}
+            {
+                "name": info["name"],
+                "created_at": info["created_at"],
+                "usage_count": info["usage_count"],
+            }
             for info in self.valid_keys.values()
         ]
 

mcli/workflow/politician_trading/models.py

@@ -238,10 +238,14 @@ class PersonWithSignificantControl:
 
     # PSC details
     name: str = ""
-    kind: str = ""  # individual-person-with-significant-control, corporate-entity-person-with-significant-control, etc.
+    kind: str = (
+        ""  # individual-person-with-significant-control, corporate-entity-person-with-significant-control, etc.
+    )
 
     # Control nature
-    natures_of_control: List[str] = field(default_factory=list)  # ownership-of-shares-75-to-100-percent, etc.
+    natures_of_control: List[str] = field(
+        default_factory=list
+    )  # ownership-of-shares-75-to-100-percent, etc.
     notified_on: Optional[datetime] = None
 
     # Personal details (may be redacted)

mcli/workflow/politician_trading/scrapers_corporate_registry.py

@@ -15,9 +15,9 @@ politician trading patterns, conflicts of interest, and asset declarations.
 import logging
 import os
 import time
+from base64 import b64encode
 from datetime import datetime, timedelta
 from typing import Dict, List, Optional
-from base64 import b64encode
 
 import requests
 
@@ -54,10 +54,9 @@ class UKCompaniesHouseScraper:
         # API uses HTTP Basic Auth with API key as username, password empty
         auth_string = f"{self.api_key}:"
         auth_header = b64encode(auth_string.encode()).decode()
-        self.session.headers.update({
-            "Authorization": f"Basic {auth_header}",
-            "User-Agent": "PoliticianTradingTracker/1.0"
-        })
+        self.session.headers.update(
+            {"Authorization": f"Basic {auth_header}", "User-Agent": "PoliticianTradingTracker/1.0"}
+        )
 
     def search_companies(self, query: str, items_per_page: int = 20) -> List[Dict]:
         """
@@ -72,10 +71,7 @@ class UKCompaniesHouseScraper:
         """
         try:
             url = f"{self.BASE_URL}/search/companies"
-            params = {
-                "q": query,
-                "items_per_page": min(items_per_page, 100)
-            }
+            params = {"q": query, "items_per_page": min(items_per_page, 100)}
 
             response = self.session.get(url, params=params, timeout=30)
             response.raise_for_status()
@@ -198,10 +194,9 @@ class InfoFinanciereAPIScraper:
 
     def __init__(self):
         self.session = requests.Session()
-        self.session.headers.update({
-            "User-Agent": "PoliticianTradingTracker/1.0",
-            "Accept": "application/json"
-        })
+        self.session.headers.update(
+            {"User-Agent": "PoliticianTradingTracker/1.0", "Accept": "application/json"}
+        )
 
     def search_publications(
         self,
@@ -209,7 +204,7 @@ class InfoFinanciereAPIScraper:
         from_date: Optional[str] = None,
         to_date: Optional[str] = None,
         page: int = 1,
-        per_page: int = 20
+        per_page: int = 20,
     ) -> List[Dict]:
         """
         Search financial publications
@@ -226,10 +221,7 @@ class InfoFinanciereAPIScraper:
         """
         try:
             url = f"{self.BASE_URL}/publications"
-            params = {
-                "page": page,
-                "per_page": min(per_page, 100)
-            }
+            params = {"page": page, "per_page": min(per_page, 100)}
 
             if query:
                 params["q"] = query
@@ -298,17 +290,12 @@ class OpenCorporatesScraper:
         # API key is optional for free tier, but recommended
 
         self.session = requests.Session()
-        self.session.headers.update({
-            "User-Agent": "PoliticianTradingTracker/1.0",
-            "Accept": "application/json"
-        })
+        self.session.headers.update(
+            {"User-Agent": "PoliticianTradingTracker/1.0", "Accept": "application/json"}
+        )
 
     def search_companies(
-        self,
-        query: str,
-        jurisdiction_code: Optional[str] = None,
-        per_page: int = 30,
-        page: int = 1
+        self, query: str, jurisdiction_code: Optional[str] = None, per_page: int = 30, page: int = 1
     ) -> List[Dict]:
         """
         Search for companies across jurisdictions
@@ -324,11 +311,7 @@ class OpenCorporatesScraper:
         """
         try:
             url = f"{self.BASE_URL}/companies/search"
-            params = {
-                "q": query,
-                "per_page": min(per_page, 100),
-                "page": page
-            }
+            params = {"q": query, "per_page": min(per_page, 100), "page": page}
 
             if jurisdiction_code:
                 params["jurisdiction_code"] = jurisdiction_code
@@ -406,7 +389,9 @@ class OpenCorporatesScraper:
             results = data.get("results", {})
             officers = results.get("officers", [])
 
-            logger.info(f"Found {len(officers)} officers for company {jurisdiction_code}/{company_number}")
+            logger.info(
+                f"Found {len(officers)} officers for company {jurisdiction_code}/{company_number}"
+            )
             return officers
 
         except Exception as e:
@@ -432,10 +417,9 @@ class XBRLFilingsScraper:
 
     def __init__(self):
         self.session = requests.Session()
-        self.session.headers.update({
-            "User-Agent": "PoliticianTradingTracker/1.0",
-            "Accept": "application/vnd.api+json"
-        })
+        self.session.headers.update(
+            {"User-Agent": "PoliticianTradingTracker/1.0", "Accept": "application/vnd.api+json"}
+        )
 
     def get_filings(
         self,
@@ -443,7 +427,7 @@ class XBRLFilingsScraper:
         from_date: Optional[str] = None,
         to_date: Optional[str] = None,
         page_number: int = 1,
-        page_size: int = 100
+        page_size: int = 100,
     ) -> List[Dict]:
         """
         Get XBRL filings with filters
@@ -460,10 +444,7 @@ class XBRLFilingsScraper:
         """
         try:
             url = f"{self.BASE_URL}/filings"
-            params = {
-                "page[number]": page_number,
-                "page[size]": min(page_size, 500)
-            }
+            params = {"page[number]": page_number, "page[size]": min(page_size, 500)}
 
             # Add filters using JSON:API filter syntax
             if country:
@@ -487,10 +468,7 @@ class XBRLFilingsScraper:
             return []
 
     def get_entities(
-        self,
-        country: Optional[str] = None,
-        page_number: int = 1,
-        page_size: int = 100
+        self, country: Optional[str] = None, page_number: int = 1, page_size: int = 100
     ) -> List[Dict]:
         """
         Get filing entities (companies)
@@ -505,10 +483,7 @@ class XBRLFilingsScraper:
         """
         try:
             url = f"{self.BASE_URL}/entities"
-            params = {
-                "page[number]": page_number,
-                "page[size]": min(page_size, 500)
-            }
+            params = {"page[number]": page_number, "page[size]": min(page_size, 500)}
 
             if country:
                 params["filter[country]"] = country
@@ -553,10 +528,9 @@ class XBRLUSScraper:
             )
 
         self.session = requests.Session()
-        self.session.headers.update({
-            "User-Agent": "PoliticianTradingTracker/1.0",
-            "Accept": "application/json"
-        })
+        self.session.headers.update(
+            {"User-Agent": "PoliticianTradingTracker/1.0", "Accept": "application/json"}
+        )
 
     def search_companies(self, query: str, limit: int = 100) -> List[Dict]:
         """
@@ -571,11 +545,7 @@ class XBRLUSScraper:
         """
         try:
             url = f"{self.BASE_URL}/entity/search"
-            params = {
-                "name": query,
-                "limit": min(limit, 2000),
-                "client_id": self.api_key
-            }
+            params = {"name": query, "limit": min(limit, 2000), "client_id": self.api_key}
 
             response = self.session.get(url, params=params, timeout=30)
             response.raise_for_status()
@@ -595,7 +565,7 @@ class XBRLUSScraper:
         entity_id: int,
         filing_date_from: Optional[str] = None,
         filing_date_to: Optional[str] = None,
-        limit: int = 100
+        limit: int = 100,
     ) -> List[Dict]:
         """
         Get filings for an entity
@@ -611,11 +581,7 @@ class XBRLUSScraper:
         """
         try:
             url = f"{self.BASE_URL}/filing/search"
-            params = {
-                "entity.id": entity_id,
-                "limit": min(limit, 2000),
-                "client_id": self.api_key
-            }
+            params = {"entity.id": entity_id, "limit": min(limit, 2000), "client_id": self.api_key}
 
             if filing_date_from:
                 params["filing_date.from"] = filing_date_from
@@ -641,7 +607,7 @@ class XBRLUSScraper:
         entity_id: Optional[int] = None,
         period_end_from: Optional[str] = None,
         period_end_to: Optional[str] = None,
-        limit: int = 100
+        limit: int = 100,
     ) -> List[Dict]:
         """
         Get XBRL facts (financial data points)
@@ -661,7 +627,7 @@ class XBRLUSScraper:
             params = {
                 "concept.local-name": concept_name,
                 "limit": min(limit, 2000),
-                "client_id": self.api_key
+                "client_id": self.api_key,
             }
 
             if entity_id:
@@ -699,7 +665,7 @@ class CorporateRegistryFetcher:
         self,
         uk_companies_house_key: Optional[str] = None,
         opencorporates_key: Optional[str] = None,
-        xbrl_us_key: Optional[str] = None
+        xbrl_us_key: Optional[str] = None,
     ):
         """
         Initialize fetcher with optional API keys
@@ -767,16 +733,10 @@ class CorporateRegistryFetcher:
             f"{len(all_officers)} officers, {len(all_psc)} PSC records"
         )
 
-        return {
-            "companies": companies,
-            "officers": all_officers,
-            "psc": all_psc
-        }
+        return {"companies": companies, "officers": all_officers, "psc": all_psc}
 
     def fetch_french_disclosures(
-        self,
-        query: Optional[str] = None,
-        days_back: int = 30
+        self, query: Optional[str] = None, days_back: int = 30
     ) -> List[Dict]:
         """
         Fetch French financial disclosures
@@ -794,19 +754,14 @@ class CorporateRegistryFetcher:
         to_date = datetime.now().strftime("%Y-%m-%d")
 
         publications = self.info_financiere.search_publications(
-            query=query,
-            from_date=from_date,
-            to_date=to_date,
-            per_page=100
+            query=query, from_date=from_date, to_date=to_date, per_page=100
         )
 
         logger.info(f"Fetched {len(publications)} French publications")
         return publications
 
     def fetch_xbrl_eu_filings(
-        self,
-        country: Optional[str] = None,
-        days_back: int = 30
+        self, country: Optional[str] = None, days_back: int = 30
     ) -> List[Dict]:
         """
         Fetch EU/UK XBRL filings
@@ -822,11 +777,7 @@ class CorporateRegistryFetcher:
 
         from_date = (datetime.now() - timedelta(days=days_back)).strftime("%Y-%m-%d")
 
-        filings = self.xbrl_filings.get_filings(
-            country=country,
-            from_date=from_date,
-            page_size=100
-        )
+        filings = self.xbrl_filings.get_filings(country=country, from_date=from_date, page_size=100)
 
         logger.info(f"Fetched {len(filings)} XBRL filings")
         return filings