from agents.offline_agent import QLearningAgent

class GetQValueUseCase:
    def execute(self, agent: QLearningAgent, state: int, action: int) -> float:
        return agent.q_table[state, action]