From ba1c0f7b38dceb8afee8bc0c105f41d03ebcf652 Mon Sep 17 00:00:00 2001
From: AI Christianson <ai.christianson@christianson.ai>
Date: Wed, 19 Feb 2025 20:06:41 -0500
Subject: [PATCH] fix tests

---
 tests/ra_aid/test_llm.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/tests/ra_aid/test_llm.py b/tests/ra_aid/test_llm.py
index 5c7c822..141765b 100644
--- a/tests/ra_aid/test_llm.py
+++ b/tests/ra_aid/test_llm.py
@@ -120,6 +120,10 @@ def test_initialize_expert_openrouter(clean_env, mock_openai, monkeypatch):
         temperature=0,
         timeout=180,
         max_retries=5,
+        default_headers={
+            "HTTP-Referer": "https://ra-aid.ai",
+            "X-Title": "RA.Aid"
+        }
     )
 
 
@@ -215,6 +219,10 @@ def test_initialize_openrouter(clean_env, mock_openai):
         temperature=0.7,
         timeout=180,
         max_retries=5,
+        default_headers={
+            "HTTP-Referer": "https://ra-aid.ai",
+            "X-Title": "RA.Aid"
+        }
     )
 
 
@@ -356,6 +364,10 @@ def test_explicit_temperature(clean_env, mock_openai, mock_anthropic, mock_gemin
         temperature=test_temp,
         timeout=180,
         max_retries=5,
+        default_headers={
+            "HTTP-Referer": "https://ra-aid.ai",
+            "X-Title": "RA.Aid"
+        }
     )
 
 
@@ -643,4 +655,8 @@ def test_initialize_openrouter_deepseek(
         temperature=0.7,
         timeout=180,
         max_retries=5,
+        default_headers={
+            "HTTP-Referer": "https://ra-aid.ai",
+            "X-Title": "RA.Aid"
+        }
     )