66from pydantic import BaseModel
77
88from tastytrade .order import (InstrumentType , NewComplexOrder , NewOrder ,
9- OrderStatus , PlacedComplexOrder , PlacedOrder ,
10- PlacedOrderResponse , PriceEffect )
9+ OrderAction , OrderStatus , PlacedComplexOrder ,
10+ PlacedOrder , PlacedOrderResponse , PriceEffect )
1111from tastytrade .session import ProductionSession , Session
1212from tastytrade .utils import (TastytradeError , TastytradeJsonDataclass ,
1313 today_in_new_york , validate_response )
@@ -335,7 +335,7 @@ class Transaction(TastytradeJsonDataclass):
335335 symbol : Optional [str ] = None
336336 instrument_type : Optional [InstrumentType ] = None
337337 underlying_symbol : Optional [str ] = None
338- action : Optional [str ] = None
338+ action : Optional [OrderAction ] = None
339339 quantity : Optional [Decimal ] = None
340340 price : Optional [Decimal ] = None
341341 regulatory_fees : Optional [Decimal ] = None
@@ -873,7 +873,7 @@ def get_live_complex_orders(
873873 def get_complex_order (
874874 self ,
875875 session : Session ,
876- order_id : str
876+ order_id : int
877877 ) -> PlacedComplexOrder :
878878 """
879879 Gets a complex order with the given ID.
@@ -894,7 +894,7 @@ def get_complex_order(
894894
895895 return PlacedComplexOrder (** data )
896896
897- def get_order (self , session : Session , order_id : str ) -> PlacedOrder :
897+ def get_order (self , session : Session , order_id : int ) -> PlacedOrder :
898898 """
899899 Gets an order with the given ID.
900900
@@ -913,7 +913,7 @@ def get_order(self, session: Session, order_id: str) -> PlacedOrder:
913913
914914 return PlacedOrder (** data )
915915
916- def delete_complex_order (self , session : Session , order_id : str ) -> None :
916+ def delete_complex_order (self , session : Session , order_id : int ) -> None :
917917 """
918918 Delete a complex order by ID.
919919
@@ -927,7 +927,7 @@ def delete_complex_order(self, session: Session, order_id: str) -> None:
927927 )
928928 validate_response (response )
929929
930- def delete_order (self , session : Session , order_id : str ) -> None :
930+ def delete_order (self , session : Session , order_id : int ) -> None :
931931 """
932932 Delete an order by ID.
933933
@@ -1136,7 +1136,7 @@ def place_complex_order(
11361136 def replace_order (
11371137 self ,
11381138 session : Session ,
1139- old_order_id : str ,
1139+ old_order_id : int ,
11401140 new_order : NewOrder
11411141 ) -> PlacedOrder :
11421142 """
0 commit comments