kiba-core 0.5.3.dev51__tar.gz → 0.5.3.dev52__tar.gz

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/CHANGELOG.md

@@ -33,6 +33,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 - [MINOR] Truncate discord notification messages to fit api
 - [MINOR] Added `blockNumber` to `EthClient.multicall` function
 - [MINOR] Added `SqlMessageQueue` and `CosmosMessageQueue` to support database backed queues
+- [MINOR] Added `shouldAllowFailures` to `EthClient.multicall` to allow individual calls to fail without reverting the whole batch
 
 ### Changed
 

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kiba-core
-Version: 0.5.3.dev51
+Version: 0.5.3.dev52
 Summary: Kiba Labs' python utilities
 Project-URL: repository, https://github.com/kibalabs/core-py
 Requires-Python: <4,>=3.12
@@ -9,33 +9,33 @@ Requires-Dist: aiofiles>=25.1.0
 Requires-Dist: pydantic>=2.13.4
 Requires-Dist: orjson>=3.11.9
 Provides-Extra: api
-Requires-Dist: fastapi>=0.139.1; extra == "api"
+Requires-Dist: fastapi>=0.141.1; extra == "api"
 Requires-Dist: uvicorn[standard]>=0.51.0; extra == "api"
 Provides-Extra: core-api
-Requires-Dist: starlette>=1.3.1; extra == "core-api"
+Requires-Dist: starlette>=1.6.0; extra == "core-api"
 Requires-Dist: uvicorn[standard]>=0.51.0; extra == "core-api"
 Provides-Extra: storage
-Requires-Dist: aiobotocore>=3.7.0; extra == "storage"
+Requires-Dist: aiobotocore>=3.9.0; extra == "storage"
 Provides-Extra: queues
-Requires-Dist: aiobotocore>=3.7.0; extra == "queues"
+Requires-Dist: aiobotocore>=3.9.0; extra == "queues"
 Provides-Extra: queue-sqs
-Requires-Dist: aiobotocore>=3.7.0; extra == "queue-sqs"
+Requires-Dist: aiobotocore>=3.9.0; extra == "queue-sqs"
 Provides-Extra: queue-aqs
 Requires-Dist: azure-storage-queue[aio]>=12.17.0; extra == "queue-aqs"
 Provides-Extra: queue-cosmos
-Requires-Dist: azure-cosmos>=4.16.2; extra == "queue-cosmos"
+Requires-Dist: azure-cosmos>=4.16.3; extra == "queue-cosmos"
 Provides-Extra: database-psql
-Requires-Dist: sqlalchemy[asyncio]>=2.0.51; extra == "database-psql"
+Requires-Dist: sqlalchemy[asyncio]>=2.0.52; extra == "database-psql"
 Requires-Dist: asyncpg>=0.31.0; extra == "database-psql"
 Provides-Extra: database-sqlite
-Requires-Dist: sqlalchemy[asyncio]>=2.0.51; extra == "database-sqlite"
+Requires-Dist: sqlalchemy[asyncio]>=2.0.52; extra == "database-sqlite"
 Requires-Dist: aiosqlite>=0.22.1; extra == "database-sqlite"
 Provides-Extra: requester
 Requires-Dist: httpx>=0.28.1; extra == "requester"
 Provides-Extra: web3
 Requires-Dist: web3>=7.16.0; extra == "web3"
 Provides-Extra: types
-Requires-Dist: types-aiobotocore[essential]>=3.7.0; extra == "types"
+Requires-Dist: types-aiobotocore[essential]>=3.9.0; extra == "types"
 Requires-Dist: types-aiofiles>=25.1.0.20260518; extra == "types"
 
 # Core-Py

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/api/authorizer.py

@@ -1,9 +1,7 @@
 import functools
 import typing
 from collections.abc import AsyncIterator
-from typing import ParamSpec
 
-from mypy_extensions import Arg
 from pydantic import BaseModel
 
 from core import logging
@@ -13,7 +11,6 @@ from core.exceptions import UnauthorizedException
 from core.http.basic_authentication import BasicAuthentication
 from core.http.jwt import Jwt
 
-_P = ParamSpec('_P')
 _AnyReturn = typing.Awaitable[typing.Any] | AsyncIterator[typing.Any]  # type: ignore[explicit-any]
 
 
@@ -43,19 +40,18 @@ async def _authorize_bearer_jwt[ApiRequest: BaseModel](request: KibaApiRequest[A
 
 def authorize_bearer_jwt[ApiRequest: BaseModel](  # type: ignore[explicit-any]
     authorizer: Authorizer,
-) -> typing.Callable[[typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], _AnyReturn]], typing.Callable[_P, typing.Any]]:
-    def decorator(func: typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], _AnyReturn]) -> typing.Callable[_P, typing.Any]:  # type: ignore[explicit-any]
+) -> typing.Callable[[typing.Callable[[KibaApiRequest[ApiRequest]], _AnyReturn]], typing.Callable[[KibaApiRequest[ApiRequest]], typing.Any]]:
+    def decorator(func: typing.Callable[[KibaApiRequest[ApiRequest]], _AnyReturn]) -> typing.Callable[[KibaApiRequest[ApiRequest]], typing.Any]:  # type: ignore[explicit-any]
         @functools.wraps(func)
         async def async_wrapper(request: KibaApiRequest[ApiRequest]) -> typing.Any:  # type: ignore[explicit-any, misc]
             request.authJwt = await _authorize_bearer_jwt(request=request, authorizer=authorizer)
-            result = func(request=request)
+            result = func(request)
             # NOTE(krishan711): this is here to support streaming responses which return an async generator
             if hasattr(result, '__aiter__'):
                 return result
             return await result
 
-        # TODO(krishan711): figure out correct typing here
-        return async_wrapper  # type: ignore[return-value]
+        return async_wrapper
 
     return decorator
 
@@ -72,26 +68,25 @@ async def get_basic_authentication_from_authorization_signature[ApiRequest: Base
     except UnauthorizedException:
         raise
     except BaseException as exception:  # noqa: BLE001
-        logging.exception(exception)  # type: ignore[arg-type]
+        logging.exception(exception)  # type: ignore[arg-type, ty:invalid-argument-type]
         raise ForbiddenException(message='AUTH_INVALID')
     return BasicAuthentication(username=signerId, password=signatureString)
 
 
 def authorize_signature[ApiRequest: BaseModel](  # type: ignore[explicit-any]
     authorizer: SignatureAuthorizer,
-) -> typing.Callable[[typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], _AnyReturn]], typing.Callable[_P, typing.Any]]:
-    def decorator(func: typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], _AnyReturn]) -> typing.Callable[_P, typing.Any]:  # type: ignore[explicit-any]
+) -> typing.Callable[[typing.Callable[[KibaApiRequest[ApiRequest]], _AnyReturn]], typing.Callable[[KibaApiRequest[ApiRequest]], typing.Any]]:
+    def decorator(func: typing.Callable[[KibaApiRequest[ApiRequest]], _AnyReturn]) -> typing.Callable[[KibaApiRequest[ApiRequest]], typing.Any]:  # type: ignore[explicit-any]
         @functools.wraps(func)
         async def async_wrapper(request: KibaApiRequest[ApiRequest]) -> typing.Any:  # type: ignore[explicit-any, misc]
             request.authBasic = await get_basic_authentication_from_authorization_signature(request=request, authorizer=authorizer)
-            result = func(request=request)
+            result = func(request)
             # NOTE(krishan711): this is here to support streaming responses which return an async generator
             if hasattr(result, '__aiter__'):
                 return result
             return await result
 
-        # TODO(krishan711): figure out correct typing here
-        return async_wrapper  # type: ignore[return-value]
+        return async_wrapper
 
     return decorator
 
@@ -112,8 +107,8 @@ class StaticTokenAuthorizer(TokenAuthorizer):
 
 def authorize_token[ApiRequest: BaseModel](  # type: ignore[explicit-any]
     authorizer: TokenAuthorizer,
-) -> typing.Callable[[typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], _AnyReturn]], typing.Callable[_P, typing.Any]]:
-    def decorator(func: typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], _AnyReturn]) -> typing.Callable[_P, typing.Any]:  # type: ignore[explicit-any]
+) -> typing.Callable[[typing.Callable[[KibaApiRequest[ApiRequest]], _AnyReturn]], typing.Callable[[KibaApiRequest[ApiRequest]], typing.Any]]:
+    def decorator(func: typing.Callable[[KibaApiRequest[ApiRequest]], _AnyReturn]) -> typing.Callable[[KibaApiRequest[ApiRequest]], typing.Any]:  # type: ignore[explicit-any]
         @functools.wraps(func)
         async def async_wrapper(request: KibaApiRequest[ApiRequest]) -> typing.Any:  # type: ignore[explicit-any, misc]
             authorization = request.headers.get('Authorization')
@@ -122,13 +117,12 @@ def authorize_token[ApiRequest: BaseModel](  # type: ignore[explicit-any]
             if not authorization.startswith('Token '):
                 raise ForbiddenException(message='AUTH_INVALID')
             await authorizer.validate_token(authorization[6:])
-            result = func(request=request)
+            result = func(request)
             # NOTE(krishan711): this is here to support streaming responses which return an async generator
             if hasattr(result, '__aiter__'):
                 return result
             return await result
 
-        # TODO(krishan711): figure out correct typing here
-        return async_wrapper  # type: ignore[return-value]
+        return async_wrapper
 
     return decorator

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/api/json_route.py

@@ -1,8 +1,6 @@
 import functools
 import typing
-from typing import ParamSpec
 
-from mypy_extensions import Arg
 from pydantic import BaseModel
 from pydantic import ValidationError
 
@@ -13,14 +11,12 @@ from core.exceptions import InternalServerErrorException
 from core.util import json_util
 from core.util.typing_util import JsonObject
 
-_P = ParamSpec('_P')
-
 
 def json_route[ApiRequest: BaseModel, ApiResponse: BaseModel](
     requestType: typing.Type[ApiRequest],
     responseType: typing.Type[ApiResponse],
-) -> typing.Callable[[typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], typing.Awaitable[ApiResponse]]], typing.Callable[_P, KibaJSONResponse]]:
-    def decorator(func: typing.Callable[[Arg(KibaApiRequest[ApiRequest], 'request')], typing.Awaitable[ApiResponse]]) -> typing.Callable[_P, KibaJSONResponse]:
+) -> typing.Callable[[typing.Callable[[KibaApiRequest[ApiRequest]], typing.Awaitable[ApiResponse]]], typing.Callable[..., typing.Awaitable[KibaJSONResponse]]]:
+    def decorator(func: typing.Callable[[KibaApiRequest[ApiRequest]], typing.Awaitable[ApiResponse]]) -> typing.Callable[..., typing.Awaitable[KibaJSONResponse]]:
         @functools.wraps(func)
         async def async_wrapper(*args: typing.Any) -> KibaJSONResponse:  # type: ignore[explicit-any, misc]
             receivedRequest = args[0]
@@ -42,12 +38,11 @@ def json_route[ApiRequest: BaseModel, ApiResponse: BaseModel](
                 raise BadRequestException(f'Invalid request: {validationErrorMessage}')
             kibaRequest: KibaApiRequest[ApiRequest] = KibaApiRequest(scope=receivedRequest.scope, receive=receivedRequest._receive, send=receivedRequest._send)  # noqa: SLF001
             kibaRequest.data = requestParams
-            receivedResponse = await func(request=kibaRequest)
+            receivedResponse = await func(kibaRequest)
             if not isinstance(receivedResponse, responseType):
                 raise InternalServerErrorException(f'Expected response to be of type {responseType}, got {type(receivedResponse)}')
             return KibaJSONResponse(content=receivedResponse.model_dump())
 
-        # TODO(krishan711): figure out correct typing here
-        return async_wrapper  # type: ignore[return-value]
+        return async_wrapper
 
     return decorator

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/api/streaming_json_route.py

@@ -2,7 +2,6 @@ import functools
 import inspect
 import typing
 from collections.abc import AsyncIterator
-from typing import ParamSpec
 
 from pydantic import BaseModel
 from pydantic import ValidationError
@@ -14,8 +13,6 @@ from core.exceptions import InternalServerErrorException
 from core.util import json_util
 from core.util.typing_util import JsonObject
 
-_P = ParamSpec('_P')
-
 
 async def _convert_to_json_generator[T: BaseModel](response_iterator: AsyncIterator[T], expectedType: typing.Type[T]) -> AsyncIterator[bytes]:
     async for content in response_iterator:
@@ -27,8 +24,8 @@ async def _convert_to_json_generator[T: BaseModel](response_iterator: AsyncItera
 def streaming_json_route[ApiRequest: BaseModel, ApiResponse: BaseModel](
     requestType: typing.Type[ApiRequest],
     responseType: typing.Type[ApiResponse],
-) -> typing.Callable[[typing.Callable[[KibaApiRequest[ApiRequest]], AsyncIterator[ApiResponse]]], typing.Callable[_P, StreamingResponse]]:
-    def decorator(func: typing.Callable[[KibaApiRequest[ApiRequest]], AsyncIterator[ApiResponse]]) -> typing.Callable[_P, StreamingResponse]:
+) -> typing.Callable[[typing.Callable[[KibaApiRequest[ApiRequest]], AsyncIterator[ApiResponse]]], typing.Callable[..., typing.Awaitable[StreamingResponse]]]:
+    def decorator(func: typing.Callable[[KibaApiRequest[ApiRequest]], AsyncIterator[ApiResponse]]) -> typing.Callable[..., typing.Awaitable[StreamingResponse]]:
         @functools.wraps(func)
         async def async_wrapper(*args: typing.Any, **kwargs: typing.Any) -> StreamingResponse:  # type: ignore[explicit-any, misc]
             receivedRequest = kwargs.get('request', args[0] if args else None)
@@ -58,7 +55,6 @@ def streaming_json_route[ApiRequest: BaseModel, ApiResponse: BaseModel](
             # NOTE(krishan711): we set content-encoding to identity to prevent gzip from trying to process it (cos it buffers all the content)
             return StreamingResponse(content=wrappedGenerator, media_type='application/x-ndjson', headers={'Content-Encoding': 'identity'})
 
-        # TODO(krishan711): figure out correct typing here
-        return async_wrapper  # type: ignore[return-value]
+        return async_wrapper
 
     return decorator

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/logging.py

@@ -4,7 +4,6 @@ import logging
 import os
 import re
 import sys
-import typing
 from collections.abc import Callable
 from collections.abc import Collection
 from logging import Formatter
@@ -153,7 +152,7 @@ def init_basic_logging(showDebug: bool = False) -> None:
         init_logger(logger=logger, loggingLevel=loggingLevel, handler=handler)
 
 
-def _serialize_numeric_value(value: Union[None, float]) -> str:
+def _serialize_numeric_value(value: Union[float, None]) -> str:
     if value is None:
         return ''
     roundedNumber = round(value, 6)
@@ -169,7 +168,7 @@ def stat(name: str, key: str, value: float = 1) -> None:
         nameValue = _serialize_string_value(value=str(name))
         keyValue = _serialize_string_value(value=str(key))
         statValue = _serialize_numeric_value(value=value)
-        STAT_LOGGER.log(level=logging.INFO, msg='', extra=typing.cast(dict[str, str], {'statName': nameValue, 'statKey': keyValue, 'statValue': statValue}))
+        STAT_LOGGER.log(level=logging.INFO, msg='', extra={'statName': nameValue, 'statKey': keyValue, 'statValue': statValue})
 
 
 # TODO(krishan711): make pathPattern mandatory in next major release
@@ -184,7 +183,7 @@ def api(action: str, path: str, query: str, pathPattern: str | None = None, resp
         API_LOGGER.log(
             level=logging.INFO,
             msg='',
-            extra=typing.cast(dict[str, str], {'apiAction': actionString, 'apiPath': pathString, 'apiPathPattern': pathPatternString, 'apiQuery': queryString, 'apiResponse': responseString or '', 'apiDuration': durationString or ''}),
+            extra={'apiAction': actionString, 'apiPath': pathString, 'apiPathPattern': pathPatternString, 'apiQuery': queryString, 'apiResponse': responseString or '', 'apiDuration': durationString or ''},
         )
 
 
@@ -204,11 +203,11 @@ def _log(level: int, msg: str, *args: Any, **kwargs: Any) -> None:  # type: igno
 
 
 def critical(msg: str, *args: Any, **kwargs: Any) -> None:  # type: ignore[explicit-any]
-    _log(level=CRITICAL, msg=msg, *args, **kwargs)  # type: ignore[misc]  # noqa: B026
+    _log(CRITICAL, msg, *args, **kwargs)
 
 
 def error(msg: str, *args: Any, **kwargs: Any) -> None:  # type: ignore[explicit-any]
-    _log(level=ERROR, msg=msg, *args, **kwargs)  # type: ignore[misc]  # noqa: B026
+    _log(ERROR, msg, *args, **kwargs)
 
 
 def exception(msg: Union[str, Exception], *args: Any, exc_info: bool = True, **kwargs: Any) -> None:  # type: ignore[explicit-any]
@@ -218,19 +217,19 @@ def exception(msg: Union[str, Exception], *args: Any, exc_info: bool = True, **k
         resolvedMessage = str(msg)
     else:
         resolvedMessage = msg
-    _log(level=ERROR, msg=resolvedMessage, exc_info=exc_info, *args, **kwargs)  # type: ignore[misc]  # noqa: B026
+    _log(ERROR, resolvedMessage, *args, exc_info=exc_info, **kwargs)
 
 
 def warning(msg: str, *args: Any, **kwargs: Any) -> None:  # type: ignore[explicit-any]
-    _log(level=WARNING, msg=msg, *args, **kwargs)  # type: ignore[misc]  # noqa: B026
+    _log(WARNING, msg, *args, **kwargs)
 
 
 def info(msg: str, *args: Any, **kwargs: Any) -> None:  # type: ignore[explicit-any]
-    _log(level=INFO, msg=msg, *args, **kwargs)  # type: ignore[misc]  # noqa: B026
+    _log(INFO, msg, *args, **kwargs)
 
 
 def debug(msg: str, *args: Any, **kwargs: Any) -> None:  # type: ignore[explicit-any]
-    _log(level=DEBUG, msg=msg, *args, **kwargs)  # type: ignore[misc]  # noqa: B026
+    _log(DEBUG, msg, *args, **kwargs)
 
 
 def basicConfig(**kwargs: Any) -> None:  # type: ignore[explicit-any]  # noqa: N802  # noqa: B026

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/queues/aqs.py

@@ -18,7 +18,7 @@ class AqsMessage(Message):
 
     @classmethod
     def from_aqs_message(cls, aqsMessage: RawAqsMessage) -> AqsMessage:
-        message = Message.parse_raw(aqsMessage.content)
+        message = Message.model_validate_json(aqsMessage.content)
         return cls(
             command=message.command,
             content=message.content,
@@ -54,7 +54,7 @@ class AqsMessageQueue(MessageQueue[AqsMessage]):
         if not self._aqsClient:
             raise InternalServerErrorException('You need to call .connect() before trying to send messages')
         message.prepare_for_send()
-        await self._aqsClient.send_message(visibility_timeout=delaySeconds, content=message.json())
+        await self._aqsClient.send_message(visibility_timeout=delaySeconds, content=message.model_dump_json())
 
     async def send_messages(self, messages: Sequence[Message], delaySeconds: int = 0) -> None:
         if not self._aqsClient:

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/queues/sql.py

@@ -8,6 +8,7 @@ from collections.abc import Sequence
 from typing import Any
 
 import sqlalchemy
+import sqlalchemy.exc
 from sqlalchemy.dialects import postgresql as sqlalchemy_psql
 
 from core.queues.message_queue import MessageQueue

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/queues/sqs.py

@@ -27,7 +27,7 @@ class SqsMessage(Message):
 
     @classmethod
     def from_sqs_message(cls, sqsMessage: RawSqsMessageTypeDef) -> SqsMessage:
-        message = Message.parse_raw(sqsMessage['Body'])
+        message = Message.model_validate_json(sqsMessage['Body'])
         return cls(
             command=message.command,
             content=message.content,
@@ -61,7 +61,7 @@ class SqsMessageQueue(MessageQueue[SqsMessage]):
         if not self._sqsClient:
             raise InternalServerErrorException('You need to call .connect() before trying to send messages')
         message.prepare_for_send()
-        await self._sqsClient.send_message(QueueUrl=self.queueUrl, DelaySeconds=delaySeconds, MessageAttributes={}, MessageBody=message.json())
+        await self._sqsClient.send_message(QueueUrl=self.queueUrl, DelaySeconds=delaySeconds, MessageAttributes={}, MessageBody=message.model_dump_json())
 
     async def send_messages(self, messages: Sequence[Message], delaySeconds: int = 0) -> None:
         if not self._sqsClient:
@@ -71,7 +71,7 @@ class SqsMessageQueue(MessageQueue[SqsMessage]):
             requests: list[SendMessageBatchRequestEntryTypeDef] = []
             for index, message in enumerate(messageChunk):
                 message.prepare_for_send()
-                requests.append({'Id': str(index), 'DelaySeconds': int(delaySeconds), 'MessageAttributes': {}, 'MessageBody': message.json()})
+                requests.append({'Id': str(index), 'DelaySeconds': int(delaySeconds), 'MessageAttributes': {}, 'MessageBody': message.model_dump_json()})
             response = await self._sqsClient.send_message_batch(QueueUrl=self.queueUrl, Entries=requests)
             failures += response.get('Failed', [])
         if len(failures) > 0:

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/s3_manager.py

@@ -94,7 +94,7 @@ class S3Manager:
             raise InternalServerErrorException('You need to call .connect() before trying to write to s3')
         targetBucket, targetKey = self._split_path_to_bucket_key(path=targetPath)
         extraArgs = self._get_extra_args(accessControl=accessControl, cacheControl=cacheControl, contentType=contentType or self._get_file_mimetype(fileName=targetKey))
-        await self._s3Client.put_object(Body=content, Bucket=targetBucket, Key=targetKey, **extraArgs)  # type: ignore[arg-type]
+        await self._s3Client.put_object(Body=content, Bucket=targetBucket, Key=targetKey, **extraArgs)  # type: ignore[arg-type, ty:invalid-argument-type]
 
     async def upload_file(self, filePath: str, targetPath: str, accessControl: str | None = None, cacheControl: str | None = None, contentType: str | None = None) -> None:
         if not self._s3Client:
@@ -124,7 +124,7 @@ class S3Manager:
         sourceBucket, sourceKey = self._split_path_to_bucket_key(path=source)
         targetBucket, targetKey = self._split_path_to_bucket_key(path=target)
         extraArgs = self._get_extra_args(accessControl=accessControl, cacheControl=cacheControl, contentType=contentType or self._get_file_mimetype(fileName=targetKey))
-        await self._s3Client.copy_object(CopySource={'Bucket': sourceBucket, 'Key': sourceKey}, Bucket=targetBucket, Key=targetKey, MetadataDirective='REPLACE', **extraArgs)  # type: ignore[arg-type]
+        await self._s3Client.copy_object(CopySource={'Bucket': sourceBucket, 'Key': sourceKey}, Bucket=targetBucket, Key=targetKey, MetadataDirective='REPLACE', **extraArgs)  # type: ignore[arg-type, ty:invalid-argument-type]
 
     async def delete_file(self, filePath: str) -> None:
         if not self._s3Client:

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/store/database.py

@@ -5,6 +5,7 @@ from collections.abc import AsyncIterator
 from typing import TypeVar
 
 import sqlalchemy
+import sqlalchemy.exc
 from sqlalchemy.engine import Result
 from sqlalchemy.ext.asyncio import AsyncConnection
 from sqlalchemy.ext.asyncio import AsyncEngine
@@ -95,11 +96,15 @@ class Database:
             await self.disconnect()
             await self.connect()
 
-    async def execute(self, query: TypedReturnsRows[ResultType], connection: DatabaseConnection | None = None) -> Result[ResultType]:
+    @typing.overload
+    async def execute(self, query: TypedReturnsRows[ResultType], connection: DatabaseConnection | None = None) -> Result[ResultType]: ...
+    @typing.overload
+    async def execute(self, query: sqlalchemy.sql.Executable, connection: DatabaseConnection | None = None) -> Result[typing.Any]: ...  # type: ignore[explicit-any]
+    async def execute(self, query: sqlalchemy.sql.Executable, connection: DatabaseConnection | None = None) -> Result[typing.Any]:  # type: ignore[explicit-any]
         if not self._engine:
             raise InternalServerErrorException(message='Connection has not been established. Please called collect() first.')
         if not connection:
             connection = self._get_context_connection()
         if not connection:
             raise InternalServerErrorException(message='No connection found. Please provide a connection or call create_context_connection() for the context.')
-        return typing.cast(Result[ResultType], await connection.execute(statement=query))
+        return typing.cast(Result[typing.Any], await connection.execute(statement=query))  # type: ignore[explicit-any]

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/store/retriever.py

@@ -4,6 +4,7 @@ from collections.abc import Sequence
 from enum import Enum
 
 import sqlalchemy
+import sqlalchemy.sql.functions
 from sqlalchemy import Table
 from sqlalchemy.sql import Select
 

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/util/json_util.py

@@ -23,13 +23,13 @@ _HAS_LOGGED_FOR_SERIALIZATION_ERROR = False
 
 
 class DatetimeEncoder(json.JSONEncoder):
-    def default(self, obj: typing.Any) -> str:  # type: ignore[explicit-any]
-        if isinstance(obj, datetime.datetime):
-            return date_util.datetime_to_string(obj)
+    def default(self, o: typing.Any) -> str:  # type: ignore[explicit-any]
+        if isinstance(o, datetime.datetime):
+            return date_util.datetime_to_string(o)
         try:
-            return typing.cast(str, super().default(obj))
+            return typing.cast(str, super().default(o))
         except TypeError:
-            return str(obj)
+            return str(o)
 
 
 def dumpb(obj: Any) -> bytes:  # type: ignore[explicit-any]

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/core/web3/eth_client.py

@@ -222,7 +222,9 @@ class EthClientInterface:
             raise TransactionFailedException(transactionReceipt=transactionReceipt)
         return transactionReceipt
 
-    async def multicall(self, contractCalls: list[ContractCall], shouldUseMulticall3: bool = True, blockNumber: int | None = None) -> list[ListAny]:
+    async def multicall(self, contractCalls: list[ContractCall], shouldUseMulticall3: bool = True, blockNumber: int | None = None, shouldAllowFailures: bool = False) -> list[ListAny]:
+        # NOTE: when shouldAllowFailures=True, a call that reverts on-chain does not fail the whole batch -
+        # its slot in the returned list is [None] instead of a decoded value. Callers must check for that per-slot.
         multicall3Address = CHAIN_ID_MULTICALL3_ADDRESS_MAP.get(self.chainId) if shouldUseMulticall3 else None
         if not multicall3Address or not shouldUseMulticall3:
             results = await asyncio.gather(*[self.call(contractCall=contractCall, blockNumber=blockNumber) for contractCall in contractCalls])
@@ -233,7 +235,7 @@ class EthClientInterface:
             functionName='aggregate3',
             arguments={
                 'calls': [
-                    {'target': contractCall.toAddress, 'allowFailure': False, 'callData': chain_util.encode_transaction_data_by_name(contractAbi=contractCall.contractAbi, functionName=contractCall.functionName, arguments=contractCall.arguments)}
+                    {'target': contractCall.toAddress, 'allowFailure': shouldAllowFailures, 'callData': chain_util.encode_transaction_data_by_name(contractAbi=contractCall.contractAbi, functionName=contractCall.functionName, arguments=contractCall.arguments)}
                     for contractCall in contractCalls
                 ]
             },
@@ -453,16 +455,16 @@ class RestEthClient(EthClientInterface):
         if 'chainId' not in params:
             if chainId is None:
                 chainId = self.chainId
-            params['chainId'] = hex(chainId)  # type: ignore[typeddict-item]
+            params['chainId'] = hex(chainId)  # type: ignore[typeddict-item, ty:invalid-assignment]
         if 'nonce' not in params:
             if nonce is None:
                 nonce = await self.get_transaction_count(address=fromAddress)
-            params['nonce'] = hex(nonce)  # type: ignore[typeddict-item]
+            params['nonce'] = hex(nonce)  # type: ignore[typeddict-item, ty:invalid-assignment]
         if 'gas' not in params:
             if gas is None:
                 response = await self._make_request(method='eth_estimateGas', params=[params])
                 gas = int(response['result'], 16)
-            params['gas'] = hex(gas)  # type: ignore[typeddict-item]
+            params['gas'] = hex(gas)  # type: ignore[typeddict-item, ty:invalid-assignment]
         if 'gasPrice' not in params:
             if 'maxPriorityFeePerGas' not in params:
                 if maxPriorityFeePerGas is None:

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/kiba_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kiba-core
-Version: 0.5.3.dev51
+Version: 0.5.3.dev52
 Summary: Kiba Labs' python utilities
 Project-URL: repository, https://github.com/kibalabs/core-py
 Requires-Python: <4,>=3.12
@@ -9,33 +9,33 @@ Requires-Dist: aiofiles>=25.1.0
 Requires-Dist: pydantic>=2.13.4
 Requires-Dist: orjson>=3.11.9
 Provides-Extra: api
-Requires-Dist: fastapi>=0.139.1; extra == "api"
+Requires-Dist: fastapi>=0.141.1; extra == "api"
 Requires-Dist: uvicorn[standard]>=0.51.0; extra == "api"
 Provides-Extra: core-api
-Requires-Dist: starlette>=1.3.1; extra == "core-api"
+Requires-Dist: starlette>=1.6.0; extra == "core-api"
 Requires-Dist: uvicorn[standard]>=0.51.0; extra == "core-api"
 Provides-Extra: storage
-Requires-Dist: aiobotocore>=3.7.0; extra == "storage"
+Requires-Dist: aiobotocore>=3.9.0; extra == "storage"
 Provides-Extra: queues
-Requires-Dist: aiobotocore>=3.7.0; extra == "queues"
+Requires-Dist: aiobotocore>=3.9.0; extra == "queues"
 Provides-Extra: queue-sqs
-Requires-Dist: aiobotocore>=3.7.0; extra == "queue-sqs"
+Requires-Dist: aiobotocore>=3.9.0; extra == "queue-sqs"
 Provides-Extra: queue-aqs
 Requires-Dist: azure-storage-queue[aio]>=12.17.0; extra == "queue-aqs"
 Provides-Extra: queue-cosmos
-Requires-Dist: azure-cosmos>=4.16.2; extra == "queue-cosmos"
+Requires-Dist: azure-cosmos>=4.16.3; extra == "queue-cosmos"
 Provides-Extra: database-psql
-Requires-Dist: sqlalchemy[asyncio]>=2.0.51; extra == "database-psql"
+Requires-Dist: sqlalchemy[asyncio]>=2.0.52; extra == "database-psql"
 Requires-Dist: asyncpg>=0.31.0; extra == "database-psql"
 Provides-Extra: database-sqlite
-Requires-Dist: sqlalchemy[asyncio]>=2.0.51; extra == "database-sqlite"
+Requires-Dist: sqlalchemy[asyncio]>=2.0.52; extra == "database-sqlite"
 Requires-Dist: aiosqlite>=0.22.1; extra == "database-sqlite"
 Provides-Extra: requester
 Requires-Dist: httpx>=0.28.1; extra == "requester"
 Provides-Extra: web3
 Requires-Dist: web3>=7.16.0; extra == "web3"
 Provides-Extra: types
-Requires-Dist: types-aiobotocore[essential]>=3.7.0; extra == "types"
+Requires-Dist: types-aiobotocore[essential]>=3.9.0; extra == "types"
 Requires-Dist: types-aiofiles>=25.1.0.20260518; extra == "types"
 
 # Core-Py

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/kiba_core.egg-info/requires.txt

@@ -3,41 +3,41 @@ pydantic>=2.13.4
 orjson>=3.11.9
 
 [api]
-fastapi>=0.139.1
+fastapi>=0.141.1
 uvicorn[standard]>=0.51.0
 
 [core-api]
-starlette>=1.3.1
+starlette>=1.6.0
 uvicorn[standard]>=0.51.0
 
 [database-psql]
-sqlalchemy[asyncio]>=2.0.51
+sqlalchemy[asyncio]>=2.0.52
 asyncpg>=0.31.0
 
 [database-sqlite]
-sqlalchemy[asyncio]>=2.0.51
+sqlalchemy[asyncio]>=2.0.52
 aiosqlite>=0.22.1
 
 [queue-aqs]
 azure-storage-queue[aio]>=12.17.0
 
 [queue-cosmos]
-azure-cosmos>=4.16.2
+azure-cosmos>=4.16.3
 
 [queue-sqs]
-aiobotocore>=3.7.0
+aiobotocore>=3.9.0
 
 [queues]
-aiobotocore>=3.7.0
+aiobotocore>=3.9.0
 
 [requester]
 httpx>=0.28.1
 
 [storage]
-aiobotocore>=3.7.0
+aiobotocore>=3.9.0
 
 [types]
-types-aiobotocore[essential]>=3.7.0
+types-aiobotocore[essential]>=3.9.0
 types-aiofiles>=25.1.0.20260518
 
 [web3]

{kiba_core-0.5.3.dev51 → kiba_core-0.5.3.dev52}/makefile

@@ -10,14 +10,13 @@ list-outdated: install
 	@ pip list -o
 
 lint-check:
-	@ uv run lint-check --new ./core
+	@ uv run lint-check ./core
 
 lint-check-ci:
-	@ uv run lint-check --new ./core --output-file lint-check-results.json --output-format annotations
+	@ uv run lint-check ./core --output-file lint-check-results.json --output-format annotations
 
 lint-fix:
-	@ uv run isort --sl -l 1000 ./core
-	@ uv run lint-check --new --fix ./core
+	@ uv run lint-check --fix ./core
 
 type-check:
 	@ uv run type-check ./core