beamlit 0.0.40rc84__py3-none-any.whl → 0.0.41rc85__py3-none-any.whl

beamlit/functions/remote/remote.py

@@ -5,14 +5,14 @@ from typing import Callable
 
 import pydantic
 import typing_extensions as t
-from langchain_core.tools.base import BaseTool, ToolException
-
-from beamlit.api.functions import get_function
+from beamlit.api.functions import get_function, list_functions
 from beamlit.authentication.authentication import AuthenticatedClient
 from beamlit.common.settings import get_settings
+from beamlit.errors import UnexpectedStatus
 from beamlit.functions.mcp.mcp import MCPClient, MCPToolkit
 from beamlit.models import Function, StoreFunctionParameter
 from beamlit.run import RunClient
+from langchain_core.tools.base import BaseTool, ToolException
 
 
 def create_dynamic_schema(name: str, parameters: list[StoreFunctionParameter]) -> type[pydantic.BaseModel]:
@@ -88,7 +88,22 @@ class RemoteToolkit:
     def initialize(self) -> None:
         """Initialize the session and retrieve tools list"""
         if self._function is None:
-            self._function = get_function.sync_detailed(self.function, client=self.client).parsed
+            try:
+                response = get_function.sync_detailed(self.function, client=self.client)
+                self._function = response.parsed
+            except UnexpectedStatus as e:
+                settings = get_settings()
+                functions = list_functions.sync_detailed(
+                    client=self.client,
+                    environment=settings.environment,
+                ).parsed
+                names = [
+                    f.metadata.name
+                    for f in functions
+                ]
+                raise RuntimeError(
+                    f"error: {e.status_code}. Available functions: {', '.join(names)}"
+                )
 
     @t.override
     def get_tools(self) -> list[BaseTool]:

{beamlit-0.0.40rc84.dist-info → beamlit-0.0.41rc85.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.40rc84
+Version: 0.0.41rc85
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12

{beamlit-0.0.40rc84.dist-info → beamlit-0.0.41rc85.dist-info}/RECORD

@@ -144,16 +144,8 @@ beamlit/deploy/format.py,sha256=U6UZEFAYLnGJJ7O2YmSdlUUFhnWNGAv6NZ-DW4KTgvI,2049
 beamlit/deploy/parser.py,sha256=Ga0poCZkoRnuTw082QnTcNGCBJncoRAnVsn8-1FsaJE,6907
 beamlit/functions/__init__.py,sha256=NcQPZZNfWhAJ1T1F6Xn21LFPMbZ7aMR2Sve3uZOkBCQ,170
 beamlit/functions/decorator.py,sha256=99oHbss0RZ6Km7Gf62otun1az9waZM0GPaOJvvwbmOI,8541
-beamlit/functions/github/__init__.py,sha256=gYnUkeegukOfbymdabuuJkScvH-_ZJygX05BoqkPn0o,49
-beamlit/functions/github/github.py,sha256=RfzWMT9nwPjsCOJr_-emN7eJHCIt2WsvkT9tq8-HfeA,568
-beamlit/functions/github/kit/__init__.py,sha256=jBwPqZv6C23_utukohxqXZwrlicNlI7PYPUj0Den7Cw,136
-beamlit/functions/github/kit/pull_request.py,sha256=wQVeRBakiqu-2ouflO8p1z7D5u07KNsitwyNRrp0KjM,1357
-beamlit/functions/math/__init__.py,sha256=wie4WME8jT-WpFRrtu-lDlHW31Mg6K2cwstjkUdLF3o,43
-beamlit/functions/math/math.py,sha256=CpoLJGwuvwCPGnVC8k9GYuIyvfUYPDQHKlZg3cx-z-A,1049
 beamlit/functions/mcp/mcp.py,sha256=c7OpmC_zIx1fc2lwoPiyJk_yMyRxWiwCpy9hvRV_C0Y,4080
-beamlit/functions/remote/remote.py,sha256=NldBFEostIX7yLw8ZE12MgVqhVpvKB4Z6FiQUBhPFsc,4207
-beamlit/functions/search/__init__.py,sha256=5NAthQ9PBwrkNg1FpLRx4flauvv0HyWuwaVS589c1Pw,49
-beamlit/functions/search/search.py,sha256=8s9ECltq7YE17j6rTxb12uY2EQY4_eTLHmwlIMThI0w,515
+beamlit/functions/remote/remote.py,sha256=8dob5SbKaBTvPLZw-6yx955vfnbrjg048QIbW8xQDvA,4844
 beamlit/models/__init__.py,sha256=DB2EBTVX5lTO7CzXvZaqiPtgc9n8SMPu59zGD5_Vvno,9652
 beamlit/models/acl.py,sha256=tH67gsl_BMaviSbTaaIkO1g9cWZgJ6VgAnYVjQSzGZY,3952
 beamlit/models/agent.py,sha256=w2f_d_nLGVHW7blklH34Nmm2TI3MCoEiQN3HMVga5XI,4002
@@ -285,6 +277,6 @@ beamlit/serve/app.py,sha256=ROS_tb9cO4GvOQKCwloyAzpYraTdIb3oG6sChXikeNw,3285
 beamlit/serve/middlewares/__init__.py,sha256=1dVmnOmhAQWvWktqHkKSIX-YoF6fmMU8xkUQuhg_rJU,148
 beamlit/serve/middlewares/accesslog.py,sha256=Mu4T4_9OvHybjA0ApzZFpgi2C8f3X1NbUk-76v634XM,631
 beamlit/serve/middlewares/processtime.py,sha256=lDAaIasZ4bwvN-HKHvZpaD9r-yrkVNZYx4abvbjbrCg,411
-beamlit-0.0.40rc84.dist-info/METADATA,sha256=IIgQeS831MisY-EbiNZqA54a05M-Cb3h_2LT3dxqT7o,3715
-beamlit-0.0.40rc84.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.40rc84.dist-info/RECORD,,
+beamlit-0.0.41rc85.dist-info/METADATA,sha256=uF6h-oK1OD8K0CafX1yGoM3gcQ1jCn1e5QyY_k88nxA,3715
+beamlit-0.0.41rc85.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.41rc85.dist-info/RECORD,,

beamlit/functions/github/__init__.py

@@ -1,3 +0,0 @@
-from .github import github
-
-__all__ = ["github"]

beamlit/functions/github/github.py

@@ -1,20 +0,0 @@
-from beamlit.common.secrets import Secret
-
-from . import kit
-
-
-def github(name: str, *args):
-    """This function kit is used to perform actions on Github."""
-    github_token = Secret.get("GITHUB_TOKEN")
-    if not github_token:
-        raise ValueError("github_token missing from configuration.")
-
-    modes = {}
-
-    for func_name in dir(kit):
-        if not func_name.startswith("_"):
-            modes[func_name] = getattr(kit, func_name)
-    if name not in modes:
-        msg = f"Invalid mode: {name}"
-        raise ValueError(msg)
-    return modes[name](*args)

beamlit/functions/github/kit/__init__.py

@@ -1,7 +0,0 @@
-"""Kit for interacting with GitHub."""
-
-from .pull_request import list_open_pull_requests
-
-__all__ = [
-    "list_open_pull_requests",
-]

beamlit/functions/github/kit/pull_request.py

@@ -1,51 +0,0 @@
-"""Functions for interacting with GitHub pull requests."""
-
-from typing import Any
-
-from github import Auth, Github, PullRequest
-from pydash import pick
-
-from beamlit.common.secrets import Secret
-
-
-def list_open_pull_requests(
-    repository: str,
-):
-    """
-    This function will fetch a list of the repository's Pull Requests (PRs).
-    It will return the title, and PR number of 5 PRs.
-    """
-    auth = Auth.Token(Secret.get("GITHUB_TOKEN"))
-    gh = Github(auth=auth)
-    repo = gh.get_repo(repository)
-    return [_format_pull_request(pr) for pr in repo.get_pulls(state="open")[:5]]
-
-
-def _format_pull_request(pr: PullRequest) -> dict[str, Any]:
-    raw_data = pr.raw_data
-    raw_data["reviewers"] = [reviewer["login"] for reviewer in raw_data["requested_reviewers"]]
-    raw_data["assignees"] = [assignee["login"] for assignee in raw_data["assignees"]]
-
-    return pick(
-        raw_data,
-        [
-            "id",
-            "title",
-            "labels",
-            "number",
-            "html_url",
-            "diff_url",
-            "patch_url",
-            "commits",
-            "additions",
-            "deletions",
-            "changed_files",
-            "comments",
-            "state",
-            "user.login",
-            "assignees",
-            "reviewers",
-            "created_at",
-            "updated_at",
-        ],
-    )

beamlit/functions/math/__init__.py

@@ -1,3 +0,0 @@
-from .math import math
-
-__all__ = ["math"]

beamlit/functions/math/math.py

@@ -1,40 +0,0 @@
-import math as math_operation
-import operator
-
-
-def math(query: str):
-    """A function for performing mathematical calculations.."""
-    safe_dict = {
-        "abs": abs,
-        "round": round,
-        "min": min,
-        "max": max,
-        "pow": math_operation.pow,
-        "sqrt": math_operation.sqrt,
-        "sin": math_operation.sin,
-        "cos": math_operation.cos,
-        "tan": math_operation.tan,
-        "pi": math_operation.pi,
-        "e": math_operation.e,
-    }
-
-    # Add basic arithmetic operators
-    safe_dict.update(
-        {
-            "+": operator.add,
-            "-": operator.sub,
-            "*": operator.mul,
-            "/": operator.truediv,
-            "**": operator.pow,
-            "%": operator.mod,
-        }
-    )
-
-    try:
-        # Replace 'x' with '*'
-        query = query.replace("x", "*")
-
-        # Evaluate the expression in a restricted environment
-        return eval(query, {"__builtins__": {}}, safe_dict)
-    except Exception as e:
-        raise ValueError(f"Invalid expression: {str(e)}")

beamlit/functions/search/__init__.py

@@ -1,3 +0,0 @@
-from .search import search
-
-__all__ = ["search"]

beamlit/functions/search/search.py

@@ -1,15 +0,0 @@
-from langchain_community.tools.tavily_search.tool import TavilySearchResults
-
-from beamlit.common.secrets import Secret
-
-
-def search(query: str):
-    """
-    A search engine optimized for comprehensive, accurate, and trusted results.
-    Useful for when you need to answer questions about current events.
-    Input should be a search query.
-    """
-    api_key = Secret.get("TAVILY_API_KEY")
-    tavily = TavilySearchResults(api_key=api_key, max_results=2)
-    result = tavily.invoke(input=query)
-    return result