Skip to content

Commit 5b29bef

Browse files
fix: AsyncPromptEngine must accept headers.
1 parent 6bef4e1 commit 5b29bef

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

jigsawstack/prompt_engine.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -204,13 +204,13 @@ def __init__(
204204
self,
205205
api_key: str,
206206
base_url: str,
207-
disable_request_logging: Union[bool, None] = False,
207+
headers: Union[Dict[str, str], None] = None,
208208
):
209-
super().__init__(api_key, base_url, disable_request_logging)
209+
super().__init__(api_key, base_url, headers)
210210
self.config = RequestConfig(
211211
base_url=base_url,
212212
api_key=api_key,
213-
disable_request_logging=disable_request_logging,
213+
headers=headers,
214214
)
215215

216216
async def create(self, params: PromptEngineCreateParams) -> PromptEngineCreateResponse:

0 commit comments

Comments
 (0)