generative-computing
diff --git a/‎pyproject.toml‎
Lines changed: 6 additions & 2 deletions b/‎pyproject.toml‎
Lines changed: 6 additions & 2 deletions
@@ -61,14 +61,18 @@ hf = [
     "trl>=0.19.0",
 ]
 
+litellm = [
+    "litellm>=1.76"
+]
+
 watsonx = [
     "ibm-watsonx-ai>=1.3.31",
 ]
 docling = [
     "docling>=2.45.0",
 ]
 
-all = ["mellea[watsonx,docling,hf]"]
+all = ["mellea[watsonx,docling,hf,litellm]"]
 
 [dependency-groups]
 # Use these like:
@@ -138,7 +142,7 @@ ignore = [
 #   "UP006",  # List vs list, etc
 #   "UP007",  # Option and Union
 #   "UP035",  # `typing.Set` is deprecated, use `set` instead"
-  "PD901",  # Avoid using the generic variable name `df` for DataFrames 
+  "PD901",  # Avoid using the generic variable name `df` for DataFrames
 ]
 
 [tool.ruff.lint.pydocstyle]
Original file line number	Diff line number	Diff line change
`@@ -61,14 +61,18 @@ hf = [`
`61`	`61`	`"trl>=0.19.0",`
`62`	`62`	`]`
`63`	`63`
	`64`	`+litellm = [`
	`65`	`+ "litellm>=1.76"`
	`66`	`+]`
	`67`	`+`
`64`	`68`	`watsonx = [`
`65`	`69`	`"ibm-watsonx-ai>=1.3.31",`
`66`	`70`	`]`
`67`	`71`	`docling = [`
`68`	`72`	`"docling>=2.45.0",`
`69`	`73`	`]`
`70`	`74`
`71`		`-all = ["mellea[watsonx,docling,hf]"]`
	`75`	`+all = ["mellea[watsonx,docling,hf,litellm]"]`
`72`	`76`
`73`	`77`	`[dependency-groups]`
`74`	`78`	`# Use these like:`
`@@ -138,7 +142,7 @@ ignore = [`
`138`	`142`	`# "UP006", # List vs list, etc`
`139`	`143`	`# "UP007", # Option and Union`
`140`	`144`	# "UP035", # `typing.Set` is deprecated, use `set` instead"
`141`		- "PD901", # Avoid using the generic variable name `df` for DataFrames
	`145`	+ "PD901", # Avoid using the generic variable name `df` for DataFrames
`142`	`146`	`]`
`143`	`147`
`144`	`148`	`[tool.ruff.lint.pydocstyle]`