update

Author: luoluoyuyu

python/functionstream-api-advanced/src/fs_api_advanced/keyed/keyed_aggregating_state.py

@@ -13,7 +13,7 @@
 from typing import Any, Generic, Optional, Protocol, Tuple, TypeVar
 
 from fs_api.store import ComplexKey, KvError, KvStore
-from fs_api_advanced.codec import Codec, PickleCodec, default_codec_for
+from fs_api_advanced.codec import Codec, default_codec_for
 
 T_agg = TypeVar("T_agg")
 ACC = TypeVar("ACC")
@@ -74,7 +74,9 @@ def from_context_auto_codec(
     ) -> "KeyedAggregatingStateFactory[T_agg, ACC, R]":
         """Create a KeyedAggregatingStateFactory with default accumulator codec from context and store name."""
         store = ctx.getOrCreateKVStore(store_name)
-        codec = default_codec_for(acc_type) if acc_type is not None else PickleCodec()
+        if acc_type is None:
+            raise KvError("keyed aggregating state from_context_auto_codec requires acc_type")
+        codec = default_codec_for(acc_type)
         return cls(store, key_group, codec, agg_func)
 
     def new_aggregating_state(

python/functionstream-api-advanced/src/fs_api_advanced/keyed/keyed_list_state.py

@@ -14,7 +14,7 @@
 from typing import Any, Generic, List, Optional, TypeVar
 
 from fs_api.store import ComplexKey, KvError, KvStore
-from fs_api_advanced.codec import Codec, PickleCodec, default_codec_for
+from fs_api_advanced.codec import Codec, default_codec_for
 
 V = TypeVar("V")
 
@@ -67,7 +67,9 @@ def from_context_auto_codec(
     ) -> "KeyedListStateFactory[V]":
         """Create a KeyedListStateFactory with default value codec from context and store name."""
         store = ctx.getOrCreateKVStore(store_name)
-        codec = default_codec_for(value_type) if value_type is not None else PickleCodec()
+        if value_type is None:
+            raise KvError("keyed list state from_context_auto_codec requires value_type")
+        codec = default_codec_for(value_type)
         return cls(store, key_group, codec)
 
     def new_keyed_list(self, key: bytes, namespace: bytes) -> "KeyedListState[V]":

python/functionstream-api-advanced/src/fs_api_advanced/keyed/keyed_map_state.py

@@ -13,7 +13,7 @@
 from typing import Any, Generic, Iterator, Optional, Tuple, TypeVar
 
 from fs_api.store import ComplexKey, KvError, KvStore
-from fs_api_advanced.codec import Codec, PickleCodec, BytesCodec, default_codec_for
+from fs_api_advanced.codec import Codec, default_codec_for
 
 from ._keyed_common import ensure_ordered_key_codec
 
@@ -69,14 +69,12 @@ def from_context_auto_codec(
     ) -> "KeyedMapStateFactory[MK, MV]":
         """Create a KeyedMapStateFactory with default codecs for MK and MV. Map key type must have an ordered default codec."""
         store = ctx.getOrCreateKVStore(store_name)
-        map_key_codec = (
-            default_codec_for(map_key_type) if map_key_type is not None else BytesCodec()
-        )
-        map_value_codec = (
-            default_codec_for(map_value_type)
-            if map_value_type is not None
-            else PickleCodec()
-        )
+        if map_key_type is None:
+            raise KvError("keyed map state from_context_auto_codec requires map_key_type")
+        if map_value_type is None:
+            raise KvError("keyed map state from_context_auto_codec requires map_value_type")
+        map_key_codec = default_codec_for(map_key_type)
+        map_value_codec = default_codec_for(map_value_type)
         return cls(store, key_group, map_key_codec, map_value_codec)
 
     def new_keyed_map(

python/functionstream-api-advanced/src/fs_api_advanced/keyed/keyed_priority_queue_state.py

@@ -13,7 +13,7 @@
 from typing import Any, Generic, Iterator, Optional, Tuple, TypeVar
 
 from fs_api.store import ComplexKey, KvError, KvStore
-from fs_api_advanced.codec import Codec, IntCodec, default_codec_for
+from fs_api_advanced.codec import Codec, default_codec_for
 
 from ._keyed_common import ensure_ordered_key_codec
 
@@ -62,7 +62,9 @@ def from_context_auto_codec(
     ) -> "KeyedPriorityQueueStateFactory[V]":
         """Create a KeyedPriorityQueueStateFactory with default value codec. V must have an ordered default codec."""
         store = ctx.getOrCreateKVStore(store_name)
-        codec = default_codec_for(item_type) if item_type is not None else IntCodec()
+        if item_type is None:
+            raise KvError("keyed priority queue state from_context_auto_codec requires item_type")
+        codec = default_codec_for(item_type)
         return cls(store, key_group, codec)
 
     def new_keyed_priority_queue(

python/functionstream-api-advanced/src/fs_api_advanced/keyed/keyed_reducing_state.py

@@ -13,7 +13,7 @@
 from typing import Any, Callable, Generic, Optional, Tuple, TypeVar
 
 from fs_api.store import ComplexKey, KvError, KvStore
-from fs_api_advanced.codec import Codec, PickleCodec, default_codec_for
+from fs_api_advanced.codec import Codec, default_codec_for
 
 V = TypeVar("V")
 
@@ -68,7 +68,9 @@ def from_context_auto_codec(
     ) -> "KeyedReducingStateFactory[V]":
         """Create a KeyedReducingStateFactory with default value codec from context and store name."""
         store = ctx.getOrCreateKVStore(store_name)
-        codec = default_codec_for(value_type) if value_type is not None else PickleCodec()
+        if value_type is None:
+            raise KvError("keyed reducing state from_context_auto_codec requires value_type")
+        codec = default_codec_for(value_type)
         return cls(store, key_group, codec, reduce_func)
 
     def new_reducing_state(

python/functionstream-api-advanced/src/fs_api_advanced/keyed/keyed_value_state.py

@@ -13,7 +13,7 @@
 from typing import Any, Generic, Optional, Tuple, TypeVar
 
 from fs_api.store import ComplexKey, KvError, KvStore
-from fs_api_advanced.codec import Codec, PickleCodec, default_codec_for
+from fs_api_advanced.codec import Codec, default_codec_for
 
 V = TypeVar("V")
 
@@ -59,7 +59,9 @@ def from_context_auto_codec(
     ) -> "KeyedValueStateFactory[V]":
         """Create a KeyedValueStateFactory with default value codec from context and store name."""
         store = ctx.getOrCreateKVStore(store_name)
-        codec = default_codec_for(value_type) if value_type is not None else PickleCodec()
+        if value_type is None:
+            raise KvError("keyed value state from_context_auto_codec requires value_type")
+        codec = default_codec_for(value_type)
         return cls(store, key_group, codec)