lmnr 0.2.15__py3-none-any.whl → 0.3.0__py3-none-any.whl

lmnr/sdk/remote_debugger.py

@@ -1,148 +0,0 @@
-from typing import Callable, Optional, Union
-from websockets.sync.client import connect
-import pydantic
-import websockets
-from lmnr.types import (
-    DeregisterDebuggerRequest,
-    NodeFunction,
-    NodeInput,
-    RegisterDebuggerRequest,
-    SDKError,
-    ToolCallError,
-    ToolCallRequest,
-    ToolCallResponse,
-)
-import json
-from concurrent.futures import ThreadPoolExecutor
-from threading import Thread
-import time
-
-
-class RemoteDebugger:
-    def __init__(
-        self,
-        project_api_key: str,
-        dev_session_id: str,
-        tools: Union[dict[str, NodeFunction], list[Callable[..., NodeInput]]] = [],
-    ):
-        # for simplicity and backwards compatibility, we allow the user to pass a list
-        if isinstance(tools, list):
-            tools = {f.__name__: NodeFunction(f.__name__, f) for f in tools}
-
-        self.project_api_key = project_api_key
-        self.url = "wss://api.lmnr.ai/v2/endpoint/ws"
-        self.tools = tools
-        self.stop_flag = False
-        self.session = dev_session_id
-        self.executor = ThreadPoolExecutor(5)
-        self.running_tasks = {}  # dict[str, Future] from request_id to Future
-    
-    def start(self) -> Optional[str]:
-        self.stop_flag = False
-        self.executor.submit(self._run)
-        return self.session
-
-    def stop(self):
-        self.stop_flag = True
-        self.executor.shutdown()
-        self.session = None
-
-    def _run(self, backoff=1):
-        assert self.session is not None, "Session ID not set"
-        request = RegisterDebuggerRequest(debuggerSessionId=self.session)
-        try:
-            self._connect_and_run(request, backoff)
-        except Exception as e:
-            print(f"Could not connect to server. Retrying in {backoff} seconds...")
-            time.sleep(backoff)
-            self._run(min(backoff * 2, 60))
-    
-    def _connect_and_run(self, request: RegisterDebuggerRequest, backoff=1):
-        with connect(
-            self.url,
-            additional_headers={"Authorization": f"Bearer {self.project_api_key}"},
-        ) as websocket:
-            websocket.send(request.model_dump_json())
-            print(self._format_session_id_and_registerd_functions())
-            req_id = None
-
-            while not self.stop_flag:
-                # first check if any of the running tasks are done
-                done_tasks = []
-                for req_id, future in self.running_tasks.items():
-                    if not future.done():
-                        continue
-                    done_tasks.append(req_id)
-                    try:
-                        response = future.result()
-                    except Exception as e:
-                        error_message = (
-                            "Error occurred while running tool" + f"{tool.__name__}: {e}"
-                        )
-                        e = ToolCallError(error=error_message, reqId=req_id)
-                        websocket.send(e.model_dump_json())
-                        continue
-                    formatted_response = None
-                    try:
-                        formatted_response = ToolCallResponse(
-                            reqId=req_id, response=response
-                        )
-                    except pydantic.ValidationError:
-                        formatted_response = ToolCallResponse(
-                            reqId=req_id, response=str(response)
-                        )
-                    websocket.send(formatted_response.model_dump_json())
-                for req_id in done_tasks:
-                    del self.running_tasks[req_id]
-                try:
-                    # blocks the thread until a message
-                    # is received or a timeout (0.1 seconds) occurs
-                    message = websocket.recv(0.1)
-                except TimeoutError:
-                    continue
-                except websockets.exceptions.ConnectionClosedError:
-                    print("Connection interrupted by server. Trying to reconnect...")
-                    self._run()
-                try:
-                    tool_call = ToolCallRequest.model_validate_json(message)
-                    req_id = tool_call.reqId
-                except Exception:
-                    raise SDKError(f"Invalid message received:\n{message}")
-                matching_tool = self.tools.get(tool_call.toolCall.function.name)
-                if matching_tool is None:
-                    error_message = (
-                        f"Tool {tool_call.toolCall.function.name} not found"
-                        + ". Registered tools: "
-                        + ", ".join(self.tools.keys())
-                    )
-                    e = ToolCallError(error=error_message, reqId=req_id)
-                    websocket.send(e.model_dump_json())
-                    continue
-                tool = matching_tool.function
-
-                # default the arguments to an empty dictionary
-                arguments = {}
-                try:
-                    arguments = json.loads(tool_call.toolCall.function.arguments)
-                except Exception:
-                    pass
-                self.running_tasks[tool_call.reqId] = self.executor.submit(tool, **arguments)
-            websocket.send(
-                DeregisterDebuggerRequest(
-                    debuggerSessionId=self.session, deregister=True
-                ).model_dump_json()
-            )
-
-    def _format_session_id_and_registerd_functions(self) -> str:
-        registered_functions = ",\n".join(["- " + k for k in self.tools.keys()])
-        return f"""
-========================================
-Dev Session ID:
-{self.session}
-========================================
-
-Registered functions:
-{registered_functions}
-
-========================================
-"""

lmnr/types.py

@@ -1,101 +0,0 @@
-import requests
-import pydantic
-import uuid
-from typing import Callable, Union, Optional
-
-
-class ChatMessage(pydantic.BaseModel):
-    role: str
-    content: str
-
-
-class ConditionedValue(pydantic.BaseModel):
-    condition: str
-    value: "NodeInput"
-
-
-NodeInput = Union[str, list[ChatMessage], ConditionedValue]  # TypeAlias
-
-
-class EndpointRunRequest(pydantic.BaseModel):
-    inputs: dict[str, NodeInput]
-    endpoint: str
-    env: dict[str, str] = pydantic.Field(default_factory=dict)
-    metadata: dict[str, str] = pydantic.Field(default_factory=dict)
-
-
-class EndpointRunResponse(pydantic.BaseModel):
-    outputs: dict[str, dict[str, NodeInput]]
-    run_id: str
-
-
-class EndpointRunError(Exception):
-    error_code: str
-    error_message: str
-
-    def __init__(self, response: requests.Response):
-        try:
-            resp_json = response.json()
-            self.error_code = resp_json["error_code"]
-            self.error_message = resp_json["error_message"]
-            super().__init__(self.error_message)
-        except Exception:
-            super().__init__(response.text)
-
-    def __str__(self) -> str:
-        try:
-            return str(
-                {"error_code": self.error_code, "error_message": self.error_message}
-            )
-        except Exception:
-            return super().__str__()
-
-
-class SDKError(Exception):
-    def __init__(self, error_message: str):
-        super().__init__(error_message)
-
-
-class ToolCallFunction(pydantic.BaseModel):
-    name: str
-    arguments: str
-
-
-class ToolCall(pydantic.BaseModel):
-    id: Optional[str]
-    type: Optional[str]
-    function: ToolCallFunction
-
-
-# TODO: allow snake_case and manually convert to camelCase
-class ToolCallRequest(pydantic.BaseModel):
-    reqId: uuid.UUID
-    toolCall: ToolCall
-
-
-class ToolCallResponse(pydantic.BaseModel):
-    reqId: uuid.UUID
-    response: NodeInput
-
-
-class ToolCallError(pydantic.BaseModel):
-    reqId: uuid.UUID
-    error: str
-
-
-class RegisterDebuggerRequest(pydantic.BaseModel):
-    debuggerSessionId: str
-
-
-class DeregisterDebuggerRequest(pydantic.BaseModel):
-    debuggerSessionId: str
-    deregister: bool
-
-
-class NodeFunction:
-    node_name: str
-    function: Callable[..., NodeInput]
-
-    def __init__(self, node_name: str, function: Callable[..., NodeInput]):
-        self.node_name = node_name
-        self.function = function

lmnr-0.2.15.dist-info/METADATA

@@ -1,187 +0,0 @@
-Metadata-Version: 2.1
-Name: lmnr
-Version: 0.2.15
-Summary: Python SDK for Laminar AI
-License: Apache-2.0
-Author: lmnr.ai
-Requires-Python: >=3.9,<4.0
-Classifier: License :: OSI Approved :: Apache Software License
-Classifier: Programming Language :: Python :: 3
-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: black (>=24.4.2,<25.0.0)
-Requires-Dist: click (>=8.1.7,<9.0.0)
-Requires-Dist: cookiecutter (>=2.6.0,<3.0.0)
-Requires-Dist: lmnr-baml (>=0.40.1,<0.41.0)
-Requires-Dist: pydantic (>=2.7.4,<3.0.0)
-Requires-Dist: pystache (>=0.6.5,<0.7.0)
-Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
-Requires-Dist: requests (>=2.32.3,<3.0.0)
-Requires-Dist: watchdog (>=4.0.2,<5.0.0)
-Requires-Dist: websockets (>=12.0,<13.0)
-Description-Content-Type: text/markdown
-
-# Laminar AI
-
-This repo provides core for code generation, Laminar CLI, and Laminar SDK.
-
-## Quickstart
-```sh
-python3 -m venv .myenv
-source .myenv/bin/activate  # or use your favorite env management tool
-
-pip install lmnr
-```
-
-## Features
-
-- Make Laminar endpoint calls from your Python code
-- Make Laminar endpoint calls that can run your own functions as tools
-- CLI to generate code from pipelines you build on Laminar or execute your own functions while you test your flows in workshop
-
-## Making Laminar endpoint calls
-
-After you are ready to use your pipeline in your code, deploy it in Laminar following the [docs](https://docs.lmnr.ai/pipeline/run-save-deploy#deploying-a-pipeline-version).
-
-Once your pipeline is deployed, you can call it from Python in just a few lines.
-
-Example use:
-
-```python
-from lmnr import Laminar
-
-l = Laminar('<YOUR_PROJECT_API_KEY>')
-result = l.run(
-    endpoint = 'my_endpoint_name',
-    inputs = {'input_node_name': 'some_value'},
-    # all environment variables
-    env = {'OPENAI_API_KEY': 'sk-some-key'},
-    # any metadata to attach to this run's trace
-    metadata = {'session_id': 'your_custom_session_id'}
-)
-```
-
-Resulting in:
-
-```python
->>> result
-EndpointRunResponse(
-    outputs={'output': {'value': [ChatMessage(role='user', content='hello')]}},
-    # useful to locate your trace
-    run_id='53b012d5-5759-48a6-a9c5-0011610e3669'
-)
-```
-
-## Making calls to pipelines that run your own logic
-
-If your pipeline contains tool call nodes, they will be able to call your local code.
-The only difference is that you need to pass references
-to the functions you want to call right into our SDK.
-
-Example use:
-
-```python
-from lmnr import Laminar, NodeInput
-
-# adding **kwargs is safer, in case an LLM produces more arguments than needed
-def my_tool(arg1: string, arg2: string, **kwargs) -> NodeInput {
-    return f'{arg1}&{arg2}'
-}
-
-l = Laminar('<YOUR_PROJECT_API_KEY>')
-result = l.run(
-    endpoint = 'my_endpoint_name',
-    inputs = {'input_node_name': 'some_value'},
-    # all environment variables
-    env = {'OPENAI_API_KEY': '<YOUR_MODEL_PROVIDER_KEY>'},
-    # any metadata to attach to this run's trace
-    metadata = {'session_id': 'your_custom_session_id'},
-    # specify as many tools as needed.
-    # Each tool name must match tool node name in the pipeline
-    tools=[my_tool]
-)
-```
-
-## LaminarRemoteDebugger
-
-If your pipeline contains local call nodes, they will be able to call code right on your machine.
-
-### Step by step instructions to connect to Laminar:
-
-#### 1. Create your pipeline with function call nodes
-
-Add function calls to your pipeline; these are signature definitions of your functions
-
-#### 2. Implement the functions
-
-At the root level, create a file: `pipeline.py`
-
-Annotate functions with the same name.
-
-Example:
-
-```python
-from lmnr import Pipeline
-
-lmnr = Pipeline()
-
-@lmnr.func("foo") # the node in the pipeline is called foo and has one parameter arg
-def custom_logic(arg: str) -> str:
-    return arg * 10
-```
-
-#### 3. Link lmnr.ai workshop to your machine
-
-1. At the root level, create a `.env` file if not already
-1. In project settings, create or copy a project api key.
-1. Add an entry in `.env` with: `LMNR_PROJECT_API_KEY=s0meKey...`
-1. In project settings create or copy a dev session. These are your individual sessions.
-1. Add an entry in `.env` with: `LMNR_DEV_SESSION_ID=01234567-89ab-cdef-0123-4567890ab`
-
-#### 4. Run the dev environment
-
-```sh
-lmnr dev
-```
-
-This will start a session, try to persist it, and reload the session on files change.
-
-## CLI for code generation
-
-### Basic usage
-
-```
-lmnr pull <pipeline_name> <pipeline_version_name> --project-api-key <PROJECT_API_KEY>
-```
-
-Note that `lmnr` CLI command will only be available from within the virtual environment
-where you have installed the package.
-
-To import your pipeline
-```python
-# submodule with the name of your pipeline will be generated in lmnr_engine.pipelines
-from lmnr_engine.pipelines.my_custom_pipeline import MyCustomPipeline
-
-
-pipeline = MyCustomPipeline()
-res = pipeline.run(
-    inputs={
-        "instruction": "Write me a short linkedin post about a dev tool for LLM developers"
-    },
-    env={
-        "OPENAI_API_KEY": <OPENAI_API_KEY>,
-    }
-)
-print(f"Pipeline run result:\n{res}")
-```
-
-### Current functionality
-- Supports graph generation for graphs with the following nodes: Input, Output, LLM, Router, Code.
-- For LLM nodes, it only supports OpenAI and Anthropic models. Structured output in LLM nodes will be supported soon.
-
-## PROJECT_API_KEY
-
-Read more [here](https://docs.lmnr.ai/api-reference/introduction#authentication) on how to get `PROJECT_API_KEY`.
-

lmnr-0.2.15.dist-info/RECORD

@@ -1,28 +0,0 @@
-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=4wu9ke-S1dvFiTK1qkQsgzjAJ2iDnxfHA-AgmhQLvuY,7539
-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
-lmnr/cli/parser/nodes/condition.py,sha256=AJny0ILXbSy1hTwsRvZvDUqts9INNx63yQSkD7Dp7KU,740
-lmnr/cli/parser/nodes/input.py,sha256=o8EfCmBbNyQL8FzmAtgnNDFlWqZmRAgkbw4HzKXZepU,539
-lmnr/cli/parser/nodes/json_extractor.py,sha256=CnVwZ-wU_Ro4WkJLw9Uk_SS3yvZ66UPa5mK4JdkM8w4,723
-lmnr/cli/parser/nodes/llm.py,sha256=Wpmo9cfNiYN9DRbj7oBS6RYcKXLwlGtF6RdF4jFQm5I,1866
-lmnr/cli/parser/nodes/output.py,sha256=1XBppSscxM01kfZhE9oOh2GgdCVzyPVe2RAxLI5HmUc,665
-lmnr/cli/parser/nodes/router.py,sha256=dmCx4ho8_GdFJXQa8UevMf_uEP7AKBv_MJ2zpLC6Vck,894
-lmnr/cli/parser/nodes/semantic_search.py,sha256=DWDPpV78XZ7vPIaPd86FbeDFAnKah4e61M1TOzwnt84,1352
-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=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=c0a6_YZJmSmIyTL8Ybu6Ln4k8KiybwqcGqWude_Pi10,5756
-lmnr/types.py,sha256=3HpLBQZr6F5YMISHYLnzzyrTwUttNqJxpyobw31YYJQ,2347
-lmnr-0.2.15.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.2.15.dist-info/METADATA,sha256=9b7MMvOj7_OGUe3k3DBi8sAd2uaVh5QMlnxF1cEb91A,5565
-lmnr-0.2.15.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-lmnr-0.2.15.dist-info/entry_points.txt,sha256=Qg7ZRax4k-rcQsZ26XRYQ8YFSBiyY2PNxYfq4a6PYXI,41
-lmnr-0.2.15.dist-info/RECORD,,