first

agent.py

@@ -0,0 +1,113 @@
+import typer
+from sentence_transformers import SentenceTransformer
+from sentence_transformers import SentenceTransformer, util
+import os
+import torch
+import torch._dynamo
+import json
+from rich import print
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+from rich.markdown import Markdown
+import warnings
+import logging
+import contextlib
+from PIL import Image
+
+
+warnings.filterwarnings('ignore')
+logging.getLogger().setLevel(logging.CRITICAL)
+torch._dynamo.config.suppress_errors = True
+logging.getLogger("torch").setLevel(logging.CRITICAL)
+logging.getLogger("torch._dynamo").setLevel(logging.CRITICAL)
+logging.getLogger("triton").setLevel(logging.CRITICAL)
+
+MODEL_NAME = "answerdotai/ModernBERT-base"
+META_FILE = 'meta.json'
+console = Console()
+
+@contextlib.contextmanager
+def suppress_outputs():
+    """Context manager to suppress all outputs, warnings, and errors temporarily"""
+    with open(os.devnull, 'w') as devnull:
+        with contextlib.redirect_stdout(devnull), contextlib.redirect_stderr(devnull):
+            with warnings.catch_warnings():
+                warnings.simplefilter("ignore")
+                yield
+
+def load_model(model_name):
+    with suppress_outputs():
+        model = SentenceTransformer(model_name)
+    return model
+
+
+def load_meta(meta_file):
+    with open(meta_file, 'r') as f:
+        return json.loads(f.read())
+
+def embed(items, model):
+    return model.encode(items)
+
+def load_meta_emb(meta, model):
+    return embed([doc['description'] for doc in meta], model)
+
+def get_meta_match(meta_emb, q_emb, model):
+    match = torch.topk(util.pytorch_cos_sim(q_emb, meta_emb), k=1)
+    match_idx, score = match.indices[0][0].item(), match.values[0][0].item()
+    return match_idx, score
+
+
+def ask(question, model, meta, meta_emb):
+    question_emb = embed(question, model)
+    match_idx, score = get_meta_match(meta_emb, question_emb, model)
+    return meta[match_idx], score
+
+def format_response(match: dict, score: float) -> Panel:
+    table = Table(show_header=False, box=None, padding=(0, 2))
+    table.add_row(
+        "[bold blue]Command:[/bold blue]",
+        match.get('name', 'N/A')
+    )
+    if 'examples' in match and match['examples']:
+        examples_md = "\n".join(f"```bash\n{example}\n```" for example in match['examples'])
+        table.add_row(
+            "[bold blue]Examples:[/bold blue]",
+            ""
+        )
+        table.add_row(
+            "",
+            Markdown(examples_md)
+        )
+    return Panel(
+        table,
+        title=f"🦦 [bold]Match found (confidence: {score:.2f})[/bold]",
+        border_style="green"
+    )
+
+
+def main():
+    console.print(Panel.fit("🦦 Suika Loading ...", title="Initializing"))
+    model = load_model(MODEL_NAME)
+    meta = load_meta(META_FILE)
+    meta_emb = load_meta_emb(meta, model)
+
+    console.print(Panel.fit("✨ Ask me any linux commands (type 'exit' to quit)", title="🦦 Suika Ready"))
+    while True:
+        question = typer.prompt("\nYour question")
+        if question.lower() == 'exit':
+            console.print("\n🦦 👋 Goodbye!")
+            break   
+        try:
+            match, score = ask(question, model, meta, meta_emb)
+            response_panel = format_response(match, score)
+            console.print(response_panel)
+            feedback = typer.confirm("\n🦦 Was this response helpful?")
+            if not feedback:
+                console.print("[yellow]🦦 I'm sorry the response wasn't helpful. Please try rephrasing your question.[/yellow]")      
+        except Exception as e:
+            console.print(f"[red]🦦 An error occurred: {str(e)}[/red]")
+            console.print("🦦 Please try again with a different question.")
+
+if __name__ == "__main__":
+    typer.run(main)

banner.txt

@@ -0,0 +1,23 @@
+▒░░░░░░░░░░░░░░░░░░░░░░░▒█████▓▓▓▓▒░░░░░
+▒░░░░░░░░░░░░░░░░░░██▓░░████████████▓▒░░
+▒▒░░░░░░░░░░░░░░░░░███▓▓█████████▓▓▒▒░░░
+▒▒▒▒░░░░░░░░░░░▒▒▒████▓▒▒░░▒▒▒▒▒░░░░░░░░
+▒▒▒▒░░░░░░░░░▒▒▒▓▓▓▓▓▒▓▓▓▒▒░░░░░░░░░░░░░
+▒▒▒░░░░░▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▓▓▒▒▒▒▒░░░░░░░░
+▒▒░░░░▒▒░░░░░▒▒▒▒▒▒▒▒▒▒▒▒▒▒▓▓▓▓▒▒▒░░░░░░
+▒▒░░▒▒▒░░░░░▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▓▒▒▒▒▒▒░░░░
+▒▒░▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▓▓▒▒▒▒▒▒▒░░░
+░░▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▒▓▓▒▒▓▓▒█▓▒▒▒▒▒▒▓░░░
+░░▒▒▒▒▒▒▒▓▓▓▓▒▒▒▒▒▒▒▓▓▓▓▓░░▒█▓░░▒▒▒▒█▓░░
+░░▒▒▒▒▒▓█▒░░▒█▓▓▒▒▒▒▓▓▓▒░░▓▒░░▒▓░▒▒▒█▓░░
+░░▒▒▒▒▒▓▓░░░░▓█▓▓▒▒▒▒▓▒▓▒▒▓░░░░█▒▓▒▓█▓░░
+░░▒▒▒▒▒▓▓░░░░▓█▓▓▒▒▒▒▒▒▒▓▒▓░░░░█░▒▓██░░░
+░░░▒▒▒▒▒▓▓░░▒▓▓▒▒▒▒▒▒▒▒▒▒░▒▓▒▒▒░░▓██▒░░░
+░░░░▒▒▒▒▒▒▒▒▒▒▒▒▒▒▓▓█▓▓▒▒▒▒░▒▓▒▒███▓░░░░
+░░░░▒▒▒▒▒▒▒▒▒▒▒▒▓██▓▓▓██▓▒▒▓█▓▓▓██▒░░░░░
+░░░░░░▒▒▒▒▒▒▓███▓▒░░░░░▒▒▓██████▓░░░░░░░
+░▒░░░░░░▒▓▓▓██▓▒░░░░░░░▒▒▒▓▓▓▓▒░░░░░░░░░
+▒▒▒▒▒▒▒░░░▒▓▓▓▓▓▓▒▓██▓▓▓▓▒▓▒▒░░░░░░░░░░░
+▒▒▒▒▒▒▒▒▒▒▓██▓▒▒▓▒███▓▒▒▒▒▓██▒░░░░░░░░░░
+▒▒▒▒▒▒▒▒▒▒▓██░░░░▓████▓░░░░██▓░░░░░░░░░░
+▒▒▒▒▒▒▒▒▒▒▓██▓▓▓███████▒░▒▓███▒░░░░░░░░░

meta.json

@@ -0,0 +1,72 @@
+[
+    {
+        "name": "ls",
+        "description": "Display directory contents and list files and folders",
+        "syntax": "ls [options] [directory]",
+        "keywords": ["list", "display", "show", "directory", "files", "folders", "contents"],
+        "examples": ["ls -la", "ls Documents/"]
+    },
+    {
+        "name": "cat",
+        "description": "View and display the contents of text files",
+        "syntax": "cat [options] [file]",
+        "keywords": ["view", "read", "display", "contents", "text", "file", "show"],
+        "examples": ["cat file.txt", "cat -n script.py"]
+    },
+    {
+        "name": "mkdir",
+        "description": "Create new directories and folder structures",
+        "syntax": "mkdir [options] directory",
+        "keywords": ["create", "new", "directory", "folder", "make"],
+        "examples": ["mkdir new_folder", "mkdir -p path/to/new/dir"]
+    },
+    {
+        "name": "rm",
+        "description": "Delete and remove files or directories permanently",
+        "syntax": "rm [options] file",
+        "keywords": ["remove", "delete", "erase", "eliminate"],
+        "examples": ["rm file.txt", "rm -rf old_directory"]
+    },
+    {
+        "name": "chmod",
+        "description": "Modify and change file or directory permissions and access rights",
+        "syntax": "chmod [options] mode file",
+        "keywords": ["permissions", "access", "rights", "modify", "change"],
+        "examples": ["chmod 755 script.sh", "chmod -R u+x directory"]
+    },
+    {
+        "name": "grep",
+        "description": "Search and find text patterns within files and content",
+        "syntax": "grep [options] pattern [file...]",
+        "keywords": ["search", "find", "text", "pattern", "content"],
+        "examples": ["grep 'pattern' file.txt", "grep -r 'text' ."]
+    },
+    {
+        "name": "df",
+        "description": "Check disk space usage and available storage",
+        "syntax": "df [options] [filesystem]",
+        "keywords": ["disk", "space", "storage", "usage", "available"],
+        "examples": ["df -h", "df -T /dev/sda1"]
+    },
+    {
+        "name": "ps",
+        "description": "List running processes and system tasks",
+        "syntax": "ps [options]",
+        "keywords": ["process", "running", "tasks", "programs", "active"],
+        "examples": ["ps aux", "ps -ef | grep python"]
+    },
+    {
+        "name": "tar",
+        "description": "Compress, archive, and package files and directories",
+        "syntax": "tar [options] [archive] [files...]",
+        "keywords": ["compress", "archive", "package", "zip", "bundle"],
+        "examples": ["tar -czf archive.tar.gz files/", "tar -xzf archive.tar.gz"]
+    },
+    {
+        "name": "cp",
+        "description": "Copy and duplicate files and directories to new locations",
+        "syntax": "cp [options] source destination",
+        "keywords": ["copy", "duplicate", "replicate", "transfer"],
+        "examples": ["cp file.txt backup/", "cp -r source_dir/ dest_dir/"]
+    }
+]

ruff.toml

@@ -0,0 +1,3 @@
+line-length = 120
+target-version = "py311"
+ignore = ["E402"]