beamlit 0.0.20rc7__py3-none-any.whl → 0.0.20rc8__py3-none-any.whl

beamlit/functions/github/__init__.py

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

beamlit/functions/github/github.py

@@ -0,0 +1,21 @@
+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)
+    return modes[name](*args)

beamlit/functions/github/kit/__init__.py

@@ -0,0 +1,7 @@
+"""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

@@ -0,0 +1,51 @@
+"""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

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

beamlit/functions/math/math.py

@@ -0,0 +1,40 @@
+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

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

beamlit/functions/search/search.py

@@ -0,0 +1,15 @@
+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

{beamlit-0.0.20rc7.dist-info → beamlit-0.0.20rc8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: beamlit
-Version: 0.0.20rc7
+Version: 0.0.20rc8
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12

{beamlit-0.0.20rc7.dist-info → beamlit-0.0.20rc8.dist-info}/RECORD

@@ -142,6 +142,14 @@ beamlit/common/settings.py,sha256=WF7_BztXwfHC1LY7DXtj7plYNFgYnCx_hk_wO1DUPJ8,51
 beamlit/common/utils.py,sha256=jouz5igBvT37Xn_e94-foCHyQczVim-UzVcoIF6RWJ4,657
 beamlit/functions/__init__.py,sha256=_RPG1Bfg54JGdIPnViAU6n9zD7E1cDNsdXi8oYGskzE,138
 beamlit/functions/decorator.py,sha256=-2newMBztweIgFuh0ABKOdxCfUzWaRxf0ym-YAgggJI,3168
+beamlit/functions/github/__init__.py,sha256=gYnUkeegukOfbymdabuuJkScvH-_ZJygX05BoqkPn0o,49
+beamlit/functions/github/github.py,sha256=FajzLCNkpXcwfgnC0l9rOGT2eSPLCz8-qrMzK9N_ZNc,598
+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/search/__init__.py,sha256=5NAthQ9PBwrkNg1FpLRx4flauvv0HyWuwaVS589c1Pw,49
+beamlit/functions/search/search.py,sha256=8s9ECltq7YE17j6rTxb12uY2EQY4_eTLHmwlIMThI0w,515
 beamlit/models/__init__.py,sha256=Jv0iTOtTs9G2zQRjbCmDjmNz8T9fX-MHoSt3b3zlRcI,8929
 beamlit/models/acl.py,sha256=MmwPgwhydRBkO1EyqZkNn7jX57HUT2a4Y06iRX91mxI,4642
 beamlit/models/agent.py,sha256=X5plQzNDoLcB2_4HoiuZ6ciw9jsalypfuehdR75SWBM,4829
@@ -288,6 +296,6 @@ beamlit/serve/app.py,sha256=riZWmczexN5t-PT172WxB5YZkoORDEl0ucS-2vxdWHg,2292
 beamlit/serve/middlewares/__init__.py,sha256=1dVmnOmhAQWvWktqHkKSIX-YoF6fmMU8xkUQuhg_rJU,148
 beamlit/serve/middlewares/accesslog.py,sha256=wM52-hcwtO-_hdM1pnsEJzerzJf1MzEyN5m85BdDccE,609
 beamlit/serve/middlewares/processtime.py,sha256=lDAaIasZ4bwvN-HKHvZpaD9r-yrkVNZYx4abvbjbrCg,411
-beamlit-0.0.20rc7.dist-info/METADATA,sha256=ib3u7kmMzztuEi5w1_vsiZ59yPJGVsZh4_KUTfZImCM,2026
-beamlit-0.0.20rc7.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-beamlit-0.0.20rc7.dist-info/RECORD,,
+beamlit-0.0.20rc8.dist-info/METADATA,sha256=l6r_tFygyBqlBCuyazZ3V0C47kxXMsLFk1x00Nd1ZVc,2026
+beamlit-0.0.20rc8.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+beamlit-0.0.20rc8.dist-info/RECORD,,