graphrag_sdk 0.2.2__tar.gz → 0.3.0__tar.gz

{graphrag_sdk-0.2.2 → graphrag_sdk-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: graphrag_sdk
-Version: 0.2.2
+Version: 0.3.0
 Summary: 
 Home-page: http://RAG-SDK.readthedocs.io
 License: MIT
@@ -15,20 +15,25 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Provides-Extra: all
+Provides-Extra: google-generativeai
+Provides-Extra: ollama
+Provides-Extra: openai
+Provides-Extra: vertexai
 Requires-Dist: backoff (>=2.2.1,<3.0.0)
 Requires-Dist: bs4 (>=0.0.2,<0.0.3)
 Requires-Dist: falkordb (>=1.0.9,<2.0.0)
 Requires-Dist: fix-busted-json (>=0.0.18,<0.0.19)
-Requires-Dist: google-generativeai (>=0.8.1,<0.9.0)
+Requires-Dist: google-generativeai (>=0.8.1,<0.9.0) ; extra == "all" or extra == "google-generativeai"
 Requires-Dist: ipykernel (>=6.29.5,<7.0.0)
-Requires-Dist: ollama (>=0.2.1,<0.3.0)
-Requires-Dist: openai (>=1.35.9,<2.0.0)
+Requires-Dist: ollama (>=0.2.1,<0.3.0) ; extra == "all" or extra == "ollama"
+Requires-Dist: openai (>=1.35.9,<2.0.0) ; extra == "all" or extra == "openai"
 Requires-Dist: pypdf (>=4.2.0,<5.0.0)
 Requires-Dist: python-abc (>=0.2.0,<0.3.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: ratelimit (>=2.2.1,<3.0.0)
 Requires-Dist: typing-extensions (>=4.12.1,<5.0.0)
-Requires-Dist: vertexai (>=1.49.0,<2.0.0)
+Requires-Dist: vertexai (>=1.49.0,<2.0.0) ; extra == "all" or extra == "vertexai"
 Project-URL: Repository, https://github.com/FalkorDB/GraphRAG-SDK
 Description-Content-Type: text/markdown
 
@@ -55,7 +60,13 @@ GraphRAG-SDK is a comprehensive solution for building Graph Retrieval-Augmented
 ### Install
 
 ```sh
-pip install graphrag_sdk
+pip install graphrag_sdk[all]
+```
+
+or (if only OpenAI is required)
+
+```sh
+pip install graphrag_sdk[openai]
 ```
 
 ### Prerequisites

{graphrag_sdk-0.2.2 → graphrag_sdk-0.3.0}/README.md

@@ -21,7 +21,13 @@ GraphRAG-SDK is a comprehensive solution for building Graph Retrieval-Augmented
 ### Install
 
 ```sh
-pip install graphrag_sdk
+pip install graphrag_sdk[all]
+```
+
+or (if only OpenAI is required)
+
+```sh
+pip install graphrag_sdk[openai]
 ```
 
 ### Prerequisites

{graphrag_sdk-0.2.2 → graphrag_sdk-0.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "graphrag_sdk"
-version = "0.2.2"
+version = "0.3.0"
 license = "MIT"
 description = ""
 authors = ["FalkorDB <info@falkordb.com>"]
@@ -15,16 +15,23 @@ falkordb = "^1.0.9"
 typing-extensions = "^4.12.1"
 bs4 = "^0.0.2"
 pypdf = "^4.2.0"
-vertexai = "^1.49.0"
 backoff = "^2.2.1"
 python-abc = "^0.2.0"
 ratelimit = "^2.2.1"
 python-dotenv = "^1.0.1"
-openai = "^1.35.9"
 fix-busted-json = "^0.0.18"
-ollama = "^0.2.1"
 ipykernel = "^6.29.5"
-google-generativeai = "^0.8.1"
+vertexai = { version = "^1.49.0", optional = true } 
+ollama = { version = "^0.2.1", optional = true }
+openai = { version = "^1.35.9", optional = true }
+google-generativeai = { version = "^0.8.1", optional = true }
+
+[tool.poetry.extras]
+all = ["vertexai", "ollama", "openai", "google-generativeai"]
+ollama = ["ollama"]
+openai = ["openai"]
+vertexai = ["vertexai"]
+google-generativeai = ["google-generativeai"]
 
 [tool.poetry.group.test.dependencies]
 pytest = "^8.2.1"