symbolicai 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

symai/shell.py

@@ -44,18 +44,22 @@ $> New-Item -ItemType Directory -Path <path> EOF
 --------------
 """
 
+
 class ShellPreProcessor(PreProcessor):
     def __call__(self, argument):
-        return f'// {argument.prop.instance!s}\n$>'
+        return f"// {argument.prop.instance!s}\n$>"
 
 
 class Shell(Expression):
     def forward(self, **kwargs) -> str:
-        @few_shot(prompt="Convert a user query to a shell command:\n",
-                     examples=[],
-                     pre_processors=[ShellPreProcessor()],
-                     post_processors=[StripPostProcessor()],
-                     stop=['EOF'], **kwargs)
+        @few_shot(
+            prompt="Convert a user query to a shell command:\n",
+            examples=[],
+            pre_processors=[ShellPreProcessor()],
+            post_processors=[StripPostProcessor()],
+            stop=["EOF"],
+            **kwargs,
+        )
         def _func(_) -> str:
             return "Sorry, something went wrong. Please check if your backend is available and try again or report an issue to the devs. :("
 
@@ -68,17 +72,19 @@ class Shell(Expression):
 
 def process_query(args) -> None:
     if args.version:
-        with ConsoleStyle('extensity') as console:
+        with ConsoleStyle("extensity") as console:
             console.print(f"ExtensityAI: Symbolic Shell v{SYMAI_VERSION}")
         return
 
     query = args.query
     if query is None or len(query) == 0:
-        with ConsoleStyle('extensity') as console:
+        with ConsoleStyle("extensity") as console:
             console.print(f"Starting ExtensityAI: Symbolic Shell v{SYMAI_VERSION}")
-        shellsv_run(auto_query_on_error=args.auto,
-                    conversation_style=args.style if args.style is not None and args.style != '' else None,
-                    verbose=args.verbose)
+        shellsv_run(
+            auto_query_on_error=args.auto,
+            conversation_style=args.style if args.style is not None and args.style != "" else None,
+            verbose=args.verbose,
+        )
         return
 
     shell = Shell(query)
@@ -95,44 +101,97 @@ def process_query(args) -> None:
     if args.delete:
         msg = msg - args.delete
     if args.exec:
-        msg = str(msg).strip().replace('EOF', '')
+        msg = str(msg).strip().replace("EOF", "")
         os.system(msg)
     if args.more:
-        cmd = msg.extract('the main command used')
-        cmd_help = cmd << 'get manual for the command'
-        expr = Try(expr=Lambda(lambda kwargs: os.system(str(kwargs['args'][0]))))
+        cmd = msg.extract("the main command used")
+        cmd_help = cmd << "get manual for the command"
+        expr = Try(expr=Lambda(lambda kwargs: os.system(str(kwargs["args"][0]))))
         with Loader(desc="Inference ...", end=""):
             expr(cmd_help)
 
-    with ConsoleStyle('code') as console:
-        msg = str(msg).strip().replace('EOF', '')
+    with ConsoleStyle("code") as console:
+        msg = str(msg).strip().replace("EOF", "")
         console.print(msg)
 
 
 def run() -> None:
     # All the logic of argparse goes in this function
-    parser = argparse.ArgumentParser(description='Welcome to the Symbolic<AI/> Shell support tool!')
-    parser.add_argument('query', type=str, nargs='?', help='The prompt for the shell query.')
-    parser.add_argument('--add', dest='add', default="", required=False, type=str,
-                        help='integrate the added text to the query.')
-    parser.add_argument('--convert', dest='convert', default="", required=False, type=str,
-                        help='convert a command to another shell. (e.g. --convert=windows or --convert=linux)')
-    parser.add_argument('--edit', dest='edit', default="", required=False, type=str,
-                        help='edit the added text to the query.')
-    parser.add_argument('--del', dest='delete', default="", required=False, type=str,
-                        help='remove the added text to the query.')
-    parser.add_argument('--more', dest='more', default=False, required=False, action=argparse.BooleanOptionalAction,
-                        help='add more context to the generated command.')
-    parser.add_argument('--exec', dest='exec', default=False, required=False, action=argparse.BooleanOptionalAction,
-                        help='execute command after creation (ATTENTION: Executing a generated command without verification may be risky!).')
-    parser.add_argument('--auto', dest='auto', default=False, required=False, action=argparse.BooleanOptionalAction,
-                        help='query the the LLM if a command resulted unsuccessfully.')
-    parser.add_argument('--style', dest='style', default=None, required=False, type=str,
-                        help='add speech style to the shell.')
-    parser.add_argument('--version', dest='version', default=False, required=False, action=argparse.BooleanOptionalAction,
-                        help='show the version of the shell.')
-    parser.add_argument('--verbose', action='store_true',
-                        help='Print verbose errors.')
+    parser = argparse.ArgumentParser(description="Welcome to the Symbolic<AI/> Shell support tool!")
+    parser.add_argument("query", type=str, nargs="?", help="The prompt for the shell query.")
+    parser.add_argument(
+        "--add",
+        dest="add",
+        default="",
+        required=False,
+        type=str,
+        help="integrate the added text to the query.",
+    )
+    parser.add_argument(
+        "--convert",
+        dest="convert",
+        default="",
+        required=False,
+        type=str,
+        help="convert a command to another shell. (e.g. --convert=windows or --convert=linux)",
+    )
+    parser.add_argument(
+        "--edit",
+        dest="edit",
+        default="",
+        required=False,
+        type=str,
+        help="edit the added text to the query.",
+    )
+    parser.add_argument(
+        "--del",
+        dest="delete",
+        default="",
+        required=False,
+        type=str,
+        help="remove the added text to the query.",
+    )
+    parser.add_argument(
+        "--more",
+        dest="more",
+        default=False,
+        required=False,
+        action=argparse.BooleanOptionalAction,
+        help="add more context to the generated command.",
+    )
+    parser.add_argument(
+        "--exec",
+        dest="exec",
+        default=False,
+        required=False,
+        action=argparse.BooleanOptionalAction,
+        help="execute command after creation (ATTENTION: Executing a generated command without verification may be risky!).",
+    )
+    parser.add_argument(
+        "--auto",
+        dest="auto",
+        default=False,
+        required=False,
+        action=argparse.BooleanOptionalAction,
+        help="query the the LLM if a command resulted unsuccessfully.",
+    )
+    parser.add_argument(
+        "--style",
+        dest="style",
+        default=None,
+        required=False,
+        type=str,
+        help="add speech style to the shell.",
+    )
+    parser.add_argument(
+        "--version",
+        dest="version",
+        default=False,
+        required=False,
+        action=argparse.BooleanOptionalAction,
+        help="show the version of the shell.",
+    )
+    parser.add_argument("--verbose", action="store_true", help="Print verbose errors.")
 
     args = parser.parse_args()
     process_query(args)