Merge a772f0a42d into sapling-pr-archive-ehhuang

This commit is contained in:
ehhuang 2025-09-21 20:46:41 -07:00 committed by GitHub
commit 1a68f6446a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -162,6 +162,9 @@ class ResponsesStore:
:param model: The model to filter by. :param model: The model to filter by.
:param order: The order to sort the responses by. :param order: The order to sort the responses by.
""" """
if not self.sql_store:
raise ValueError("Responses store is not initialized")
if not order: if not order:
order = Order.desc order = Order.desc
@ -189,6 +192,9 @@ class ResponsesStore:
""" """
Get a response object with automatic access control checking. Get a response object with automatic access control checking.
""" """
if not self.sql_store:
raise ValueError("Responses store is not initialized")
row = await self.sql_store.fetch_one( row = await self.sql_store.fetch_one(
"openai_responses", "openai_responses",
where={"id": response_id}, where={"id": response_id},
@ -202,6 +208,9 @@ class ResponsesStore:
return OpenAIResponseObjectWithInput(**row["response_object"]) return OpenAIResponseObjectWithInput(**row["response_object"])
async def delete_response_object(self, response_id: str) -> OpenAIDeleteResponseObject: async def delete_response_object(self, response_id: str) -> OpenAIDeleteResponseObject:
if not self.sql_store:
raise ValueError("Responses store is not initialized")
row = await self.sql_store.fetch_one("openai_responses", where={"id": response_id}) row = await self.sql_store.fetch_one("openai_responses", where={"id": response_id})
if not row: if not row:
raise ValueError(f"Response with id {response_id} not found") raise ValueError(f"Response with id {response_id} not found")