reflex 0.7.2.dev1__tar.gz → 0.7.2.post1__tar.gz

{reflex-0.7.2.dev1 → reflex-0.7.2.post1}/PKG-INFO

@@ -1,45 +1,51 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.3
 Name: reflex
-Version: 0.7.2.dev1
+Version: 0.7.2.post1
 Summary: Web apps in pure Python.
+License: Apache-2.0
 Keywords: web,framework
-Author: Elijah Ahianyo
-Author-Email: Nikhil Rao <nikhil@reflex.dev>, Alek Petuskey <alek@reflex.dev>, Masen Furer <masen@reflex.dev>, =?utf-8?q?Thomas_Brand=C3=A9ho?= <thomas@reflex.dev>, Khaleel Al-Adhami <khaleel@reflex.dev>
-Maintainer-Email: Masen Furer <masen@reflex.dev>, =?utf-8?q?Thomas_Brand=C3=A9ho?= <thomas@reflex.dev>, Khaleel Al-Adhami <khaleel@reflex.dev>
-License-Expression: Apache-2.0
-Project-URL: homepage, https://reflex.dev
-Project-URL: repository, https://github.com/reflex-dev/reflex
-Project-URL: documentation, https://reflex.dev/docs/getting-started/introduction
-Requires-Python: <4.0,>=3.10
-Requires-Dist: fastapi!=0.111.0,!=0.111.1,>=0.96.0
-Requires-Dist: gunicorn<24.0,>=20.1.0
-Requires-Dist: jinja2<4.0,>=3.1.2
-Requires-Dist: psutil<8.0,>=5.9.4
-Requires-Dist: pydantic<3.0,>=1.10.21
-Requires-Dist: python-multipart<0.1,>=0.0.5
-Requires-Dist: python-socketio<6.0,>=5.7.0
-Requires-Dist: redis<6.0,>=4.3.5
-Requires-Dist: rich<14.0,>=13.0.0
-Requires-Dist: sqlmodel<0.1,>=0.0.14
-Requires-Dist: typer<1.0,>=0.15.1
-Requires-Dist: uvicorn>=0.20.0
-Requires-Dist: starlette-admin<1.0,>=0.11.0
-Requires-Dist: alembic<2.0,>=1.11.1
-Requires-Dist: platformdirs<5.0,>=3.10.0
-Requires-Dist: distro<2.0,>=1.8.0; platform_system == "Linux"
-Requires-Dist: python-engineio!=4.6.0
-Requires-Dist: wrapt<2.0,>=1.17.0
-Requires-Dist: packaging<25.0,>=23.1
-Requires-Dist: reflex-hosting-cli>=0.1.29
-Requires-Dist: charset-normalizer<4.0,>=3.3.2
-Requires-Dist: wheel<1.0,>=0.42.0
-Requires-Dist: build<2.0,>=1.0.3
-Requires-Dist: setuptools>=75.0
-Requires-Dist: httpx<1.0,>=0.25.1
-Requires-Dist: twine<7.0,>=4.0.0
-Requires-Dist: tomlkit<1.0,>=0.12.4
-Requires-Dist: lazy_loader>=0.4
-Requires-Dist: typing_extensions>=4.6.0
+Author: Nikhil Rao
+Author-email: nikhil@reflex.dev
+Requires-Python: >=3.10,<4.0
+Classifier: Development Status :: 4 - Beta
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: alembic (>=1.11.1,<2.0)
+Requires-Dist: build (>=1.0.3,<2.0)
+Requires-Dist: charset-normalizer (>=3.3.2,<4.0)
+Requires-Dist: distro (>=1.8.0,<2.0) ; sys_platform == "linux"
+Requires-Dist: fastapi (>=0.96.0,!=0.111.0,!=0.111.1)
+Requires-Dist: gunicorn (>=20.1.0,<24.0)
+Requires-Dist: httpx (>=0.25.1,<1.0)
+Requires-Dist: jinja2 (>=3.1.2,<4.0)
+Requires-Dist: lazy_loader (>=0.4)
+Requires-Dist: packaging (>=23.1,<25.0)
+Requires-Dist: platformdirs (>=3.10.0,<5.0)
+Requires-Dist: psutil (>=5.9.4,<7.0)
+Requires-Dist: pydantic (>=1.10.21,<3.0)
+Requires-Dist: python-engineio (!=4.6.0)
+Requires-Dist: python-multipart (>=0.0.5,<0.1)
+Requires-Dist: python-socketio (>=5.7.0,<6.0)
+Requires-Dist: redis (>=4.3.5,<6.0)
+Requires-Dist: reflex-hosting-cli (>=0.1.29)
+Requires-Dist: rich (>=13.0.0,<14.0)
+Requires-Dist: setuptools (>=75.0)
+Requires-Dist: sqlmodel (>=0.0.14,<0.1)
+Requires-Dist: starlette-admin (>=0.11.0,<1.0)
+Requires-Dist: tomlkit (>=0.12.4,<1.0)
+Requires-Dist: twine (>=4.0.0,<7.0)
+Requires-Dist: typer (>=0.15.1,<1.0)
+Requires-Dist: typing_extensions (>=4.6.0)
+Requires-Dist: uvicorn (>=0.20.0)
+Requires-Dist: wheel (>=0.42.0,<1.0)
+Requires-Dist: wrapt (>=1.17.0,<2.0)
+Project-URL: Documentation, https://reflex.dev/docs/getting-started/introduction
+Project-URL: Homepage, https://reflex.dev
+Project-URL: Repository, https://github.com/reflex-dev/reflex
 Description-Content-Type: text/markdown
 
 
@@ -301,3 +307,4 @@ We are actively looking for contributors, no matter your skill level or experien
 ## License
 
 Reflex is open-source and licensed under the [Apache License 2.0](LICENSE).
+

reflex-0.7.2.post1/pyproject.toml

@@ -0,0 +1,137 @@
+[tool.poetry]
+name = "reflex"
+version = "0.7.2post1"
+description = "Web apps in pure Python."
+license = "Apache-2.0"
+authors = [
+  "Nikhil Rao <nikhil@reflex.dev>",
+  "Alek Petuskey <alek@reflex.dev>",
+  "Masen Furer <masen@reflex.dev>",
+  "Elijah Ahianyo <elijah@reflex.dev>",
+  "Thomas Brandého <thomas@reflex.dev>",
+]
+readme = "README.md"
+homepage = "https://reflex.dev"
+repository = "https://github.com/reflex-dev/reflex"
+documentation = "https://reflex.dev/docs/getting-started/introduction"
+keywords = ["web", "framework"]
+classifiers = ["Development Status :: 4 - Beta"]
+
+[tool.poetry.dependencies]
+python = ">=3.10, <4.0"
+fastapi = ">=0.96.0,!=0.111.0,!=0.111.1"
+gunicorn = ">=20.1.0,<24.0"
+jinja2 = ">=3.1.2,<4.0"
+psutil = ">=5.9.4,<7.0"
+pydantic = ">=1.10.21,<3.0"
+python-multipart = ">=0.0.5,<0.1"
+python-socketio = ">=5.7.0,<6.0"
+redis = ">=4.3.5,<6.0"
+rich = ">=13.0.0,<14.0"
+sqlmodel = ">=0.0.14,<0.1"
+typer = ">=0.15.1,<1.0"
+uvicorn = ">=0.20.0"
+starlette-admin = ">=0.11.0,<1.0"
+alembic = ">=1.11.1,<2.0"
+platformdirs = ">=3.10.0,<5.0"
+distro = { version = ">=1.8.0,<2.0", platform = "linux" }
+python-engineio = "!=4.6.0"
+wrapt = ">=1.17.0,<2.0"
+packaging = ">=23.1,<25.0"
+reflex-hosting-cli = ">=0.1.29"
+charset-normalizer = ">=3.3.2,<4.0"
+wheel = ">=0.42.0,<1.0"
+build = ">=1.0.3,<2.0"
+setuptools = ">=75.0"
+httpx = ">=0.25.1,<1.0"
+twine = ">=4.0.0,<7.0"
+tomlkit = ">=0.12.4,<1.0"
+lazy_loader = ">=0.4"
+typing_extensions = ">=4.6.0"
+
+[tool.poetry.group.dev.dependencies]
+pytest = ">=7.1.2,<9.0"
+pytest-mock = ">=3.10.0,<4.0"
+pyright = ">=1.1.394, <1.2"
+darglint = ">=1.8.1,<2.0"
+dill = ">=0.3.8"
+toml = ">=0.10.2,<1.0"
+pytest-asyncio = ">=0.24.0"
+pytest-cov = ">=4.0.0,<7.0"
+ruff = "0.9.6"
+pandas = ">=2.1.1,<3.0"
+pillow = ">=10.0.0,<12.0"
+plotly = ">=5.13.0,<6.0"
+asynctest = ">=0.13.0,<1.0"
+pre-commit = ">=3.2.1"
+selenium = ">=4.11.0,<5.0"
+pytest-benchmark = ">=4.0.0,<6.0"
+playwright = ">=1.46.0"
+pytest-playwright = ">=0.5.1"
+pytest-codspeed = "^3.1.2"
+
+[tool.poetry.scripts]
+reflex = "reflex.reflex:cli"
+
+[build-system]
+requires = ["poetry-core>=1.5.1"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.pyright]
+reportIncompatibleMethodOverride = false
+
+[tool.ruff]
+target-version = "py310"
+output-format = "concise"
+lint.isort.split-on-trailing-comma = false
+lint.select = [
+  "ANN001",
+  "B",
+  "C4",
+  "D",
+  "E",
+  "ERA",
+  "F",
+  "FURB",
+  "I",
+  "N",
+  "PERF",
+  "PGH",
+  "PTH",
+  "RUF",
+  "SIM",
+  "T",
+  "TRY",
+  "W",
+]
+lint.ignore = [
+  "B008",
+  "D205",
+  "E501",
+  "F403",
+  "SIM115",
+  "RUF006",
+  "RUF008",
+  "RUF012",
+  "TRY0",
+]
+lint.pydocstyle.convention = "google"
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["F401"]
+"tests/*.py" = ["ANN001", "D100", "D103", "D104", "B018", "PERF", "T", "N"]
+"benchmarks/*.py" = ["ANN001", "D100", "D103", "D104", "B018", "PERF", "T", "N"]
+"reflex/.templates/*.py" = ["D100", "D103", "D104"]
+"*.pyi" = ["D301", "D415", "D417", "D418", "E742", "N", "PGH"]
+"pyi_generator.py" = ["N802"]
+"reflex/constants/*.py" = ["N"]
+"*/blank.py" = ["I001"]
+
+[tool.pytest.ini_options]
+filterwarnings = "ignore:fields may not start with an underscore:RuntimeWarning"
+asyncio_default_fixture_loop_scope = "function"
+asyncio_mode = "auto"
+
+[tool.codespell]
+skip = "docs/*,*.html,examples/*, *.pyi, poetry.lock"
+ignore-words-list = "te, TreeE"

{reflex-0.7.2.dev1 → reflex-0.7.2.post1}/reflex/reflex.py

@@ -13,7 +13,7 @@ from reflex import constants
 from reflex.config import environment, get_config
 from reflex.custom_components.custom_components import custom_components_cli
 from reflex.state import reset_disk_state_manager
-from reflex.utils import console, redir, telemetry
+from reflex.utils import console, telemetry
 
 # Disable typer+rich integration for help panels
 typer.core.rich = None  # pyright: ignore [reportPrivateImportUsage]
@@ -70,10 +70,6 @@ def _init(
     # Show system info
     exec.output_system_info()
 
-    if ai:
-        redir.reflex_build_redirect()
-        return
-
     # Validate the app name.
     app_name = prerequisites.validate_app_name(name)
     console.rule(f"[bold]Initializing {app_name}")
@@ -87,7 +83,7 @@ def _init(
     prerequisites.initialize_frontend_dependencies()
 
     # Initialize the app.
-    template = prerequisites.initialize_app(app_name, template)
+    template = prerequisites.initialize_app(app_name, template, ai)
 
     # Initialize the .gitignore.
     prerequisites.initialize_gitignore()

{reflex-0.7.2.dev1 → reflex-0.7.2.post1}/reflex/state.py

@@ -1396,6 +1396,29 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
         for substate in self.substates.values():
             substate.reset()
 
+    @classmethod
+    @functools.lru_cache
+    def _is_client_storage(cls, prop_name_or_field: str | ModelField) -> bool:
+        """Check if the var is a client storage var.
+
+        Args:
+            prop_name_or_field: The name of the var or the field itself.
+
+        Returns:
+            Whether the var is a client storage var.
+        """
+        if isinstance(prop_name_or_field, str):
+            field = cls.get_fields().get(prop_name_or_field)
+        else:
+            field = prop_name_or_field
+        return field is not None and (
+            isinstance(field.default, ClientStorageBase)
+            or (
+                isinstance(field.type_, type)
+                and issubclass(field.type_, ClientStorageBase)
+            )
+        )
+
     def _reset_client_storage(self):
         """Reset client storage base vars to their default values."""
         # Client-side storage is reset during hydrate so that clearing cookies
@@ -1403,10 +1426,7 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
         fields = self.get_fields()
         for prop_name in self.base_vars:
             field = fields[prop_name]
-            if isinstance(field.default, ClientStorageBase) or (
-                isinstance(field.type_, type)
-                and issubclass(field.type_, ClientStorageBase)
-            ):
+            if self._is_client_storage(field):
                 setattr(self, prop_name, copy.deepcopy(field.default))
 
         # Recursively reset the substate client storage.
@@ -2342,8 +2362,9 @@ class UpdateVarsInternalState(State):
         for var, value in vars.items():
             state_name, _, var_name = var.rpartition(".")
             var_state_cls = State.get_class_substate(state_name)
-            var_state = await self.get_state(var_state_cls)
-            setattr(var_state, var_name, value)
+            if var_state_cls._is_client_storage(var_name):
+                var_state = await self.get_state(var_state_cls)
+                setattr(var_state, var_name, value)
 
 
 class OnLoadInternalState(State):

{reflex-0.7.2.dev1 → reflex-0.7.2.post1}/reflex/utils/prerequisites.py

@@ -37,7 +37,7 @@ from redis.exceptions import RedisError
 from reflex import constants, model
 from reflex.compiler import templates
 from reflex.config import Config, environment, get_config
-from reflex.utils import console, net, path_ops, processes
+from reflex.utils import console, net, path_ops, processes, redir
 from reflex.utils.exceptions import (
     GeneratedCodeHasNoFunctionDefsError,
     SystemPackageMissingError,
@@ -1695,6 +1695,31 @@ def validate_and_create_app_using_remote_template(
     )
 
 
+def generate_template_using_ai(template: str | None = None) -> str:
+    """Generate a template using AI(Flexgen).
+
+    Args:
+        template: The name of the template.
+
+    Returns:
+        The generation hash.
+
+    Raises:
+        Exit: If the template and ai flags are used.
+    """
+    if template is None:
+        # If AI is requested and no template specified, redirect the user to reflex.build.
+        return redir.reflex_build_redirect()
+    elif is_generation_hash(template):
+        # Otherwise treat the template as a generation hash.
+        return template
+    else:
+        console.error(
+            "Cannot use `--template` option with `--ai` option. Please remove `--template` option."
+        )
+        raise typer.Exit(2)
+
+
 def fetch_remote_templates(
     template: str,
 ) -> tuple[str, dict[str, Template]]:
@@ -1719,12 +1744,15 @@ def fetch_remote_templates(
     return template, available_templates
 
 
-def initialize_app(app_name: str, template: str | None = None) -> str | None:
+def initialize_app(
+    app_name: str, template: str | None = None, ai: bool = False
+) -> str | None:
     """Initialize the app either from a remote template or a blank app. If the config file exists, it is considered as reinit.
 
     Args:
         app_name: The name of the app.
         template: The name of the template to use.
+        ai: Whether to use AI to generate the template.
 
     Returns:
         The name of the template.
@@ -1740,6 +1768,11 @@ def initialize_app(app_name: str, template: str | None = None) -> str | None:
         telemetry.send("reinit")
         return
 
+    generation_hash = None
+    if ai:
+        generation_hash = generate_template_using_ai(template)
+        template = constants.Templates.DEFAULT
+
     templates: dict[str, Template] = {}
 
     # Don't fetch app templates if the user directly asked for DEFAULT.
@@ -1748,7 +1781,11 @@ def initialize_app(app_name: str, template: str | None = None) -> str | None:
 
     if template is None:
         template = prompt_for_template_options(get_init_cli_prompt_options())
-        if template == constants.Templates.CHOOSE_TEMPLATES:
+        if template == constants.Templates.AI:
+            generation_hash = generate_template_using_ai()
+            # change to the default to allow creation of default app
+            template = constants.Templates.DEFAULT
+        elif template == constants.Templates.CHOOSE_TEMPLATES:
             console.print(
                 f"Go to the templates page ({constants.Templates.REFLEX_TEMPLATES_URL}) and copy the command to init with a template."
             )
@@ -1763,6 +1800,11 @@ def initialize_app(app_name: str, template: str | None = None) -> str | None:
             app_name=app_name, template=template, templates=templates
         )
 
+    # If a reflex.build generation hash is available, download the code and apply it to the main module.
+    if generation_hash:
+        initialize_main_module_index_from_generation(
+            app_name, generation_hash=generation_hash
+        )
     telemetry.send("init", template=template)
 
     return template

{reflex-0.7.2.dev1 → reflex-0.7.2.post1}/reflex/utils/pyi_generator.py

@@ -348,7 +348,7 @@ def _extract_class_props_as_ast_nodes(
     all_props = []
     kwargs = []
     for target_class in clzs:
-        event_triggers = target_class._create([]).get_event_triggers()
+        event_triggers = target_class().get_event_triggers()
         # Import from the target class to ensure type hints are resolvable.
         exec(f"from {target_class.__module__} import *", type_hint_globals)
         for name, value in target_class.__annotations__.items():
@@ -575,7 +575,7 @@ def _generate_component_create_functiondef(
             return ast.Name(id=f"{' | '.join(map(ast.unparse, all_count_args_type))}")
         return ast.Name(id="EventType[Any]")
 
-    event_triggers = clz._create([]).get_event_triggers()
+    event_triggers = clz().get_event_triggers()
 
     # event handler kwargs
     kwargs.extend(

{reflex-0.7.2.dev1 → reflex-0.7.2.post1}/reflex/utils/redir.py

@@ -1,6 +1,7 @@
 """Utilities to handle redirection to browser UI."""
 
 import time
+import uuid
 import webbrowser
 
 import httpx
@@ -47,6 +48,14 @@ def open_browser_and_wait(
     return response
 
 
-def reflex_build_redirect() -> None:
-    """Open the browser window to reflex.build."""
-    open_browser(constants.Templates.REFLEX_BUILD_FRONTEND)
+def reflex_build_redirect() -> str:
+    """Open the browser window to reflex.build and wait for the user to select a generation.
+
+    Returns:
+        The selected generation hash.
+    """
+    token = str(uuid.uuid4())
+    target_url = constants.Templates.REFLEX_BUILD_URL.format(reflex_init_token=token)
+    poll_url = constants.Templates.REFLEX_BUILD_POLL_URL.format(reflex_init_token=token)
+    response = open_browser_and_wait(target_url, poll_url)
+    return response.json()["generation_hash"]

{reflex-0.7.2.dev1 → reflex-0.7.2.post1}/reflex/vars/base.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 import contextlib
-import copy
 import dataclasses
 import datetime
 import functools
@@ -2147,7 +2146,7 @@ class ComputedVar(Var[RETURN_TYPE]):
             "fget": kwargs.pop("fget", self._fget),
             "initial_value": kwargs.pop("initial_value", self._initial_value),
             "cache": kwargs.pop("cache", self._cache),
-            "deps": kwargs.pop("deps", copy.copy(self._static_deps)),
+            "deps": kwargs.pop("deps", self._static_deps),
             "auto_deps": kwargs.pop("auto_deps", self._auto_deps),
             "interval": kwargs.pop("interval", self._update_interval),
             "backend": kwargs.pop("backend", self._backend),
@@ -2319,7 +2318,7 @@ class ComputedVar(Var[RETURN_TYPE]):
         if not _isinstance(value, self._var_type, nested=1, treat_var_as_type=False):
             console.error(
                 f"Computed var '{type(instance).__name__}.{self._js_expr}' must return"
-                f" a value of type '{self._var_type}', got '{value}' of type {type(value)}."
+                f" type '{self._var_type}', got '{type(value)}'."
             )
 
     def _deps(

reflex-0.7.2.dev1/benchmarks/__init__.py

@@ -1,3 +0,0 @@
-"""Reflex benchmarks."""
-
-WINDOWS_SKIP_REASON = "Takes too much time as a result of npm"

reflex-0.7.2.dev1/benchmarks/benchmark_compile_times.py

@@ -1,147 +0,0 @@
-"""Extracts the compile times from the JSON files in the specified directory and inserts them into the database."""
-
-from __future__ import annotations
-
-import argparse
-import json
-import os
-from pathlib import Path
-
-from utils import send_data_to_posthog
-
-
-def extract_stats_from_json(json_file: str) -> list[dict]:
-    """Extracts the stats from the JSON data and returns them as a list of dictionaries.
-
-    Args:
-        json_file: The JSON file to extract the stats data from.
-
-    Returns:
-        list[dict]: The stats for each test.
-    """
-    with Path(json_file).open() as file:
-        json_data = json.load(file)
-
-    # Load the JSON data if it is a string, otherwise assume it's already a dictionary
-    data = json.loads(json_data) if isinstance(json_data, str) else json_data
-
-    # Initialize an empty list to store the stats for each test
-    test_stats = []
-
-    # Iterate over each test in the 'benchmarks' list
-    for test in data.get("benchmarks", []):
-        group = test.get("group", None)
-        stats = test.get("stats", {})
-        full_name = test.get("fullname")
-        file_name = (
-            full_name.split("/")[-1].split("::")[0].strip(".py") if full_name else None
-        )
-        test_name = test.get("name", "Unknown Test")
-
-        test_stats.append(
-            {
-                "test_name": test_name,
-                "group": group,
-                "stats": stats,
-                "full_name": full_name,
-                "file_name": file_name,
-            }
-        )
-    return test_stats
-
-
-def insert_benchmarking_data(
-    os_type_version: str,
-    python_version: str,
-    performance_data: list[dict],
-    commit_sha: str,
-    pr_title: str,
-    branch_name: str,
-    event_type: str,
-    pr_id: str,
-):
-    """Insert the benchmarking data into the database.
-
-    Args:
-        os_type_version: The OS type and version to insert.
-        python_version: The Python version to insert.
-        performance_data: The performance data of reflex web to insert.
-        commit_sha: The commit SHA to insert.
-        pr_title: The PR title to insert.
-        branch_name: The name of the branch.
-        event_type: Type of github event(push, pull request, etc).
-        pr_id: Id of the PR.
-    """
-    # Prepare the event data
-    properties = {
-        "os": os_type_version,
-        "python_version": python_version,
-        "distinct_id": commit_sha,
-        "pr_title": pr_title,
-        "branch_name": branch_name,
-        "event_type": event_type,
-        "performance": performance_data,
-        "pr_id": pr_id,
-    }
-
-    send_data_to_posthog("simple_app_benchmark", properties)
-
-
-def main():
-    """Runs the benchmarks and inserts the results."""
-    # Get the commit SHA and JSON directory from the command line arguments
-    parser = argparse.ArgumentParser(description="Run benchmarks and process results.")
-    parser.add_argument(
-        "--os", help="The OS type and version to insert into the database."
-    )
-    parser.add_argument(
-        "--python-version", help="The Python version to insert into the database."
-    )
-    parser.add_argument(
-        "--commit-sha", help="The commit SHA to insert into the database."
-    )
-    parser.add_argument(
-        "--benchmark-json",
-        help="The JSON file containing the benchmark results.",
-    )
-    parser.add_argument(
-        "--pr-title",
-        help="The PR title to insert into the database.",
-    )
-    parser.add_argument(
-        "--branch-name",
-        help="The current branch",
-        required=True,
-    )
-    parser.add_argument(
-        "--event-type",
-        help="The github event type",
-        required=True,
-    )
-    parser.add_argument(
-        "--pr-id",
-        help="ID of the PR.",
-        required=True,
-    )
-    args = parser.parse_args()
-
-    # Get the PR title from env or the args. For the PR merge or push event, there is no PR title, leaving it empty.
-    pr_title = args.pr_title or os.getenv("PR_TITLE", "")
-
-    # Get the results of pytest benchmarks
-    cleaned_benchmark_results = extract_stats_from_json(args.benchmark_json)
-    # Insert the data into the database
-    insert_benchmarking_data(
-        os_type_version=args.os,
-        python_version=args.python_version,
-        performance_data=cleaned_benchmark_results,
-        commit_sha=args.commit_sha,
-        pr_title=pr_title,
-        branch_name=args.branch_name,
-        event_type=args.event_type,
-        pr_id=args.pr_id,
-    )
-
-
-if __name__ == "__main__":
-    main()