246 lines
6.4 KiB
JavaScript
246 lines
6.4 KiB
JavaScript
|
// eslint-disable-next-line import/no-extraneous-dependencies
|
||
|
|
import { loadPyodide } from "pyodide";
|
||
|
|
import { Tool } from "@langchain/core/tools";
|
||
|
|
export class PythonInterpreterTool extends Tool {
|
||
|
|
static lc_name() {
|
||
|
|
return "PythonInterpreterTool";
|
||
|
|
}
|
||
|
|
constructor(options) {
|
||
|
|
super(options);
|
||
|
|
Object.defineProperty(this, "name", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: "python_interpreter"
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "description", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: `Evaluates python code in a sandbox environment. The environment resets on every execution. You must send the whole script every time and print your outputs. Script should be pure python code that can be evaluated. Packages available:
|
||
|
|
${this.availableDefaultPackages}`
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "pyodideInstance", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: void 0
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "stdout", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: ""
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "stderr", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: ""
|
||
|
|
});
|
||
|
|
this.pyodideInstance = options.instance;
|
||
|
|
this.pyodideInstance.setStderr({
|
||
|
|
batched: (text) => {
|
||
|
|
this.stderr += text;
|
||
|
|
},
|
||
|
|
});
|
||
|
|
this.pyodideInstance.setStdout({
|
||
|
|
batched: (text) => {
|
||
|
|
this.stdout += text;
|
||
|
|
},
|
||
|
|
});
|
||
|
|
}
|
||
|
|
async addPackage(packageName) {
|
||
|
|
await this.pyodideInstance.loadPackage(packageName);
|
||
|
|
this.description += `, ${packageName}`;
|
||
|
|
}
|
||
|
|
get availableDefaultPackages() {
|
||
|
|
return [
|
||
|
|
"asciitree",
|
||
|
|
"astropy",
|
||
|
|
"atomicwrites",
|
||
|
|
"attrs",
|
||
|
|
"autograd",
|
||
|
|
"awkward-cpp",
|
||
|
|
"bcrypt",
|
||
|
|
"beautifulsoup4",
|
||
|
|
"biopython",
|
||
|
|
"bitarray",
|
||
|
|
"bitstring",
|
||
|
|
"bleach",
|
||
|
|
"bokeh",
|
||
|
|
"boost-histogram",
|
||
|
|
"brotli",
|
||
|
|
"cachetools",
|
||
|
|
"Cartopy",
|
||
|
|
"cbor-diag",
|
||
|
|
"certifi",
|
||
|
|
"cffi",
|
||
|
|
"cffi_example",
|
||
|
|
"cftime",
|
||
|
|
"click",
|
||
|
|
"cligj",
|
||
|
|
"cloudpickle",
|
||
|
|
"cmyt",
|
||
|
|
"colorspacious",
|
||
|
|
"contourpy",
|
||
|
|
"coolprop",
|
||
|
|
"coverage",
|
||
|
|
"cramjam",
|
||
|
|
"cryptography",
|
||
|
|
"cssselect",
|
||
|
|
"cycler",
|
||
|
|
"cytoolz",
|
||
|
|
"decorator",
|
||
|
|
"demes",
|
||
|
|
"deprecation",
|
||
|
|
"distlib",
|
||
|
|
"docutils",
|
||
|
|
"exceptiongroup",
|
||
|
|
"fastparquet",
|
||
|
|
"fiona",
|
||
|
|
"fonttools",
|
||
|
|
"freesasa",
|
||
|
|
"fsspec",
|
||
|
|
"future",
|
||
|
|
"galpy",
|
||
|
|
"gensim",
|
||
|
|
"geopandas",
|
||
|
|
"gmpy2",
|
||
|
|
"gsw",
|
||
|
|
"h5py",
|
||
|
|
"html5lib",
|
||
|
|
"idna",
|
||
|
|
"igraph",
|
||
|
|
"imageio",
|
||
|
|
"iniconfig",
|
||
|
|
"jedi",
|
||
|
|
"Jinja2",
|
||
|
|
"joblib",
|
||
|
|
"jsonschema",
|
||
|
|
"kiwisolver",
|
||
|
|
"lazy-object-proxy",
|
||
|
|
"lazy_loader",
|
||
|
|
"lightgbm",
|
||
|
|
"logbook",
|
||
|
|
"lxml",
|
||
|
|
"MarkupSafe",
|
||
|
|
"matplotlib",
|
||
|
|
"matplotlib-pyodide",
|
||
|
|
"micropip",
|
||
|
|
"mne",
|
||
|
|
"more-itertools",
|
||
|
|
"mpmath",
|
||
|
|
"msgpack",
|
||
|
|
"msprime",
|
||
|
|
"multidict",
|
||
|
|
"munch",
|
||
|
|
"mypy",
|
||
|
|
"netcdf4",
|
||
|
|
"networkx",
|
||
|
|
"newick",
|
||
|
|
"nlopt",
|
||
|
|
"nltk",
|
||
|
|
"nose",
|
||
|
|
"numcodecs",
|
||
|
|
"numpy",
|
||
|
|
"opencv-python",
|
||
|
|
"optlang",
|
||
|
|
"orjson",
|
||
|
|
"packaging",
|
||
|
|
"pandas",
|
||
|
|
"parso",
|
||
|
|
"patsy",
|
||
|
|
"peewee",
|
||
|
|
"Pillow",
|
||
|
|
"pillow_heif",
|
||
|
|
"pkgconfig",
|
||
|
|
"pluggy",
|
||
|
|
"protobuf",
|
||
|
|
"py",
|
||
|
|
"pyb2d",
|
||
|
|
"pyclipper",
|
||
|
|
"pycparser",
|
||
|
|
"pycryptodome",
|
||
|
|
"pydantic",
|
||
|
|
"pyerfa",
|
||
|
|
"Pygments",
|
||
|
|
"pyheif",
|
||
|
|
"pyinstrument",
|
||
|
|
"pynacl",
|
||
|
|
"pyodide-http",
|
||
|
|
"pyodide-tblib",
|
||
|
|
"pyparsing",
|
||
|
|
"pyproj",
|
||
|
|
"pyrsistent",
|
||
|
|
"pyshp",
|
||
|
|
"pytest",
|
||
|
|
"pytest-benchmark",
|
||
|
|
"python-dateutil",
|
||
|
|
"python-magic",
|
||
|
|
"python-sat",
|
||
|
|
"python_solvespace",
|
||
|
|
"pytz",
|
||
|
|
"pywavelets",
|
||
|
|
"pyxel",
|
||
|
|
"pyyaml",
|
||
|
|
"rebound",
|
||
|
|
"reboundx",
|
||
|
|
"regex",
|
||
|
|
"retrying",
|
||
|
|
"RobotRaconteur",
|
||
|
|
"ruamel.yaml",
|
||
|
|
"rust-panic-test",
|
||
|
|
"scikit-image",
|
||
|
|
"scikit-learn",
|
||
|
|
"scipy",
|
||
|
|
"screed",
|
||
|
|
"setuptools",
|
||
|
|
"shapely",
|
||
|
|
"simplejson",
|
||
|
|
"six",
|
||
|
|
"smart_open",
|
||
|
|
"soupsieve",
|
||
|
|
"sourmash",
|
||
|
|
"sparseqr",
|
||
|
|
"sqlalchemy",
|
||
|
|
"statsmodels",
|
||
|
|
"svgwrite",
|
||
|
|
"swiglpk",
|
||
|
|
"sympy",
|
||
|
|
"termcolor",
|
||
|
|
"texttable",
|
||
|
|
"threadpoolctl",
|
||
|
|
"tomli",
|
||
|
|
"tomli-w",
|
||
|
|
"toolz",
|
||
|
|
"tqdm",
|
||
|
|
"traits",
|
||
|
|
"tskit",
|
||
|
|
"typing-extensions",
|
||
|
|
"uncertainties",
|
||
|
|
"unyt",
|
||
|
|
"webencodings",
|
||
|
|
"wordcloud",
|
||
|
|
"wrapt",
|
||
|
|
"xarray",
|
||
|
|
"xgboost",
|
||
|
|
"xlrd",
|
||
|
|
"xyzservices",
|
||
|
|
"yarl",
|
||
|
|
"yt",
|
||
|
|
"zarr",
|
||
|
|
].join(", ");
|
||
|
|
}
|
||
|
|
static async initialize(options) {
|
||
|
|
const instance = await loadPyodide(options);
|
||
|
|
return new this({ ...options, instance });
|
||
|
|
}
|
||
|
|
async _call(script) {
|
||
|
|
this.stdout = "";
|
||
|
|
this.stderr = "";
|
||
|
|
await this.pyodideInstance.runPythonAsync(script);
|
||
|
|
return JSON.stringify({ stdout: this.stdout, stderr: this.stderr });
|
||
|
|
}
|
||
|
|
}
|