beamlit 0.0.20rc7__tar.gz → 0.0.20rc9__tar.gz

{beamlit-0.0.20rc7 → beamlit-0.0.20rc9}/PKG-INFO

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

{beamlit-0.0.20rc7 → beamlit-0.0.20rc9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beamlit"
-version = "0.0.20.pre-7"
+version = "0.0.20.pre-9"
 description = "Add your description here"
 readme = "README.md"
 authors = [

beamlit-0.0.20rc9/src/beamlit/functions/github/__init__.py

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

beamlit-0.0.20rc9/src/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-0.0.20rc9/src/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-0.0.20rc9/src/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-0.0.20rc9/src/beamlit/functions/math/__init__.py

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

beamlit-0.0.20rc9/src/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-0.0.20rc9/src/beamlit/functions/search/__init__.py

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

beamlit-0.0.20rc9/src/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 → beamlit-0.0.20rc9}/src/beamlit/serve/app.py

@@ -1,4 +1,6 @@
 import importlib
+import os
+import sys
 import traceback
 from logging import getLogger
 from uuid import uuid4
@@ -11,6 +13,9 @@ from beamlit.common.settings import get_settings, init
 
 from .middlewares import AccessLogMiddleware, AddProcessTimeHeader
 
+sys.path.insert(0, os.getcwd())
+sys.path.insert(0, os.path.join(os.getcwd(), "src"))
+
 
 def import_module():
     settings = get_settings()