add more keys for different llm providers

AgentOps-AI · Jan 13, 2025 · 751873b · 751873b
1 parent d83210b
commit 751873b
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/tests/fixtures/vcr.py b/tests/fixtures/vcr.py
@@ -57,6 +57,25 @@ def vcr_config():
         ("x-ratelimit-remaining-tokens", "REDACTED"),
         ("x-ratelimit-reset-requests", "REDACTED"),
         ("x-ratelimit-reset-tokens", "REDACTED"),
+        # Mistral headers
+        ("x-mistral-api-key", "REDACTED"),
+        # Groq headers
+        ("x-groq-api-key", "REDACTED"),
+        # LiteLLM headers
+        ("x-litellm-api-key", "REDACTED"),
+        # Ollama headers
+        ("x-ollama-api-key", "REDACTED"),
+        # TaskWeaver headers
+        ("x-taskweaver-api-key", "REDACTED"),
+        # Additional provider version headers
+        ("anthropic-version", "REDACTED"),
+        ("cohere-version", "REDACTED"),
+        ("x-stainless-lang", "REDACTED"),
+        ("x-stainless-arch", "REDACTED"),
+        ("x-stainless-os", "REDACTED"),
+        ("x-stainless-async", "REDACTED"),
+        ("x-stainless-runtime", "REDACTED"),
+        ("x-stainless-runtime-version", "REDACTED"),
     ]
 
     def filter_response_headers(response):