openbb-cli 1.1.6__tar.gz → 1.1.8__tar.gz

{openbb_cli-1.1.6 → openbb_cli-1.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: openbb-cli
-Version: 1.1.6
+Version: 1.1.8
 Summary: Investment Research for Everyone, Anywhere.
 License: AGPL-3.0-only
 Author: OpenBB
@@ -12,12 +12,12 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: openbb[all] (>=4.4.0,<5.0.0)
+Requires-Dist: openbb[all] (>=4.4.2,<5.0.0)
 Requires-Dist: openpyxl (>=3.1.5,<4.0.0)
 Requires-Dist: prompt-toolkit (>=3.0.50,<4.0.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: pywry (>=0.6.2,<0.7.0)
-Requires-Dist: rich (>=13.9.4,<14.0.0)
+Requires-Dist: rich (>=14.0.0,<15.0.0)
 Project-URL: Documentation, https://docs.openbb.co/cli
 Project-URL: Homepage, https://openbb.co
 Project-URL: Repository, https://github.com/OpenBB-finance/OpenBB

{openbb_cli-1.1.6 → openbb_cli-1.1.8}/openbb_cli/argparse_translator/argparse_translator.py

@@ -178,14 +178,13 @@ class ArgparseTranslator:
         """Return the argparse action type for the given parameter."""
         param_type = self.type_hints[param.name]
         type_origin = get_origin(param_type)
-
-        if param_type == bool or (
+        if param_type is bool or (
             type_origin is Union and bool in get_args(param_type)
         ):
             return "store_true"
         return "store"
 
-    def _get_type_and_choices(
+    def _get_type_and_choices(  # noqa: PLR0912 # pylint: disable=too-many-return-statements, too-many-branches, too-many-statements
         self, param: inspect.Parameter
     ) -> Tuple[Type[Any], Tuple[Any, ...]]:
         """Return the type and choices for the given parameter."""
@@ -196,6 +195,9 @@ class ArgparseTranslator:
 
         if type_origin is Literal:
             choices = get_args(param_type)
+            # Special handling for boolean literals
+            if all(isinstance(choice, bool) for choice in choices):
+                return bool, ()
             param_type = type(choices[0])  # type: ignore
 
         if type_origin is list:
@@ -203,30 +205,66 @@ class ArgparseTranslator:
 
             if get_origin(param_type) is Literal:
                 choices = get_args(param_type)
+                # Special handling for boolean literals in lists
+                if all(isinstance(choice, bool) for choice in choices):
+                    return bool, ()
                 param_type = type(choices[0])  # type: ignore
 
         if type_origin is Union:
             union_args = get_args(param_type)
-            if str in union_args:
+            # Check if Union contains bool type
+            if bool in union_args and (str in union_args or len(union_args) == 2):
+                return bool, ()
+
+            # Check if Union contains Literal types and extract all choices
+            literal_choices: list = []
+            for arg in union_args:
+                if get_origin(arg) is Literal:
+                    literal_choices.extend(get_args(arg))
+
+            if literal_choices:
+                # Check if all choices are boolean
+                if all(isinstance(choice, bool) for choice in literal_choices):
+                    return bool, ()
+                # We have Literal types in the Union, use their choices
+                choices = tuple(literal_choices)
+                param_type = type(choices[0])  # type: ignore
+            elif str in union_args:
                 param_type = str
 
             # check if it's an Optional, which would be a Union with NoneType
             if type(None) in get_args(param_type):
                 # remove NoneType from the args
-                args = [arg for arg in get_args(param_type) if arg != type(None)]
+                args = [arg for arg in get_args(param_type) if arg is not None]
                 # if there is only one arg left, use it
-                if len(args) > 1:
-                    raise ValueError(
-                        "Union with NoneType should have only one type left"
-                    )
-                param_type = args[0]
-
-                if get_origin(param_type) is Literal:
-                    choices = get_args(param_type)
-                    param_type = type(choices[0])  # type: ignore
+                if len(args) == 1:
+                    param_type = args[0]
+
+                    if get_origin(param_type) is Literal:
+                        choices = get_args(param_type)
+                        # Special handling for boolean literals
+                        if all(isinstance(choice, bool) for choice in choices):
+                            return bool, ()
+                        param_type = type(choices[0])  # type: ignore
+                elif len(args) > 1:
+                    # Handle Union with multiple types (not just Optional)
+                    # Try to extract Literal types again from the filtered args
+                    literal_choices = []
+                    for arg in args:
+                        if get_origin(arg) is Literal:
+                            literal_choices.extend(get_args(arg))
+
+                    if literal_choices:
+                        # Check if all choices are boolean
+                        if all(isinstance(choice, bool) for choice in literal_choices):
+                            return bool, ()
+                        choices = tuple(set(literal_choices))
+                        param_type = type(choices[0])  # type: ignore
 
         # if there are custom choices, override
-        choices = self._get_argument_custom_choices(param) or choices  # type: ignore
+        custom_choices = self._get_argument_custom_choices(param)
+        if custom_choices and param_type is not bool:
+            choices = tuple(custom_choices)
 
         return param_type, choices
 
@@ -329,13 +367,21 @@ class ArgparseTranslator:
 
             required = not self._param_is_default(param)
 
+            # Get the appropriate action based on the parameter type
+            action = self._get_action_type(param)
+
+            # For boolean parameters with action="store_true", we should not use any choices
+            if param_type is bool:
+                choices = ()
+                action = "store_true"
+
             argument = ArgparseArgumentModel(
                 name=param.name,
                 type=param_type,
                 dest=param.name,
                 default=param.default,
                 required=required,
-                action=self._get_action_type(param),
+                action=action,
                 help=self._get_argument_custom_help(param),
                 nargs=self._get_nargs(param),
                 choices=choices,

{openbb_cli-1.1.6 → openbb_cli-1.1.8}/openbb_cli/argparse_translator/obbject_registry.py

@@ -89,7 +89,9 @@ class Registry:
             if hasattr(obbject, "results") and obbject.results:
                 data_schema = (
                     obbject.results[0].model_json_schema()
-                    if obbject.results and isinstance(obbject.results, list)
+                    if obbject.results
+                    and isinstance(obbject.results, list)
+                    and hasattr(obbject.results[0], "model_json_schema")
                     else ""
                 )
                 if data_schema and "title" in data_schema:

{openbb_cli-1.1.6 → openbb_cli-1.1.8}/openbb_cli/controllers/base_platform_controller.py

@@ -34,7 +34,7 @@ class PlatformController(BaseController):
 
     CHOICES_GENERATION = True
 
-    def __init__(
+    def __init__(  # pylint: disable=too-many-positional-arguments
         self,
         name: str,
         parent_path: List[str],
@@ -174,6 +174,9 @@ class PlatformController(BaseController):
                     title = f"{self.PATH}{translator.func.__name__}"
 
                     if obbject:
+                        if isinstance(obbject, list):
+                            obbject = OBBject(results=obbject)
+
                         if isinstance(obbject, OBBject):
                             if (
                                 session.max_obbjects_exceeded()

{openbb_cli-1.1.6 → openbb_cli-1.1.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "openbb-cli"
-version = "1.1.6"
+version = "1.1.8"
 description = "Investment Research for Everyone, Anywhere."
 authors = ["OpenBB <hello@openbb.co>"]
 packages = [{ include = "openbb_cli" }]
@@ -17,11 +17,11 @@ openbb = 'openbb_cli.cli:main'
 python = "^3.9,<3.13"
 
 # OpenBB dependencies
-openbb = { version = "^4.4.0", extras = ["all"] }
+openbb = { version = "^4.4.2", extras = ["all"] }
 
 # CLI dependencies
 prompt-toolkit = "^3.0.50"
-rich = "^13.9.4"
+rich = "^14.0.0"
 python-dotenv = "^1.0.1"
 openpyxl = "^3.1.5"
 pywry = "^0.6.2"