ahuang11's picture
Upload 3 files
dcc5dd4
raw
history blame
23.4 kB
OPENAI_MODELS = [
"gpt-4-1106-preview",
"gpt-4",
"gpt-3.5-turbo",
"gpt-3.5-turbo-16k",
]
ARGUMENTS = [
{
"name": "k_value",
"nickname": "k",
"help_text": "Number of chunks to return",
"type": int,
"default": 15,
},
{
"name": "libraries",
"nickname": "l",
"help_text": "Limit your chat to a list of libraries. Usage: -l library1 library2 library3",
"type": list,
},
{
"name": "model",
"nickname": "m",
"help_text": "Specify the model. Default: gpt-4",
"type": str,
"default": "gpt-4"
},
{
"name": "cite_sources",
"nickname": "c",
"help_text": "Determines whether or not the AI model cites its sources",
"type": bool,
"default": True,
},
{
"name": "local",
"nickname": "n",
"help_text": "Uses LMStudio for local models",
"type": bool,
"default": False,
},
{
"name": "context_window",
"nickname": "w",
"help_text": "Context window (if using local models)",
"type": int,
"default": 3000,
},
]
LIBRARIES = [
"python",
"boto3",
"urllib3",
"botocore",
"setuptools",
"requests",
"typing-extensions",
"certifi",
"charset-normalizer",
"wheel",
"cryptography",
"python-dateutil",
"idna",
"pyyaml",
"google-api-core",
"six",
"pytz",
"numpy",
"importlib-metadata",
"pip",
"packaging",
"zipp",
"awscli",
"aiobotocore",
"protobuf",
"click",
"pandas",
"pyasn1",
"rsa",
"fsspec",
"pyjwt",
"jmespath",
"markupsafe",
"s3fs",
"attrs",
"cffi",
"psutil",
"lxml",
"pydantic",
"colorama",
"platformdirs",
"googleapis-common-protos",
"pycparser",
"google-auth",
"pyopenssl",
"virtualenv",
"cachetools",
"werkzeug",
"jinja2",
"jsonschema",
"filelock",
"flask",
"sqlalchemy",
"pyparsing",
"docutils",
"async-timeout",
"tzlocal",
"oauthlib",
"pluggy",
"tomli",
"aiohttp",
"grpcio",
"requests-oauthlib",
"pyarrow",
"pytest",
"wrapt",
"tqdm",
"soupsieve",
"dnspython",
"isodate",
"azure-core",
"frozenlist",
"coverage",
"pygments",
"websocket-client",
"beautifulsoup4",
"pillow",
"greenlet",
"importlib-resources",
"distlib",
"yarl",
"multidict",
"scipy",
"decorator",
"aiofiles",
"et-xmlfile",
"openpyxl",
"google-cloud-storage",
"google-cloud-core",
"httptools",
"chardet",
"iniconfig",
"asn1crypto",
"tomlkit",
"tabulate",
"more-itertools",
"requests-toolbelt",
"google-resumable-media",
"paramiko",
"aioconsole",
"deprecated",
"gitpython",
"pynacl",
"google-api-python-client",
"pymysql",
"psycopg2",
"rpds-py",
"proto-plus",
"anyio",
"itsdangerous",
"msal",
"referencing",
"azure-storage-blob",
"jsonschema-specifications",
"bcrypt",
"pathspec",
"scikit-learn",
"smmap",
"msgpack",
"matplotlib",
"poetry-core",
"keyring",
"joblib",
"regex",
"mypy-extensions",
"wcwidth",
"docker",
"sniffio",
"google-auth-oauthlib",
"kiwisolver",
"portalocker",
"pexpect",
"ptyprocess",
"jaraco-classes",
"dill",
"pyrsistent",
"ruamel-yaml",
"gitdb",
"pycryptodomex",
"sqlparse",
"msrest",
"google-crc32c",
"sagemaker",
"tenacity",
"prompt-toolkit",
"google-cloud-bigquery",
"tzdata",
"snowflake-connector-python",
"gunicorn",
"cython",
"py4j",
"py",
"markdown",
"azure-identity",
"httplib2",
"future",
"fonttools",
"alembic",
"markdown-it-py",
"cachecontrol",
"awswrangler",
"rich",
"msal-extensions",
"tornado",
"threadpoolctl",
"jedi",
"marshmallow",
"google-auth-httplib2",
"traitlets",
"cloudpickle",
"shellingham",
"redis",
"pycodestyle",
"backoff",
"python-dotenv",
"scramp",
"toml",
"h11",
"pytest-cov",
"termcolor",
"trove-classifiers",
"annotated-types",
"uritemplate",
"ipython",
"pyzmq",
"networkx",
"xmltodict",
"uvicorn",
"pyspark",
"pg8000",
"mccabe",
"ply",
"prometheus-client",
"prometheus-python",
"redshift-connector",
"oscrypto",
"dulwich",
"webencodings",
"pyodbc",
"pycryptodome",
"httpx",
"sortedcontainers",
"httpcore",
"jeepney",
"mako",
"babel",
"poetry",
"secretstorage",
"defusedxml",
"isort",
"jsonpointer",
"blinker",
"black",
"jupyter-client",
"typing-inspect",
"jupyter-core",
"pymongo",
"mdit-py-plugins",
"datadog",
"contourpy",
"adal",
"pkginfo",
"parso",
"tensorboard",
"toolz",
"pyflakes",
"absl-py",
"sentry-sdk",
"xlrd",
"requests-aws4auth",
"flake8",
"jsonpath-ng",
"python-json-logger",
"nbconvert",
"pickleshare",
"build",
"mdurl",
"backcall",
"fastapi",
"rapidfuzz",
"argcomplete",
"python-utils",
"transformers",
"matplotlib-inline",
"setuptools-scm",
"nbformat",
"ipykernel",
"databricks-cli",
"notebook",
"fastjsonschema",
"jupyter-server",
"mistune",
"huggingface-hub",
"kubernetes",
"debugpy",
"starlette",
"arrow",
"asttokens",
"progressbar2",
"tensorflow",
"google-cloud-pubsub",
"websockets",
"astroid",
"jsonpatch",
"asynctest",
"aioitertools",
"imageio",
"simplejson",
"appdirs",
"pyproject-hooks",
"pylint",
"pbr",
"lazy-object-proxy",
"multiprocess",
"smart-open",
"altair",
"h5py",
"asgiref",
"backports-zoneinfo",
"tinycss2",
"entrypoints",
"bleach",
"oauth2client",
"llvmlite",
"numba",
"cattrs",
"crashtest",
"mlflow",
"send2trash",
"shapely",
"elasticsearch",
"comm",
"cleo",
"orjson",
"pendulum",
"pytest-runner",
"nbclient",
"aenum",
"pygithub",
"identify",
"msrestazure",
"nodeenv",
"mypy",
"flatbuffers",
"great-expectations",
"mock",
"jupyterlab-server",
"zope-interface",
"pytzdata",
"loguru",
"argon2-cffi",
"tokenizers",
"typeguard",
"overrides",
"tox",
"requests-file",
"humanfriendly",
"json5",
"xlsxwriter",
"pysocks",
"google-pasta",
"cfgv",
"pyathena",
"gast",
"azure-storage-file-datalake",
"ipywidgets",
"rfc3339-validator",
"executing",
"jupyterlab",
"pre-commit",
"django",
"querystring-parser",
"contextlib2",
"cached-property",
"installer",
"deepdiff",
"pure-eval",
"tensorflow-serving-api",
"nltk",
"semver",
"retry",
"hvac",
"pipenv",
"uri-template",
"torch",
"execnet",
"html5lib",
"typer",
"croniter",
"lockfile",
"slack-sdk",
"watchdog",
"dataclasses",
"gremlinpython",
"types-pyyaml",
"tensorflow-io-gcs-filesystem",
"setproctitle",
"azure-mgmt-core",
"responses",
"sphinx",
"statsmodels",
"text-unidecode",
"dataclasses-json",
"pandocfilters",
"pytest-xdist",
"async-lru",
"click-plugins",
"opentelemetry-api",
"selenium",
"safetensors",
"opencv-python",
"python-slugify",
"xgboost",
"distro",
"plotly",
"sentencepiece",
"webcolors",
"types-requests",
"rfc3986",
"terminado",
"jupyter-lsp",
"rfc3986-validator",
"configparser",
"argon2-cffi-bindings",
"cmake",
"fastavro",
"docopt",
"unidecode",
"retrying",
"types-urllib3",
"apache-airflow",
"pytest-mock",
"fqdn",
"isoduration",
"tblib",
"prettytable",
"semantic-version",
"sympy",
"seaborn",
"confluent-kafka",
"azure-keyvault-secrets",
"opt-einsum",
"faker",
"jsonpickle",
"mpmath",
"patsy",
"azure-mgmt-resource",
"libclang",
"opencensus",
"antlr4-python3-runtime",
"pysftp",
"ordered-set",
"pymssql",
"db-dtypes",
"astunparse",
"inflection",
"gcsfs",
"thrift",
"parsedatetime",
"dask",
"deprecation",
"scikit-image",
"azure-datalake-store",
"moto",
"zeep",
"makefun",
"pyhcl",
"boto",
"libcst",
"graphviz",
"stevedore",
"gspread",
"snowballstemmer",
"ujson",
"zope-event",
"gevent",
"pyproj",
"checkov",
"python-gnupg",
"pathos",
"trio",
"trio-websocket",
"azure-eventhub",
"typed-ast",
"kombu",
"shap",
"pox",
"ppft",
"datasets",
"apscheduler",
"torchvision",
"click-man",
"accelerate",
"coloredlogs",
"xxhash",
"brotli",
"mypy-boto3-rds",
"docstring-parser",
"applicationinsights",
"apache-beam",
"structlog",
"tldextract",
"lightgbm",
"email-validator",
"wandb",
"cligj",
"kafka-python",
"pybind11",
"fire",
"celery",
"wsproto",
"pywavelets",
"numexpr",
"authlib",
"datetime",
"colorlog",
"pathlib2",
"uamqp",
"texttable",
"pytest-asyncio",
"google-cloud-logging",
"azure-cosmos",
"delta-spark",
"ecdsa",
"nvidia-cudnn-cu11",
"enum34",
"flask-cors",
"slicer",
"spacy",
"fiona",
"python-jose",
"watchtower",
"unicodecsv",
"imagesize",
"schema",
"alabaster",
"kfp",
"geopandas",
"marshmallow-enum",
"apache-airflow-providers-common-sql",
"pyfunctional",
"dbt-core",
"validators",
"keras-preprocessing",
"holidays",
"python-daemon",
"readme-renderer",
"djangorestframework",
"pandas-gbq",
"azure-storage-queue",
"azure-servicebus",
"hypothesis",
"tifffile",
"sshtunnel",
"graphframes",
"lz4",
"kfp-server-api",
"python-magic",
"invoke",
"avro-python3",
"parse",
"kfp-pipeline-spec",
"freezegun",
"constructs",
"outcome",
"python-multipart",
"billiard",
"monotonic",
"pip-tools",
"vine",
"fasteners",
"ddtrace",
"databricks-sql-connector",
"pycountry",
"azure-keyvault-keys",
"sendgrid",
"click-repl",
"srsly",
"pika",
"chex",
"thinc",
"ijson",
"jira",
"docker-pycreds",
"hpack",
"opencv-python-headless",
"blis",
"flask-sqlalchemy",
"fuzzywuzzy",
"xlwt",
"imbalanced-learn",
"qtconsole",
"pydata-google-auth",
"h2",
"pyproject-api",
"sh",
"lit",
"hyperframe",
"stringcase",
"astor",
"langchain-guides",
"langchain",
"wasabi",
"pytest-metadata",
"bitarray",
"pathtools",
"catalogue",
"nose",
"yapf",
"distributed",
"amqp",
"pathy",
"qtpy",
"types-pytz",
"boto3-stubs",
"triton",
"office365-rest-python-client",
"hatchling",
"jupyter-console",
"slackclient",
"atomicwrites",
"starkbank-ecdsa",
"omegaconf",
"editables",
"uvloop",
"humanize",
"knack",
"botocore-stubs",
"iso8601",
"smdebug-rulesconfig",
"crcmod",
"torchmetrics",
"fastparquet",
"python-levenshtein",
"pytimeparse",
"mypy-boto3-s3",
"einops",
"pywin32",
"jpype1",
"pydeequ",
"cog",
"azure-cli",
"pymeeus",
"types-six",
"murmurhash",
"ansible",
"pyspnego",
"inflect",
"phonenumbers",
"flask-wtf",
"cymem",
"preshed",
"cdk-nag",
"aws-requests-auth",
"google-cloud-audit-log",
"ua-parser",
"jsondiff",
"yamllint",
"nbclassic",
"cerberus",
"lazy-loader",
"dacite",
"statsd",
"cssselect",
"dpath",
"apispec",
"gensim",
"django-cors-headers",
"ruff",
"gradio",
"convertdate",
"scp",
"geopy",
"sqlalchemy-utils",
"azure-data-tables",
"pypdf2",
"partd",
"graphql-core",
"python-gitlab",
"ninja",
"ratelimit",
"junit-xml",
"levenshtein",
"fabric",
"pydot",
"azure-storage-file-share",
"pytorch-lightning",
"watchfiles",
"types-setuptools",
"requests-mock",
"strip-hints",
"keras-applications",
"pyotp",
"mashumaro",
"apache-airflow-providers-http",
"ipaddress",
"timm",
"click-didyoumean",
"bytecode",
"parameterized",
"netaddr",
"flask-appbuilder",
"pyperclip",
"openapi-spec-validator",
"onnx",
"marshmallow-sqlalchemy",
"locket",
"lark",
"mysqlclient",
"confection",
"pytest-html",
"azure-cosmosdb-table",
"agate",
"geographiclib",
"types-paramiko",
"pytest-rerunfailures",
"pyserial",
"spacy-loggers",
"flask-login",
"flask-jwt-extended",
"azure-devops",
"xarray",
"spark-nlp",
"dateparser",
"onnxruntime",
"twisted",
"lightning-utilities",
"wtforms",
"jaydebeapi",
"bokeh",
"natsort",
"google-cloud-bigtable",
"grpcio-health-checking",
"tensorflow-text",
"twine",
"commonmark",
"grpcio-reflection",
"flask-caching",
"cron-descriptor",
"pyaml",
"geoip2",
"nh3",
"autopep8",
"python-editor",
"logbook",
"ftfy",
"cachelib",
"datadog-api-client",
"jupyter",
"hologram",
"protobuf3-to-dict",
"ndg-httpsclient",
"promise",
"azureml-core",
"pydub",
"jax",
"flit-core",
"zstandard",
"cssselect2",
"minimal-snowplow-tracker",
"dbt-extractor",
"connexion",
"azure-keyvault-certificates",
"configargparse",
"aniso8601",
"cairocffi",
"hyperlink",
"cramjam",
"elasticsearch-dsl",
"mypy-boto3-redshift-data",
"h3",
"cairosvg",
"maxminddb",
"pytz-deprecation-shim",
"reportlab",
"langcodes",
"pytest-forked",
"pymupdf",
"ansible-core",
"cloudevents",
"leather",
"ddsketch",
"jaxlib",
"oldest-supported-numpy",
"tiktoken",
"supervisor",
"diskcache",
"functions-framework",
"hdfs",
"apache-airflow-providers-ssh",
"gradio-client",
"azure-multiapi-storage",
"funcsigs",
"azure-kusto-data",
"envier",
"pyhive",
"types-protobuf",
"django-filter",
"elastic-transport",
"parse-type",
"types-python-dateutil",
"boltons",
"python-docx",
"twilio",
"twilio-python",
"pgpy",
"korean-lunar-calendar",
"azure-eventgrid",
"async-generator",
"globus-sdk",
"apache-airflow-providers-imap",
"sentence-transformers",
"mkdocs-material",
"aws-xray-sdk",
"resolvelib",
"linkify-it-py",
"setuptools-rust",
"google",
"terminaltables",
"keystoneauth1",
"apache-airflow-providers-ftp",
"javaproperties",
"sqlalchemy-redshift",
"jdcal",
"pep517",
"incremental",
"limits",
"unittest-xml-reporting",
"frozendict",
"service-identity",
"factory-boy",
"ml-dtypes",
"addict",
"uc-micro-py",
"shortuuid",
"pypandoc",
"blessed",
"cx-oracle",
"requests-ntlm",
"django-extensions",
"apache-airflow-providers-amazon",
"python-keystoneclient",
"bracex",
"cmdstanpy",
"apache-airflow-providers-sqlite",
"cookiecutter",
"types-cryptography",
"flask-session",
"timezonefinder",
"magicattr",
"pymsteams",
"pylint-plugin-utils",
"voluptuous",
"langsmith",
"cinemagoer",
"passlib",
"imdbpy",
"emoji",
"databricks-api",
"configobj",
"bandit",
"ultralytics",
"w3lib",
"dirac",
"backports-functools-lru-cache",
"tableauserverclient",
"automat",
"pypika",
"pydash",
"py-cpuinfo",
"mmh3",
"tokenize-rt",
"python-swiftclient",
"tensorflow-hub",
"librosa",
"webdriver-manager",
"constantly",
"user-agents",
"injector",
"youtube-dl",
"pdfminer-six",
"markdown2",
"ffmpy",
"mergedeep",
"netifaces",
"databricks-sdk",
"azure-keyvault-administration",
"ephem",
"flax",
"urllib3-secure-extra",
"looker-sdk",
"kornia",
"python3-openid",
"userpath",
"polars",
"tensorboardx",
"openapi-schema-validator",
"jellyfish",
"ray",
"django-storages",
"asyncpg",
"dynamodb-json",
"pycocotools",
"lunarcalendar",
"types-redis",
"dm-tree",
"flask-limiter",
"scapy",
"sacremoses",
"hiredis",
"netcdf4",
"pyhocon",
"cmaes",
"feedparser",
"firebase-admin",
"yacs",
"prison",
"pytest-localserver",
"polling2",
"flask-babel",
"influxdb",
"binaryornot",
"psycopg3",
"sarif-om",
"jschema-to-python",
"cfn-flip",
"google-apitools",
"ipdb",
"pyrfc3339",
"filterpy",
"py-spy",
"wcmatch",
"launchdarkly-server-sdk",
"pyelftools",
"logging-azure-rest",
"python-jenkins",
"apache-airflow-providers-cncf-kubernetes",
"google-ads",
"clickclick",
"streamlit",
"pylint-django",
"yq",
"findspark",
"pycares",
"mkdocs",
"pytimeparse2",
"ldap3",
"pyee",
"pydocstyle",
"catboost",
"sqlalchemy-jsonfield",
"optuna",
"aws-lambda-powertools",
"lru-dict",
"rasterio",
"cartoframes",
"carto",
"aiodns",
"pyrestcli",
"opentracing",
"tensorflow-datasets",
"apache-airflow-providers-google",
"jsonlines",
"azure",
"backports-weakref",
"diff-cover",
"cftime",
"azure-kusto-ingest",
"qrcode",
"redis-py-cluster",
"diffusers",
"grpclib",
"pypdf",
"thrift-sasl",
"django-debug-toolbar",
"dynaconf",
"django-redis",
"salesforce-bulk",
"kazoo",
"configupdater",
"comtypes",
"langdetect",
"hydra-core",
"pytest-django",
"pywin32-ctypes",
"pyminizip",
"pathvalidate",
"google-re2",
"idna-ssl",
"dagster-pandas",
"toposort",
"expiringdict",
"rdflib",
"etils",
"rich-argparse",
"xyzservices",
"bottle",
"oslo-utils",
"prophet",
"pdfplumber",
"azure-mgmt-subscription",
"parsl",
"jsii",
"click-option-group",
"analytics-python",
"home-run",
"funcx",
"funcx-common",
"lmdb",
"zict",
"multi-key-dict",
"hatch-fancy-pypi-readme",
"googlemaps",
"pyudev",
"atlassian-python-api",
"dohq-artifactory",
"oslo-i18n",
"whitenoise",
"aiosqlite",
"python-engineio",
"enum-compat",
"affine",
"fs",
"flake8-bugbear",
"hyperopt",
"multipledispatch",
"oslo-serialization",
"pygeohash",
"somnium",
"kaleido",
"python-snappy",
"python-pptx",
"gql",
"pymdown-extensions",
"wexpect",
"types-pyopenssl",
"foundationdb",
"jsonschema-spec",
"iopath",
"snuggs",
"strict-rfc3339",
"tablib",
"orderedmultidict",
"sqlglot",
"fakeredis",
"pystan",
"python-socketio",
"robotframework",
"pkgconfig",
"pycairo",
"python-consul",
"curlify",
"types-toml",
"backports-tempfile",
"multimethod",
"pynamodb",
"docker-compose",
"munch",
"torchaudio",
"elementpath",
"mypy-boto3-lambda",
"python-decouple",
"mypy-boto3-dynamodb",
"pylev",
"pmdarima",
"drf-yasg",
"path",
"pyxlsb",
"pandasql",
"pipdeptree",
"debtcollector",
"nvidia-ml-py",
"pyinstaller-hooks-contrib",
"dvclive",
"koalas",
"arviz",
"coreapi",
"sqlalchemy-bigquery",
"pyquery",
"webob",
"faiss-cpu",
"flower",
"cloudformation-cli",
"azureml-dataset-runtime",
"azure-mgmt",
"cloudformation-cli-java-plugin",
"pyinstaller",
"python-box",
"pympler",
"mypy-boto3-secretsmanager",
"marshmallow-oneofschema",
"schedule",
"resampy",
"bitstring",
"timeout-decorator",
"furl",
"bidict",
"setuptools-git",
"jsonmerge",
"htmlmin",
"plumbum",
"gdown",
"evergreen-py",
"tableauhyperapi",
"xformers",
"yt-dlp",
"waitress",
"mypy-boto3-cloudformation",
"tld",
"pipx",
"fake-useragent",
"junitparser",
"pylint-flask",
"jaraco-functools",
"geomet",
"yappi",
"flask-openid",
"apache-airflow-providers-snowflake",
"ciso8601",
"paho-mqtt",
"aiohttp-retry",
"smbprotocol",
"mypy-protobuf",
"msgpack-python",
"dockerpty",
"cssutils",
"djangorestframework-simplejwt",
"wordcloud",
"pytest-env",
"django-environ",
"s3cmd",
"graphene",
"soundfile",
"html2text",
"dagster-dbt",
"apache-airflow-providers-databricks",
"python-nvd3",
"pygobject",
"azureml-sdk",
"click-default-group",
"azureml-dataprep",
"pygit2",
"boto3-type-annotations",
"imagehash",
"ec2-metadata",
"requests-futures",
"rx",
"geventhttpclient",
"wget",
"xmlschema",
"python-rapidjson",
"playwright",
"flatten-json",
"collections-extended",
"myst-parser",
"flask-restful",
"facebook-business",
"pdpyras",
"python-crfsuite",
"pydeck",
"dash-core-components",
"publication",
"zthreading",
"cheroot",
"minio",
"uwsgi",
"portpicker",
"simplegeneric",
"python-crontab",
"basicsr",
"facexlib",
"testpath",
"json-log-formatter",
"ghp-import",
"sseclient-py",
"ansi2html",
"jproperties",
"django-timezone-field",
"duckdb",
"pygsheets",
"pyzstd",
"opencv-contrib-python",
"pyyaml-env-tag",
"pyaes",
"pooch",
"funcy",
"appnope",
"cerberus-python-client",
"realesrgan",
"readchar",
"cassandra-driver",
"requests-unixsocket",
"pyproject-metadata",
"dictdiffer",
"pypng",
"ffmpeg-python",
"locust",
"pymc",
"modelx",
"ffn",
"finance-py",
"gs-quant",
"tf-quant-finance",
"finta",
"qstrader",
"blankly",
"ta-lib-python",
"zipline",
"bt",
"backtrader",
"pyalgotrade",
"pandas-ta",
"ta",
"finmarket-py",
"zvt",
"py-portfolio-opt",
"eiten",
"backtesting-py",
"quantstats",
"qtpylib",
"freqtrade",
"qlib",
"jesse",
"finrl",
"bulbea",
"octobot",
"tda-api",
"vectorbt",
"lean",
"pybroker",
"pyfolio",
"empyrical",
"finquant",
"riskfolio-lib",
"alphalens",
"arch",
"pyflux",
"tsfresh",
"gluonts",
"yfinance",
"alpha-vantage",
"pandas-datareader",
"yahoo-finance",
"findatapy",
"wallstreet",
"alpaca-trade-api-python",
"investpy",
"xlwings",
"dtale",
"mplfinance",
"keras",
"opensearch-py",
"openai",
"dash",
"stripe",
]