Merge pull request #15 from JigsawStack/check

Improve sdk

Author: iamNarcisse

jigsawstack/prompt_engine.py

@@ -16,7 +16,7 @@ class PromptEngineRunParams(TypedDict):
     prompt:str
     inputs: NotRequired[List[object]]
     return_prompt: Union[str, List[object], Dict[str, str]]
-    grok_key: NotRequired[str]
+    # grok_key: NotRequired[str]
     input_values: NotRequired[Dict[str, str]]
 
 class PromptEngineExecuteParams(TypedDict):
@@ -30,7 +30,7 @@ class PromptEngineRunResponse(TypedDict):
 
 class PromptEngineCreateParams(TypedDict):
     prompt: str
-    grok_key: NotRequired[str]
+    # grok_key: NotRequired[str]
     inputs: NotRequired[List[object]]
     return_prompt: Union[str, List[object], Dict[str, str]]
 
@@ -100,15 +100,15 @@ def delete(self, id:str) -> PromptEngineDeleteResponse:
             path=path,params={},verb="DELETE").perform_with_content()
         return resp
 
-    def run(self, params:PromptEngineRunParams) -> PromptEngineRunResponse:
+    def run_prompt_direct(self, params:PromptEngineRunParams) -> PromptEngineRunResponse:
         path = "/prompt_engine/run"
         resp = Request(
             api_key=self.api_key,
             api_url=self.api_url,
             path=path,params=cast(Dict[Any, Any], params),verb="post").perform_with_content()
         return resp
 
-    def execute(self, params:PromptEngineExecuteParams) -> PromptEngineRunResponse:
+    def run(self, params:PromptEngineExecuteParams) -> PromptEngineRunResponse:
         id = params.get("id")
         path = f"/prompt_engine/{id}"
         resp = Request(

setup.py

@@ -6,7 +6,7 @@
 
 setup(
     name="jigsawstack",
-    version="0.1.9",
+    version="0.1.10",
     description="JigsawStack Python SDK",
     long_description=open("README.md", encoding="utf8").read(),
     long_description_content_type="text/markdown",