From 4dc3bd35485f0c524dc26db308f72fefbb01d7cd Mon Sep 17 00:00:00 2001 From: lidia9 Date: Wed, 30 Apr 2025 12:18:47 +0200 Subject: [PATCH] Fix START_TIME import for tests - ci --- src/cai/util.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/cai/util.py b/src/cai/util.py index ca0af158..63cedaf5 100644 --- a/src/cai/util.py +++ b/src/cai/util.py @@ -21,6 +21,11 @@ from dataclasses import dataclass, field from typing import Dict, Optional import time +# Instead of direct import +try: + from cai.cli import START_TIME +except ImportError: + START_TIME = None # Shared stats tracking object to maintain consistent costs across calls @dataclass