Update types

glifio · Jan 19, 2024 · 61d9266 · 61d9266
1 parent b153010
commit 61d9266
Show file tree

Hide file tree

Showing 3 changed files with 56 additions and 57 deletions.
diff --git a/mock/FEVMQueries.go b/mock/FEVMQueries.go
diff --git a/sdk/agent_queries.go b/sdk/agent_queries.go
@@ -375,7 +375,7 @@ func (q *fevmQueries) AgentVersion(ctx context.Context, agentAddr common.Address
 	return agentVersion, deployerVersion, nil
 }
 
-func (q *fevmQueries) AgentCollateralStats(ctx context.Context, agentAddr common.Address, blockNumber *big.Int) (terminate.PreviewAgentTerminationSummary, error) {
+func (q *fevmQueries) PreviewAgentTerminationQuick(ctx context.Context, agentAddr common.Address) (terminate.PreviewAgentTerminationSummary, error) {
 	agentID, err := q.AgentID(ctx, agentAddr)
 	if err != nil {
 		return terminate.PreviewAgentTerminationSummary{}, err
@@ -395,7 +395,7 @@ func (q *fevmQueries) AgentCollateralStats(ctx context.Context, agentAddr common
 		vestingBal.Add(vestingBal, miner.Vesting)
 	}
 
-	agentLiquidFIL, err := q.AgentLiquidAssets(ctx, agentAddr, blockNumber)
+	agentLiquidFIL, err := q.AgentLiquidAssets(ctx, agentAddr, nil)
 	if err != nil {
 		return terminate.PreviewAgentTerminationSummary{}, err
 	}

diff --git a/types/types.go b/types/types.go
@@ -37,8 +37,8 @@ type FEVMQueries interface {
 	AgentPrincipal(ctx context.Context, agentAddr common.Address, blockNumber *big.Int) (*big.Int, error)
 	AgentInterestOwed(ctx context.Context, agentAddr common.Address, tsk *filtypes.TipSet) (*big.Int, error)
 	AgentFaultyEpochStart(ctx context.Context, agentAddr common.Address) (*big.Int, error)
-	AgentCollateralStats(ctx context.Context, agentAddr common.Address, blockNumber *big.Int) (terminate.PreviewAgentTerminationSummary, error)
 	PreviewAgentTermination(ctx context.Context, agentAddr common.Address, tipset *filtypes.TipSet) (terminate.PreviewAgentTerminationSummary, error)
+	PreviewAgentTerminationQuick(ctx context.Context, agentAddr common.Address) (terminate.PreviewAgentTerminationSummary, error)
 	// agent factory methods
 	AgentFactoryAgentCount(ctx context.Context, blockNumber *big.Int) (*big.Int, error)
 	// infinity pool methods