Add method to get jobs count from TEE pool contract (#38)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vana"
-version = "0.22.0"
+version = "0.23.0"
 description = ""
 authors = ["Tim Nunamaker <[email protected]>", "Volodymyr Isai <[email protected]>", "Kahtaf Alam <[email protected]>"]
 readme = "README.md"

vana/__init__.py

@@ -15,7 +15,7 @@
 # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 # DEALINGS IN THE SOFTWARE.
 
-__version__ = "0.22.0"
+__version__ = "0.23.0"
 
 import rich
 

vana/client.py

@@ -149,3 +149,20 @@ def claim(self):
         """
         claim_fn = self.tee_pool_contract.functions.claim()
         return self.chain_manager.send_transaction(claim_fn, self.wallet.hotkey)
+
+    def get_jobs_count(self) -> int:
+        """
+        Get the total number of jobs from the TEE Pool contract
+
+        Returns:
+            int: Total number of jobs in the queue
+        """
+        try:
+            # Call the TEE Pool contract's jobCount method
+            job_count_fn = self.tee_pool_contract.functions.jobsCount()
+            job_count = self.chain_manager.read_contract_fn(job_count_fn)
+            return job_count
+        except Exception as e:
+            vana.logging.error(f"Error getting job count: {str(e)}")
+            # Return 0 if unable to get count to avoid breaking progress calculation
+            return 0