Skip to content

Commit 0ea52a3

Browse files
fix: Added name param to PromptEngineCreateParams
1 parent 4f6659c commit 0ea52a3

File tree

1 file changed

+13
-4
lines changed

1 file changed

+13
-4
lines changed

jigsawstack/prompt_engine.py

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@ class PromptEngineRunResponse(TypedDict):
4848

4949

5050
class PromptEngineCreateParams(TypedDict):
51+
name: NotRequired[str]
5152
prompt: str
5253
inputs: NotRequired[List[object]]
5354
return_prompt: Union[str, List[object], Dict[str, str]]
@@ -119,10 +120,14 @@ def create(self, params: PromptEngineCreateParams) -> PromptEngineCreateResponse
119120

120121
def get(self, id: str) -> PromptEngineGetResponse:
121122
path = f"/prompt_engine/{id}"
122-
resp = Request(config=self.config, path=path, params={}, verb="get").perform_with_content()
123+
resp = Request(
124+
config=self.config, path=path, params={}, verb="get"
125+
).perform_with_content()
123126
return resp
124127

125-
def list(self, params: Union[PromptEngineListParams, None] = None) -> PromptEngineListResponse:
128+
def list(
129+
self, params: Union[PromptEngineListParams, None] = None
130+
) -> PromptEngineListResponse:
126131
if params is None:
127132
params = {}
128133

@@ -137,7 +142,9 @@ def list(self, params: Union[PromptEngineListParams, None] = None) -> PromptEngi
137142
base_path="/prompt_engine",
138143
params=params,
139144
)
140-
resp = Request(config=self.config, path=path, params={}, verb="get").perform_with_content()
145+
resp = Request(
146+
config=self.config, path=path, params={}, verb="get"
147+
).perform_with_content()
141148
return resp
142149

143150
def delete(self, id: str) -> PromptEngineDeleteResponse:
@@ -213,7 +220,9 @@ def __init__(
213220
headers=headers,
214221
)
215222

216-
async def create(self, params: PromptEngineCreateParams) -> PromptEngineCreateResponse:
223+
async def create(
224+
self, params: PromptEngineCreateParams
225+
) -> PromptEngineCreateResponse:
217226
path = "/prompt_engine"
218227
resp = await AsyncRequest(
219228
config=self.config,

0 commit comments

Comments
 (0)