fix: set pillow and unstructured to be installed by default (#675)

Co-authored-by: zeromquan <[email protected]>
Co-authored-by: Wendong <[email protected]>

Author: 3 people

camel/agents/knowledge_graph_agent.py

@@ -13,7 +13,10 @@
 # =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
 from typing import Optional, Union
 
-from unstructured.documents.elements import Element
+try:
+    from unstructured.documents.elements import Element
+except ImportError:
+    Element = None
 
 from camel.agents import ChatAgent
 from camel.messages import BaseMessage

camel/storages/graph_storages/graph_element.py

@@ -16,7 +16,10 @@
 from dataclasses import dataclass, field
 from typing import List, Union
 
-from unstructured.documents.elements import Element
+try:
+    from unstructured.documents.elements import Element
+except ImportError:
+    Element = None
 
 
 @dataclass
@@ -72,3 +75,8 @@ class Config:
     nodes: List[Node]
     relationships: List[Relationship]
     source: Element
+
+    def __post_init__(self):
+        if Element is None:
+            raise ImportError("""The 'unstructured' package is required to use
+                              the 'source' attribute.""")

poetry.lock

@@ -39,6 +39,7 @@ anthropic = "^0.29.0"
 docstring-parser = "^0.15"
 pydantic = ">=1.9,<3"
 curl_cffi = "0.6.2"
+pillow ="^10.2.0"
 
 # model-platforms
 litellm = { version = "^1.38.1", optional = true }
@@ -67,7 +68,6 @@ requests_oauthlib = { version = "^1.3.1", optional = true }
 prance = { version = "^23.6.21.0", optional = true }
 openapi-spec-validator = { version = "^0.7.1", optional = true }
 unstructured = { extras = ["all-docs"], version = "^0.10.30", optional = true }
-pillow = { version = "^10.2.0", optional = true }
 slack-sdk = { version = "^3.27.2", optional = true }
 pydub = { version = "^0.25.1", optional = true }
 pygithub = { version = "^2.3.0", optional = true }