117 lines
3.2 KiB
TOML
117 lines
3.2 KiB
TOML
[tool.poetry]
|
|
name = "cognee"
|
|
version = "0.1.1"
|
|
description = "Cognee - is a library for enriching LLM context with a semantic layer for better understanding and reasoning."
|
|
authors = ["Vasilije Markovic"]
|
|
readme = "README.md"
|
|
license = "Apache-2.0"
|
|
homepage = "https://www.cognee.ai"
|
|
repository = "https://github.com/topoteretes/cognee"
|
|
classifiers = [
|
|
"Development Status :: 4 - Beta",
|
|
"Intended Audience :: Developers",
|
|
"License :: OSI Approved :: Apache Software License",
|
|
"Topic :: Software Development :: Libraries",
|
|
"Operating System :: MacOS :: MacOS X",
|
|
"Operating System :: POSIX :: Linux",
|
|
"Operating System :: Microsoft :: Windows"
|
|
]
|
|
|
|
[tool.poetry.dependencies]
|
|
python = "~3.11"
|
|
openai = "1.14.3"
|
|
python-dotenv = "1.0.1"
|
|
fastapi = "^0.109.2"
|
|
uvicorn = "0.22.0"
|
|
boto3 = "^1.26.125"
|
|
gunicorn = "^20.1.0"
|
|
sqlalchemy = "^2.0.21"
|
|
asyncpg = "^0.28.0"
|
|
instructor = "^0.6.8"
|
|
networkx = "^3.2.1"
|
|
graphviz = "^0.20.1"
|
|
debugpy = "^1.8.0"
|
|
pyarrow = "^15.0.0"
|
|
pylint = "^3.0.3"
|
|
aiosqlite = "^0.20.0"
|
|
pandas = "^2.2.0"
|
|
greenlet = "^3.0.3"
|
|
ruff = "^0.2.2"
|
|
filetype = "^1.2.0"
|
|
nltk = "^3.8.1"
|
|
dlt = "^0.4.6"
|
|
duckdb = {version = "^0.10.0", extras = ["dlt"]}
|
|
overrides = "^7.7.0"
|
|
aiofiles = "^23.2.1"
|
|
qdrant-client = "^1.8.0"
|
|
duckdb-engine = "^0.11.2"
|
|
graphistry = "^0.33.5"
|
|
tenacity = "^8.2.3"
|
|
weaviate-client = "^4.5.4"
|
|
scikit-learn = "^1.4.1.post1"
|
|
fastembed = "^0.2.5"
|
|
pypdf = "^4.1.0"
|
|
anthropic = "^0.21.3"
|
|
xmltodict = "^0.13.0"
|
|
jinja2 = "^3.1.3"
|
|
matplotlib = "^3.8.3"
|
|
|
|
[tool.poetry.extras]
|
|
dbt = ["dbt-core", "dbt-redshift", "dbt-bigquery", "dbt-duckdb", "dbt-snowflake", "dbt-athena-community", "dbt-databricks"]
|
|
gcp = ["grpcio", "google-cloud-bigquery", "db-dtypes", "gcsfs"]
|
|
# bigquery is alias on gcp extras
|
|
bigquery = ["grpcio", "google-cloud-bigquery", "pyarrow", "db-dtypes", "gcsfs"]
|
|
postgres = ["psycopg2-binary", "psycopg2cffi"]
|
|
redshift = ["psycopg2-binary", "psycopg2cffi"]
|
|
parquet = ["pyarrow"]
|
|
duckdb = ["duckdb"]
|
|
filesystem = ["s3fs", "botocore"]
|
|
s3 = ["s3fs", "botocore"]
|
|
gs = ["gcsfs"]
|
|
az = ["adlfs"]
|
|
snowflake = ["snowflake-connector-python"]
|
|
motherduck = ["duckdb", "pyarrow"]
|
|
cli = ["pipdeptree", "cron-descriptor"]
|
|
athena = ["pyathena", "pyarrow", "s3fs", "botocore"]
|
|
weaviate = ["weaviate-client"]
|
|
mssql = ["pyodbc"]
|
|
synapse = ["pyodbc", "adlfs", "pyarrow"]
|
|
databricks = ["databricks-sql-connector"]
|
|
lancedb = ["lancedb"]
|
|
pinecone = ["pinecone-client"]
|
|
neo4j = ["neo4j", "py2neo"]
|
|
notebook =[ "ipykernel","overrides", "ipywidgets", "jupyterlab", "jupyterlab_widgets", "jupyterlab-server", "jupyterlab-git"]
|
|
|
|
[tool.poetry.group.dev.dependencies]
|
|
pytest = "^7.4.0"
|
|
pytest-asyncio = "^0.21.1"
|
|
coverage = "^7.3.2"
|
|
mypy = "^1.7.1"
|
|
notebook = "^7.1.1"
|
|
|
|
[tool.poetry.group.docs.dependencies]
|
|
mkdocs = "^1.4.3"
|
|
mkdocs-material = {extras = ["imaging"], version = "^9.5.9"}
|
|
mkdocstrings = "^0.22.0"
|
|
mkdocstrings-python = "^1.1.2"
|
|
pytest-examples = "^0.0.10"
|
|
mkdocs-jupyter = "^0.24.6"
|
|
mkdocs-minify-plugin = "^0.8.0"
|
|
mkdocs-redirects = "^1.2.1"
|
|
|
|
[tool.poetry.group.test-docs.dependencies]
|
|
fastapi = "^0.109.2"
|
|
redis = "^5.0.1"
|
|
diskcache = "^5.6.3"
|
|
pandas = "^2.2.0"
|
|
tabulate = "^0.9.0"
|
|
|
|
|
|
[tool.ruff] # https://beta.ruff.rs/docs/
|
|
line-length = 100
|
|
ignore = ["F401"]
|
|
ignore-init-module-imports = true
|
|
|
|
[build-system]
|
|
requires = ["poetry-core"]
|
|
build-backend = "poetry.core.masonry.api"
|