vision-agent 0.2.14__py3-none-any.whl → 0.2.15__py3-none-any.whl

vision_agent/agent/execution.py

@@ -1,287 +0,0 @@
-"""This code is based off of code from CodeGeeX https://github.com/THUDM/CodeGeeX"""
-
-import contextlib
-import faulthandler
-import io
-import multiprocessing
-import os
-import platform
-import signal
-import tempfile
-import traceback
-import typing
-from pathlib import Path
-from typing import Dict, Generator, List, Optional, Union
-
-IMPORT_HELPER = """
-import math
-import re
-import sys
-import copy
-import datetime
-import itertools
-import collections
-import heapq
-import statistics
-import functools
-import hashlib
-import numpy
-import numpy as np
-import string
-from typing import *
-from collections import *
-from vision_agent.tools.tools_v2 import *
-"""
-
-
-def unsafe_execute(code: str, timeout: float, result: List) -> None:
-    with create_tempdir() as dir:
-        code_path = Path(dir) / "code.py"
-        with open(code_path, "w") as f:
-            f.write(code)
-
-        # These system calls are needed when cleaning up tempdir.
-        import os
-        import shutil
-
-        rmtree = shutil.rmtree
-        rmdir = os.rmdir
-        chdir = os.chdir
-
-        # Disable functionalities that can make destructive changes to the test.
-        reliability_guard()
-
-        try:
-            with swallow_io() as s:
-                with time_limit(timeout):
-                    # WARNING
-                    # This program exists to execute untrusted model-generated code. Although
-                    # it is highly unlikely that model-generated code will do something overtly
-                    # malicious in response to this test suite, model-generated code may act
-                    # destructively due to a lack of model capability or alignment.
-                    # Users are strongly encouraged to sandbox this evaluation suite so that it
-                    # does not perform destructive actions on their host or network.
-                    # Once you have read this disclaimer and taken appropriate precautions,
-                    # uncomment the following line and proceed at your own risk:
-                    code = compile(code, code_path, "exec")  # type: ignore
-                    exec(code)
-            result.append({"output": s.getvalue(), "passed": True})
-        except TimeoutError:
-            result.append({"output": "Timed out", "passed": False})
-        except AssertionError:
-            result.append({"output": f"{traceback.format_exc()}", "passed": False})
-        except BaseException:
-            result.append({"output": f"{traceback.format_exc()}", "passed": False})
-
-        # Needed for cleaning up.
-        shutil.rmtree = rmtree
-        os.rmdir = rmdir
-        os.chdir = chdir
-
-        code_path.unlink()
-
-
-def check_correctness(
-    code: str,
-    timeout: float = 3.0,
-) -> Dict[str, Union[str, bool]]:
-    """Evaluates the functional correctness of a completion by running the test suite
-    provided in the problem.
-    """
-
-    manager = multiprocessing.Manager()
-    result = manager.list()
-
-    # p = multiprocessing.Process(target=unsafe_execute, args=(tmp_dir,))
-    p = multiprocessing.Process(target=unsafe_execute, args=(code, timeout, result))
-    p.start()
-    p.join(timeout=timeout + 1)
-    if p.is_alive():
-        p.kill()
-
-    if not result:
-        result.append("timed out")
-
-    return {
-        "code": code,
-        "result": result[0]["output"],
-        "passed": result[0]["passed"],
-    }
-
-
-# Copyright (c) OpenAI (https://openai.com)
-
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-# ============================================================================
-
-
-class redirect_stdin(contextlib._RedirectStream):
-    _stream = "stdin"
-
-
-@contextlib.contextmanager
-def chdir(root: str) -> Generator[None, None, None]:
-    if root == ".":
-        yield
-        return
-    cwd = os.getcwd()
-    os.chdir(root)
-    try:
-        yield
-    except BaseException as exc:
-        raise exc
-    finally:
-        os.chdir(cwd)
-
-
-class WriteOnlyStringIO(io.StringIO):
-    """StringIO that throws an exception when it's read from"""
-
-    def read(self, *args, **kwargs):  # type: ignore
-        raise IOError
-
-    def readline(self, *args, **kwargs):  # type: ignore
-        raise IOError
-
-    def readlines(self, *args, **kwargs):  # type: ignore
-        raise IOError
-
-    def readable(self, *args, **kwargs):  # type: ignore
-        """Returns True if the IO object can be read."""
-        return False
-
-
-@contextlib.contextmanager
-def create_tempdir() -> Generator[str, None, None]:
-    with tempfile.TemporaryDirectory() as dirname:
-        with chdir(dirname):
-            yield dirname
-
-
-@contextlib.contextmanager
-def swallow_io() -> Generator[WriteOnlyStringIO, None, None]:
-    stream = WriteOnlyStringIO()
-    with contextlib.redirect_stdout(stream):
-        with contextlib.redirect_stderr(stream):
-            with redirect_stdin(stream):
-                yield stream
-
-
-@typing.no_type_check
-@contextlib.contextmanager
-def time_limit(seconds: float) -> Generator[None, None, None]:
-    def signal_handler(signum, frame):
-        raise TimeoutError("Timed out!")
-
-    if platform.uname().system != "Windows":
-        signal.setitimer(signal.ITIMER_REAL, seconds)
-        signal.signal(signal.SIGALRM, signal_handler)
-        try:
-            yield
-        finally:
-            signal.setitimer(signal.ITIMER_REAL, 0)
-
-
-@typing.no_type_check
-def reliability_guard(maximum_memory_bytes: Optional[int] = None) -> None:
-    """
-    This disables various destructive functions and prevents the generated code
-    from interfering with the test (e.g. fork bomb, killing other processes,
-    removing filesystem files, etc.)
-
-    WARNING
-    This function is NOT a security sandbox. Untrusted code, including, model-
-    generated code, should not be blindly executed outside of one. See the
-    Codex paper for more information about OpenAI's code sandbox, and proceed
-    with caution.
-    """
-
-    if maximum_memory_bytes is not None:
-        if platform.uname().system != "Windows":
-            import resource
-
-            resource.setrlimit(
-                resource.RLIMIT_AS, (maximum_memory_bytes, maximum_memory_bytes)
-            )
-            resource.setrlimit(
-                resource.RLIMIT_DATA, (maximum_memory_bytes, maximum_memory_bytes)
-            )
-            if not platform.uname().system == "Darwin":
-                resource.setrlimit(
-                    resource.RLIMIT_STACK, (maximum_memory_bytes, maximum_memory_bytes)
-                )
-
-    faulthandler.disable()
-
-    import builtins
-
-    builtins.exit = None
-    builtins.quit = None
-
-    import os
-
-    os.environ["OMP_NUM_THREADS"] = "1"
-
-    os.kill = None
-    os.system = None
-    # os.putenv = None  # this causes numpy to fail on import
-    os.remove = None
-    os.removedirs = None
-    os.rmdir = None
-    os.fchdir = None
-    os.setuid = None
-    os.fork = None
-    os.forkpty = None
-    os.killpg = None
-    os.rename = None
-    os.renames = None
-    os.truncate = None
-    os.replace = None
-    os.unlink = None
-    os.fchmod = None
-    os.fchown = None
-    os.chmod = None
-    os.chown = None
-    os.chroot = None
-    os.fchdir = None
-    os.lchflags = None
-    os.lchmod = None
-    os.lchown = None
-    os.getcwd = None
-    os.chdir = None
-
-    import shutil
-
-    shutil.rmtree = None
-    shutil.move = None
-    shutil.chown = None
-
-    import subprocess
-
-    subprocess.Popen = None  # type: ignore
-
-    __builtins__["help"] = None
-
-    import sys
-
-    sys.modules["ipdb"] = None
-    sys.modules["joblib"] = None
-    sys.modules["resource"] = None
-    sys.modules["psutil"] = None
-    sys.modules["tkinter"] = None

vision_agent-0.2.14.dist-info/RECORD

@@ -1,30 +0,0 @@
-vision_agent/__init__.py,sha256=GVLHCeK_R-zgldpbcPmOzJat-BkadvkuRCMxDvTIcXs,108
-vision_agent/agent/__init__.py,sha256=6AIN_lkqAz83dRN2WKpW7m1lQH1m-8BA2L-oUCFSKd4,163
-vision_agent/agent/agent.py,sha256=X7kON-g9ePUKumCDaYfQNBX_MEFE-ax5PnRp7-Cc5Wo,529
-vision_agent/agent/agent_coder.py,sha256=65ZEF3_K2aAG0vve-Q9xOPP9uw30_vVD2Li6NE6BFRY,6073
-vision_agent/agent/agent_coder_prompts.py,sha256=CJe3v7xvHQ32u3RQAXQga_Tk_4UgU64RBAMHZ3S70KY,5538
-vision_agent/agent/easytool.py,sha256=oMHnBg7YBtIPgqQUNcZgq7uMgpPThs99_UnO7ERkMVg,11511
-vision_agent/agent/easytool_prompts.py,sha256=Bikw-PPLkm78dwywTlnv32Y1Tw6JMeC-R7oCnXWLcTk,4656
-vision_agent/agent/execution.py,sha256=wX8LwXDq_0g_bTPikNiaW6nz5bUC7fUlNQsQHe_7Ww0,8582
-vision_agent/agent/reflexion.py,sha256=4gz30BuFMeGxSsTzoDV4p91yE0R8LISXp28IaOI6wdM,10506
-vision_agent/agent/reflexion_prompts.py,sha256=G7UAeNz_g2qCb2yN6OaIC7bQVUkda4m3z42EG8wAyfE,9342
-vision_agent/agent/vision_agent.py,sha256=5W5Xr_h4yDMsFvIk2JWcfMlYoPYmTv3JZnrDDumuZgM,26842
-vision_agent/agent/vision_agent_prompts.py,sha256=MZSIwovYgB-f-kdJ6btaNDVXptJn47bfOL3-Zn6NiC0,8573
-vision_agent/fonts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vision_agent/fonts/default_font_ch_en.ttf,sha256=1YM0Z3XqLDjSNbF7ihQFSAIUdjF9m1rtHiNC_6QosTE,1594400
-vision_agent/image_utils.py,sha256=1dggPBhW8_hUXDItCRLa23h-hdBwS50cjL4v1hsoUbg,7586
-vision_agent/llm/__init__.py,sha256=BoUm_zSAKnLlE8s-gKTSQugXDqVZKPqYlWwlTLdhcz4,48
-vision_agent/llm/llm.py,sha256=1BkrSVBWEClyqLc0Rmyw4heLhi_ZVm6JO7-i1wd1ziw,5383
-vision_agent/lmm/__init__.py,sha256=nnNeKD1k7q_4vLb1x51O_EUTYaBgGfeiCx5F433gr3M,67
-vision_agent/lmm/lmm.py,sha256=gK90vMxh0OcGSuIZQikBkDXm4pfkdFk1R2y7rtWDl84,10539
-vision_agent/tools/__init__.py,sha256=uWySwcIeQMH57PVN6lVIknTx-SFmN_J0mvn_HbGlXcQ,451
-vision_agent/tools/prompts.py,sha256=V1z4YJLXZuUl_iZ5rY0M5hHc_2tmMEUKr0WocXKGt4E,1430
-vision_agent/tools/tool_utils.py,sha256=kY-hBDIrapI-030nZuasXU83P6X3GR0Y_gOR32bnedw,747
-vision_agent/tools/tools.py,sha256=8JzNtn_uKTyc-bztjnaGCY7ctRnfW5dRS-ppxaP-1RE,46427
-vision_agent/tools/tools_v2.py,sha256=RxeaBTTkhqvATQGuYKiopeU4L2m0GbpPo-ypDmQ9UfY,5407
-vision_agent/tools/video.py,sha256=xTElFSFp1Jw4ulOMnk81Vxsh-9dTxcWUO6P9fzEi3AM,7653
-vision_agent/type_defs.py,sha256=4LTnTL4HNsfYqCrDn9Ppjg9bSG2ZGcoKSSd9YeQf4Bw,1792
-vision_agent-0.2.14.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-vision_agent-0.2.14.dist-info/METADATA,sha256=P4mOwafNaTHHf5GMoe-spAfPUmQHNWT7UKSBZ5LU-Vo,8997
-vision_agent-0.2.14.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-vision_agent-0.2.14.dist-info/RECORD,,