put back prompt engine

Author: winzamark123

jigsawstack/__init__.py

@@ -15,6 +15,7 @@
 from .exceptions import JigsawStackError
 from .image_generation import ImageGeneration, AsyncImageGeneration
 from .classification import Classification, AsyncClassification
+from .prompt_engine import PromptEngine, AsyncPromptEngine
 
 
 class JigsawStack:
@@ -25,6 +26,7 @@ class JigsawStack:
     web: Web
     search: Search
     classification: Classification
+    prompt_engine: PromptEngine
     api_key: str
     api_url: str
     headers: Dict[str, str]
@@ -125,6 +127,12 @@ def __init__(
             disable_request_logging=disable_request_logging,
         ).classify
 
+        self.prompt_engine = PromptEngine(
+            api_key=api_key,
+            api_url=api_url,
+            disable_request_logging=disable_request_logging,
+        )
+
 
 class AsyncJigsawStack:
     validate: AsyncValidate
@@ -133,6 +141,7 @@ class AsyncJigsawStack:
     vision: AsyncVision
     image_generation: AsyncImageGeneration
     store: AsyncStore
+    prompt_engine: AsyncPromptEngine
     api_key: str
     api_url: str
     disable_request_logging: bool
@@ -235,6 +244,12 @@ def __init__(
             disable_request_logging=disable_request_logging,
         ).classify
 
+        self.prompt_engine = AsyncPromptEngine(
+            api_key=api_key,
+            api_url=api_url,
+            disable_request_logging=disable_request_logging,
+        )
+
 
 # Create a global instance of the Web class
 __all__ = ["JigsawStack", "Search", "JigsawStackError", "AsyncJigsawStack"]