qtype 0.0.3__py3-none-any.whl → 0.0.5__py3-none-any.whl

qtype/cli.py

@@ -7,10 +7,33 @@ import importlib
 import logging
 from pathlib import Path
 
+try:
+    from importlib.metadata import entry_points
+except ImportError:
+    # Fallback for Python < 3.10 (though you require 3.10+)
+    from importlib_metadata import entry_points  # type: ignore[no-redef]
+
 
 def _discover_commands(subparsers: argparse._SubParsersAction) -> None:
     """Automatically discover and register command modules.
 
+    This function discovers commands from two sources:
+    1. Built-in commands from the local commands directory
+    2. Plugin commands from entry points defined by third-party packages
+
+    Args:
+        subparsers: The subparsers object to add commands to.
+    """
+    # First, discover built-in commands from the local commands directory
+    _discover_local_commands(subparsers)
+
+    # Then, discover plugin commands from entry points
+    _discover_plugin_commands(subparsers)
+
+
+def _discover_local_commands(subparsers: argparse._SubParsersAction) -> None:
+    """Discover and register built-in command modules from the local commands directory.
+
     Args:
         subparsers: The subparsers object to add commands to.
     """
@@ -29,15 +52,61 @@ def _discover_commands(subparsers: argparse._SubParsersAction) -> None:
                 module.parser(subparsers)
             else:
                 logging.warning(
-                    f"Command module {module_name} does not have a 'parser' function"
+                    f"Built-in command module {module_name} does not have a 'parser' function"
                 )
         except Exception as e:
             logging.error(
-                f"Failed to load command module {module_name}: {e}",
+                f"Failed to load built-in command module {module_name}: {e}",
                 exc_info=True,
             )
 
 
+def _discover_plugin_commands(subparsers: argparse._SubParsersAction) -> None:
+    """Discover and register plugin command modules from entry points.
+
+    Third-party packages can register commands by defining entry points in their
+    setup.py or pyproject.toml file like this:
+
+    [project.entry-points."qtype.commands"]
+    my-command = "my_package.qtype_commands:my_command_parser"
+
+    Args:
+        subparsers: The subparsers object to add commands to.
+    """
+    try:
+        # Get all entry points for the 'qtype.commands' group
+        eps = entry_points(group="qtype.commands")
+
+        for entry_point in eps:
+            try:
+                # Load the parser function from the entry point
+                parser_func = entry_point.load()
+
+                # Call the parser function to set up the subparser
+                if callable(parser_func):
+                    parser_func(subparsers)
+                    logging.debug(
+                        f"Successfully loaded plugin command '{entry_point.name}' "
+                        f"from {entry_point.value}"
+                    )
+                else:
+                    logging.warning(
+                        f"Plugin entry point '{entry_point.name}' "
+                        f"({entry_point.value}) is not callable"
+                    )
+            except Exception as e:
+                logging.error(
+                    f"Failed to load plugin command '{entry_point.name}' "
+                    f"from {entry_point.value}: {e}",
+                    exc_info=True,
+                )
+    except Exception as e:
+        logging.error(
+            f"Failed to discover plugin commands: {e}",
+            exc_info=True,
+        )
+
+
 def main() -> None:
     """
     Main entry point for the QType CLI.

qtype/interpreter/api.py

@@ -23,13 +23,16 @@ class APIExecutor:
         self.host = host
         self.port = port
 
-    def create_app(self, name: Optional[str]) -> FastAPI:
+    def create_app(
+        self,
+        name: Optional[str],
+        fast_api_args: dict = {
+            "docs_url": "/docs",
+            "redoc_url": "/redoc",
+        },
+    ) -> FastAPI:
         """Create FastAPI app with dynamic endpoints."""
-        app = FastAPI(
-            title=name or "QType API",
-            docs_url="/docs",  # Swagger UI
-            redoc_url="/redoc",
-        )
+        app = FastAPI(title=name or "QType API", **fast_api_args)
 
         flows = self.definition.flows if self.definition.flows else []
 

{qtype-0.0.3.dist-info → qtype-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qtype
-Version: 0.0.3
+Version: 0.0.5
 Summary: DSL for Generative AI Prototyping
 Author-email: Lou Kratz <lou.kratz+qtype@bazaarvoice.com>
 License-Expression: Apache-2.0

{qtype-0.0.3.dist-info → qtype-0.0.5.dist-info}/RECORD

@@ -1,5 +1,5 @@
 qtype/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qtype/cli.py,sha256=TjqZ6N0RGRZAcFm8Q6WkzXS_oV6cvYvsmbFdP8x9a-o,2113
+qtype/cli.py,sha256=XSVtfzpjTbQBQ60iyb5oupwi-mFpGH5jT8H1bJQdjis,4697
 qtype/loader.py,sha256=4Ty5B0qWNQeehexh0anlP7hDmsso0bj7-IVt5VqnHWA,10992
 qtype/commands/__init__.py,sha256=Qo4M07zm5I63r8STxDjvt5fhP1jygdXTsExNGELkefc,257
 qtype/commands/convert.py,sha256=Mkk8BAJDOmZ_nPsa0YtzjdxxBMScWk-qeYye4ojWA0o,2453
@@ -20,7 +20,7 @@ qtype/dsl/domain_types.py,sha256=VglaF7eTVjUmLvubQeNQo2ZA0pRooQXZ-OPJ6edQBIk,153
 qtype/dsl/model.py,sha256=mAgqfG1us1mT53nYd0gAIwOSgudLSDlwQjUlG5MDfRg,21169
 qtype/dsl/validator.py,sha256=yuDFPHfErPTIpZXCIyBrwpZJAitMEW0jNL8867KixUA,17130
 qtype/interpreter/__init__.py,sha256=IaRF90JLFbsTLKz9LTOMI_Pz4xwVaEyXPNaXV7sLou8,43
-qtype/interpreter/api.py,sha256=DrBV0fF6f4HiGSgXzHE07FwOr5Qe7uBQ2yQ6UjDOUTU,3872
+qtype/interpreter/api.py,sha256=9hwc8pPrfbUDxHfEERsnEtPRBVjU1ziSyBgGIR5gWDw,3924
 qtype/interpreter/conversions.py,sha256=d_hyWWR0PlhvRf2EiNt1sEzhFH8gIl3vMrVnSOFUZGk,5612
 qtype/interpreter/exceptions.py,sha256=Il8IF0UAtYWQXwvOVQCY-csfRzC1iOejHM1G-nF5EfY,288
 qtype/interpreter/flow.py,sha256=2u1wRahNFQaRRklnU4uW7_UKSD73-uZe_WiYlKitXQg,1233
@@ -42,9 +42,9 @@ qtype/semantic/errors.py,sha256=dmdxGkVHrTwfiEHj5RFu2q11dAtefsivpLdJd6QynK0,119
 qtype/semantic/generate.py,sha256=U4r5Yo1b0wTBgw3dLnqjYD6wD6lc0TqdwUjhIV_yQRI,13085
 qtype/semantic/model.py,sha256=Hd3rsTdvkHtAeIrPqpJf0yjHa697TnrUbm3pbrIgSK0,11047
 qtype/semantic/resolver.py,sha256=D3CmRmr_ACL54ZCbc9HxXCuCZ37bNtSzlPLMpOeeOcw,3439
-qtype-0.0.3.dist-info/licenses/LICENSE,sha256=1KA5EgYBSR0O6nCH2HEvk6Di53YKJ9r_VCR7G8G8qAY,11341
-qtype-0.0.3.dist-info/METADATA,sha256=F4WqzZVkHDvHI8gl7bzB7P0bGbQcYGuUoB-Wb4SOkLE,4809
-qtype-0.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-qtype-0.0.3.dist-info/entry_points.txt,sha256=5y4vj8RLvgl2tXSj-Hm7v5-Tn3kP4-UonjNoN-mfaQE,41
-qtype-0.0.3.dist-info/top_level.txt,sha256=ONroH5B0mZ51jr7NSWCK0weFwwCO7wBLmyVS1YqNU14,6
-qtype-0.0.3.dist-info/RECORD,,
+qtype-0.0.5.dist-info/licenses/LICENSE,sha256=1KA5EgYBSR0O6nCH2HEvk6Di53YKJ9r_VCR7G8G8qAY,11341
+qtype-0.0.5.dist-info/METADATA,sha256=0T5PLI4kg58aPEnqie7J1Oxuc2edhz53R7S4Qi01uD4,4809
+qtype-0.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+qtype-0.0.5.dist-info/entry_points.txt,sha256=5y4vj8RLvgl2tXSj-Hm7v5-Tn3kP4-UonjNoN-mfaQE,41
+qtype-0.0.5.dist-info/top_level.txt,sha256=ONroH5B0mZ51jr7NSWCK0weFwwCO7wBLmyVS1YqNU14,6
+qtype-0.0.5.dist-info/RECORD,,