Skip to content

Commit

Permalink
Fixed extras grouping
Browse files Browse the repository at this point in the history
  • Loading branch information
slincoln-systemtwo committed Nov 7, 2024
1 parent fc6ea9a commit 6a144ec
Showing 1 changed file with 21 additions and 12 deletions.
33 changes: 21 additions & 12 deletions pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "SigmAIQ"
version = "0.4.4"
version = "0.4.5"
description = "Wrapper and tools for pySigma and Sigma rules"
authors = ["Stephen Lincoln <stephen.lincoln@attackiq.com>", "AttackIQ <support@attackiq.com>"]
readme = "README.md"
Expand Down Expand Up @@ -41,6 +41,15 @@ pysigma-pipeline-sysmon = "1.0.4"
pysigma-pipeline-windows = "1.2.0"
importlib-resources = "^5.13.0"

# Optional LLM dependencies, needed here for extras install with pip
langchain = {version = "^0.2.9", optional = true}
openai = {version = "*", optional = true}
tiktoken = {version = "*", optional = true}
faiss-cpu = {version = "*", optional = true}
langchain-community = {version = "^0.2.7", optional = true}
langchain-openai = {version = "^0.1.17", optional = true}
langchain-core = {version = "^0.2.20", optional = true}

[tool.poetry.group.llm]
optional = true

Expand All @@ -63,6 +72,17 @@ black = "^24.8.0"
ruff = "^0.7.2"
pytest-asyncio = "^0.24.0"

[tool.poetry.extras]
llm = [
"langchain",
"openai",
"tiktoken",
"faiss-cpu",
"langchain-community",
"langchain-openai",
"langchain-core"
]

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
Expand Down Expand Up @@ -123,14 +143,3 @@ testpaths = ["tests"]
pythonpath = ["."]
addopts = "--cov=sigmaiq --cov-report=term-missing"
asyncio_mode = "auto"

[tool.poetry.extras]
llm = [
"langchain",
"openai",
"tiktoken",
"faiss-cpu",
"langchain-community",
"langchain-openai",
"langchain-core"
]

0 comments on commit 6a144ec

Please sign in to comment.