pyproject.toml

[tool.poetry]
name = "local-llm-function-calling"
version = "0.1.23"
description = "A tool for generating function arguments and choosing what function to call with local LLMs"
authors = ["rizerphe <44440399+rizerphe@users.noreply.github.com>"]
readme = "README.md"
homepage = "https://github.com/rizerphe/local-llm-function-calling"
documentation = "https://local-llm-function-calling.readthedocs.io/"
keywords = ["llm", "jsonschema", "huggingface", "transformers", "local", "llama.cpp"]
license = "MIT"
packages = [
    {include = "local_llm_function_calling"},
    {include = "local_llm_function_calling/py.typed"},
]

[tool.poetry.dependencies]
python = "^3.11"
transformers = "^4.30.2"
json-schema-enforcer = "^0.1.3"
torch = "^2.0.1"
llama-cpp-python = {version = "^0.1.83", optional = true}

[tool.poetry.extras]
llama-cpp = ["llama-cpp-python"]


[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"