Fix bug

Certora · Jan 9, 2025 · f5d0f2c · f5d0f2c
1 parent 9ca87b9
commit f5d0f2c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -45,9 +45,7 @@ jobs:
           poetry run ruff format --check src
 
       - name: Run unit tests
-        run: |
-          echo $ANTHROPIC_API_KEY
-          poetry run pytest src/quorum/tests --maxfail=1 --disable-warnings --tb=short
+        run: poetry run pytest src/quorum/tests --maxfail=1 --disable-warnings --tb=short
 
       - name: Run Quorum setup
         run: poetry run quorum setup --working-dir workdir

diff --git a/src/quorum/entry_points/implementations/ipfs_validator.py b/src/quorum/entry_points/implementations/ipfs_validator.py
@@ -74,10 +74,15 @@ def run_ipfs_validator(args: argparse.Namespace):
     # Initialize the IPFS Validation Chain
     ipfs_validation_chain = IPFSValidationChain()
 
-    # Execute the Chain
-    answer = ipfs_validation_chain.execute(
-        prompt_templates=args.prompt_templates, ipfs=ipfs, payload=payload
-    )
+    try:
+        # Execute the Chain
+        answer = ipfs_validation_chain.execute(
+            prompt_templates=args.prompt_templates, ipfs=ipfs, payload=payload
+        )
+    except Exception as e:
+        pp.pprint(f"Got and error while running LLM: {e}", pp.Colors.FAILURE)
+        pp.pprint("Verify your API key and try again.", pp.Colors.FAILURE)
+        return
 
     if answer.incompatibilities:
         pp.pprint("Found incompatibilities:", pp.Colors.FAILURE)