@@ -197,9 +197,7 @@ def _record_version_headers(self, headers: httpx.Headers) -> None:
197197 self ._last_server_version = headers .get ("X-AgentFlow-Version" )
198198 self ._last_latest_version = headers .get ("X-AgentFlow-Latest-Version" )
199199 self ._last_deprecated = headers .get ("X-AgentFlow-Deprecated" )
200- self ._last_deprecation_warning = headers .get (
201- "X-AgentFlow-Deprecation-Warning"
202- )
200+ self ._last_deprecation_warning = headers .get ("X-AgentFlow-Deprecation-Warning" )
203201
204202 async def _get_entity (
205203 self ,
@@ -219,9 +217,7 @@ def _parse_contract_versions(
219217 return {}
220218 entity , separator , version = contract_version .partition (":" )
221219 if not separator or not entity or not version :
222- raise ValueError (
223- "contract_version must use '<entity>:<version>' format."
224- )
220+ raise ValueError ("contract_version must use '<entity>:<version>' format." )
225221 return {entity : version [1 :] if version .startswith ("v" ) else version }
226222
227223 async def _apply_contract_version (
@@ -234,27 +230,14 @@ async def _apply_contract_version(
234230 return payload
235231 contract = await self ._get_contract (entity_type , version )
236232 fields = contract .get ("fields" , [])
237- required_fields = [
238- field ["name" ]
239- for field in fields
240- if field .get ("required" )
241- ]
242- missing_fields = [
243- field_name
244- for field_name in required_fields
245- if field_name not in payload
246- ]
233+ required_fields = [field ["name" ] for field in fields if field .get ("required" )]
234+ missing_fields = [field_name for field_name in required_fields if field_name not in payload ]
247235 if missing_fields :
248236 raise AgentFlowError (
249- "Contract validation failed. Missing required fields: "
250- + ", " .join (missing_fields )
237+ "Contract validation failed. Missing required fields: " + ", " .join (missing_fields )
251238 )
252239 allowed_fields = {field ["name" ] for field in fields }
253- return {
254- name : value
255- for name , value in payload .items ()
256- if name in allowed_fields
257- }
240+ return {name : value for name , value in payload .items () if name in allowed_fields }
258241
259242 async def _get_contract (self , entity_type : str , version : str ) -> dict [str , Any ]:
260243 cache_key = (entity_type , version )
@@ -355,11 +338,7 @@ async def _query_page(
355338 payload ["limit" ] = limit
356339 if cursor is not None :
357340 payload ["cursor" ] = cursor
358- headers = (
359- {"Idempotency-Key" : idempotency_key }
360- if idempotency_key is not None
361- else None
362- )
341+ headers = {"Idempotency-Key" : idempotency_key } if idempotency_key is not None else None
363342 return await self ._request ("POST" , "/v1/query" , json = payload , headers = headers )
364343
365344 async def query (
@@ -404,8 +383,7 @@ async def search(
404383 async def list_contracts (self ) -> list [ContractSummary ]:
405384 payload = await self ._request ("GET" , "/v1/contracts" )
406385 return [
407- ContractSummary .model_validate (contract )
408- for contract in payload .get ("contracts" , [])
386+ ContractSummary .model_validate (contract ) for contract in payload .get ("contracts" , [])
409387 ]
410388
411389 async def get_contract (
@@ -438,11 +416,7 @@ async def validate_contract(
438416 * ,
439417 idempotency_key : str | None = None ,
440418 ) -> ContractValidation :
441- headers = (
442- {"Idempotency-Key" : idempotency_key }
443- if idempotency_key is not None
444- else None
445- )
419+ headers = {"Idempotency-Key" : idempotency_key } if idempotency_key is not None else None
446420 response = await self ._request (
447421 "POST" ,
448422 f"/v1/contracts/{ entity } /validate" ,
@@ -500,11 +474,7 @@ async def batch(
500474 * ,
501475 idempotency_key : str | None = None ,
502476 ) -> dict [str , Any ]:
503- headers = (
504- {"Idempotency-Key" : idempotency_key }
505- if idempotency_key is not None
506- else None
507- )
477+ headers = {"Idempotency-Key" : idempotency_key } if idempotency_key is not None else None
508478 return await self ._request (
509479 "POST" ,
510480 "/v1/batch" ,
0 commit comments