add vcr

i-dot-ai · Nov 15, 2024 · 6c16776 · 6c16776
1 parent 41ab927
commit 6c16776
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/redbox-core/poetry.lock b/redbox-core/poetry.lock
diff --git a/redbox-core/pyproject.toml b/redbox-core/pyproject.toml
@@ -54,5 +54,6 @@ env_files = [
     ".env"
 ]
 markers = [
-    "ai: marks tests as using a live LLM (deselect with '-m \"not ai\"')"
+    "ai: marks tests as using a live LLM (deselect with '-m \"not ai\"')",
+    "vcr",
 ]
diff --git a/redbox-core/tests/test_tools.py b/redbox-core/tests/test_tools.py
@@ -3,6 +3,7 @@
 from uuid import UUID, uuid4
 
 import pytest
+import requests
 from elasticsearch import Elasticsearch
 from langchain_core.embeddings.fake import FakeEmbeddings
 from langchain_core.messages import HumanMessage
@@ -212,14 +213,14 @@ def test_wikipedia_tool():
         assert metadata.creator_type == ChunkCreatorType.wikipedia
 
 
-@pytest.mark.vcr()
 @pytest.mark.parametrize(
     "is_filter, relevant_return, query, keyword",
     [
         (False, False, "UK government use of AI", "artificial intelligence"),
         (True, True, "UK government use of AI", "artificial intelligence"),
     ],
 )
+@pytest.mark.vcr()
 def test_gov_filter_AI(is_filter, relevant_return, query, keyword):
     def run_tool(is_filter):
         tool = build_govuk_search_tool(num_results=1, filter=is_filter)