lmnr 0.2.10__py3-none-any.whl → 0.2.12__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,47 @@ 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")
+
+    print(f"Sys path: {sys.path}")
+
+    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)
+
+    print(f"Sys path: {sys.path}")
+
+    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/cli/zip.py

@@ -7,6 +7,10 @@ def zip_directory(directory_path: Path, zip_file_path: Path):
     with zipfile.ZipFile(zip_file_path, "w", zipfile.ZIP_DEFLATED) as zipf:
         for root, _, files in os.walk(directory_path):
             for file in files:
+                # Don't include the zip file itself, otherwise goes to infinite loop
+                if file == zip_file_path.name:
+                    continue
+
                 file_path = os.path.join(root, file)
                 arcname = os.path.relpath(file_path, directory_path)
                 zipf.write(file_path, arcname)

{lmnr-0.2.10.dist-info → lmnr-0.2.12.dist-info}/METADATA

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

{lmnr-0.2.10.dist-info → lmnr-0.2.12.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=HWnsIc5kFajq_VVJdJzRzEeNhBiVLgKaVp-GmC1itxI,5581
 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
@@ -15,14 +15,14 @@ lmnr/cli/parser/nodes/semantic_search.py,sha256=DWDPpV78XZ7vPIaPd86FbeDFAnKah4e6
 lmnr/cli/parser/nodes/types.py,sha256=OVXj-iMEDY9nPKCX1-zddtoszZcUL3CXYYryI7O3et0,6094
 lmnr/cli/parser/parser.py,sha256=yDa-ysAkh6si_hHU8Gw8EdtNWc4pFc5RbvgWEXGEPys,2370
 lmnr/cli/parser/utils.py,sha256=1oy6BApHXOF7BTXbP8v3Oi9bwOdWZjoxDlRIOfXVxro,1169
-lmnr/cli/zip.py,sha256=DpAfSVg6k0Ly55SY7rsIZkX0XG6U6IPD-HzsNi5Cf-U,445
+lmnr/cli/zip.py,sha256=u2-LcYtQdZ_FIW0-PM-WGjclNPoB8v6OecrI79PyLPw,607
 lmnr/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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.10.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.2.10.dist-info/METADATA,sha256=STv0WzrtahyFBUaq_uLDuPVts5pd3dLgCfjdeQJ0Oyk,5565
-lmnr-0.2.10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-lmnr-0.2.10.dist-info/entry_points.txt,sha256=Qg7ZRax4k-rcQsZ26XRYQ8YFSBiyY2PNxYfq4a6PYXI,41
-lmnr-0.2.10.dist-info/RECORD,,
+lmnr-0.2.12.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
+lmnr-0.2.12.dist-info/METADATA,sha256=ioChA7mM_fEj2KfrvXEtsHnyH5uTMgkPD46b-HcvkfE,5565
+lmnr-0.2.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+lmnr-0.2.12.dist-info/entry_points.txt,sha256=Qg7ZRax4k-rcQsZ26XRYQ8YFSBiyY2PNxYfq4a6PYXI,41
+lmnr-0.2.12.dist-info/RECORD,,