lmnr 0.2.11__py3-none-any.whl → 0.2.13__py3-none-any.whl

lmnr/cli/cli.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+import sys
 import requests
 from dotenv import load_dotenv
 import os
@@ -8,6 +9,8 @@ from cookiecutter.main import cookiecutter
 from pydantic.alias_generators import to_pascal
 
 from lmnr.cli.zip import zip_directory
+from lmnr.sdk.registry import Registry as Pipeline
+from lmnr.sdk.remote_debugger import RemoteDebugger
 
 from .parser.parser import runnable_graph_to_template_vars
 
@@ -134,3 +137,43 @@ def deploy(endpoint_id, project_api_key):
         logging.exception("Error in deploying code")
     finally:
         Path.unlink(zip_file_path, missing_ok=True)
+
+
+@cli.command(name="dev")
+@click.option(
+    "-p",
+    "--project-api-key",
+    help="Project API key",
+)
+def dev(project_api_key):
+    project_api_key = project_api_key or os.environ.get("LMNR_PROJECT_API_KEY")
+    if not project_api_key:
+        load_dotenv()
+        project_api_key = os.environ.get("LMNR_PROJECT_API_KEY")
+    if not project_api_key:
+        raise ValueError("LMNR_PROJECT_API_KEY is not set")
+
+    cur_dir = os.getcwd()  # e.g. /Users/username/project_name
+    parent_dir, name = os.path.split(cur_dir)  # e.g. /Users/username, project_name
+
+    # Needed to __import__ pipeline.py
+    if sys.path[0] != parent_dir:
+        sys.path.insert(0, parent_dir)
+    # Needed to import src in pipeline.py and other files
+    if cur_dir not in sys.path:
+        sys.path.insert(0, cur_dir)
+
+    module_name = f"{name}.pipeline"
+    __import__(module_name)
+    module = sys.modules[module_name]
+
+    matches = [v for v in module.__dict__.values() if isinstance(v, Pipeline)]
+    if not matches:
+        raise ValueError("No Pipeline found in the module")
+    if len(matches) > 1:
+        raise ValueError("Multiple Pipelines found in the module")
+    pipeline = matches[0]
+
+    tools = pipeline.functions
+    debugger = RemoteDebugger(project_api_key, tools)
+    debugger.start()

{lmnr-0.2.11.dist-info → lmnr-0.2.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lmnr
-Version: 0.2.11
+Version: 0.2.13
 Summary: Python SDK for Laminar AI
 License: Apache-2.0
 Author: lmnr.ai

{lmnr-0.2.11.dist-info → lmnr-0.2.13.dist-info}/RECORD

@@ -1,7 +1,7 @@
 lmnr/__init__.py,sha256=BBJ87AiHC_OpaYOCzF8QSsf7eO3LlJPOCBXHNKurbE4,235
 lmnr/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lmnr/cli/__main__.py,sha256=8hDtWlaFZK24KhfNq_ZKgtXqYHsDQDetukOCMlsbW0Q,59
-lmnr/cli/cli.py,sha256=qG3-aY2EJIW3BSD-_s1vUsxHcxC9kiuIeAO8aVUq3ik,4103
+lmnr/cli/cli.py,sha256=6xmIxRAMx7th0S9rfkzUeWyt-oWaiwPWk4YeVzXhElQ,5509
 lmnr/cli/parser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lmnr/cli/parser/nodes/__init__.py,sha256=2MkPdKulb1kuNe6aT71CaqBA8iBrXyb5pq5bu_EvCb8,1052
 lmnr/cli/parser/nodes/code.py,sha256=8lTPBibUzaw_t-9QoPljhxH3KA4CLn9DJjA-iWpprOA,933
@@ -21,8 +21,8 @@ lmnr/sdk/endpoint.py,sha256=0HjcxMUcJz-klFZO2f5xtTaoLjcaEb8vrJ_YldTWUc8,7467
 lmnr/sdk/registry.py,sha256=sEYQFOjO72YvgBSEkBrvoewFExoyBzx6nELgBarvD6Y,755
 lmnr/sdk/remote_debugger.py,sha256=BYAN13KUDxH412qD3HXdDH0RfokrePquDt35fzB7GUg,5010
 lmnr/types.py,sha256=3HpLBQZr6F5YMISHYLnzzyrTwUttNqJxpyobw31YYJQ,2347
-lmnr-0.2.11.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.2.11.dist-info/METADATA,sha256=Vy8sFZj2301oMMAdDWMnqa23iWBSzBuy68WhEgvlVZU,5565
-lmnr-0.2.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-lmnr-0.2.11.dist-info/entry_points.txt,sha256=Qg7ZRax4k-rcQsZ26XRYQ8YFSBiyY2PNxYfq4a6PYXI,41
-lmnr-0.2.11.dist-info/RECORD,,
+lmnr-0.2.13.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
+lmnr-0.2.13.dist-info/METADATA,sha256=G0wy4T7F8TlHpFXky86Bq5hV3nt1VNZ6hf9XUVbraew,5565
+lmnr-0.2.13.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+lmnr-0.2.13.dist-info/entry_points.txt,sha256=Qg7ZRax4k-rcQsZ26XRYQ8YFSBiyY2PNxYfq4a6PYXI,41
+lmnr-0.2.13.dist-info/RECORD,,