From 768e6559479eedf44c02389312d789a14d4a1cbf Mon Sep 17 00:00:00 2001
From: Zishan <zishan@Zishans-MacBook-Pro.local>
Date: Wed, 12 Feb 2025 17:32:10 +0100
Subject: [PATCH] adjust pytest

---
 proxy/routes/anthropic.py                     |  2 +-
 .../anthropic/test_claude_agent_streaming.py  | 11 ++++++++---
 tests/anthropic/test_claude_weather_agent.py  | 19 +++++++++++++------
 3 files changed, 22 insertions(+), 10 deletions(-)

diff --git a/proxy/routes/anthropic.py b/proxy/routes/anthropic.py
index a7f8173..e320ac9 100644
--- a/proxy/routes/anthropic.py
+++ b/proxy/routes/anthropic.py
@@ -52,7 +52,7 @@ def validate_headers(
 @proxy.post(
     "/{dataset_name}/anthropic/{endpoint:path}",
     dependencies=[Depends(validate_headers)],
-) 
+)
 async def anthropic_proxy(
     dataset_name: str,
     endpoint: str,
diff --git a/tests/anthropic/test_claude_agent_streaming.py b/tests/anthropic/test_claude_agent_streaming.py
index cc77249..cb7e3a8 100644
--- a/tests/anthropic/test_claude_agent_streaming.py
+++ b/tests/anthropic/test_claude_agent_streaming.py
@@ -6,9 +6,14 @@ import os
 # from invariant import testing
 import datetime
 import pytest
+import sys
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-@pytest.mark.skipif(not os.getenv("ANTHROPIC_API_KEY"),reason="Anthropic API keys not set")
-def test_streaming_response_without_toolcall():    
+from util import *  # needed for pytest fixtures
+
+pytest_plugins = ("pytest_asyncio")
+@pytest.mark.skipif(not os.getenv("ANTHROPIC_API_KEY"), reason="No ANTHROPIC_API_KEY set")
+def test_streaming_response_without_toolcall(proxy_url):    
     # Example queries
     dataset_name = "claude_streaming_agent_test" + str(datetime.datetime.now().strftime("%Y%m%d%H%M%S"))
     invariant_api_key = os.environ.get("INVARIANT_API_KEY")
@@ -19,7 +24,7 @@ def test_streaming_response_without_toolcall():
                 "Invariant-Authorization": f"Bearer {invariant_api_key}"
             },
     ),
-    base_url=f"http://localhost/api/v1/proxy/{dataset_name}/anthropic",
+    base_url=f"{proxy_url}/api/v1/proxy/{dataset_name}/anthropic",
     )
 
     cities = ["Zurich", "New York", "London"]
diff --git a/tests/anthropic/test_claude_weather_agent.py b/tests/anthropic/test_claude_weather_agent.py
index 53cbccc..ad3d2ee 100644
--- a/tests/anthropic/test_claude_weather_agent.py
+++ b/tests/anthropic/test_claude_weather_agent.py
@@ -5,10 +5,17 @@ from typing import Dict
 import anthropic
 import pytest
 from httpx import Client
+import sys
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from util import *  # needed for pytest fixtures
+
+pytest_plugins = ("pytest_asyncio",)
 
 
 class WeatherAgent:
-    def __init__(self):
+    def __init__(self,proxy_url):
+        # proxy_url = "http://localhost"
         dataset_name = "claude_weather_agent_test" + str(
             datetime.datetime.now().strftime("%Y%m%d%H%M%S")
         )
@@ -17,11 +24,11 @@ class WeatherAgent:
             http_client=Client(
                 headers={"Invariant-Authorization": f"Bearer {invariant_api_key}"},
             ),
-            base_url=f"http://localhost/api/v1/proxy/{dataset_name}/anthropic",
+            base_url=f"{proxy_url}/api/v1/proxy/{dataset_name}/anthropic",
         )
         self.get_weather_function = {
             "name": "get_weather",
-            "description": "Get the current weather in a given locatiofn",
+            "description": "Get the current weather in a given location",
             "input_schema": {
                 "type": "object",
                 "properties": {
@@ -87,11 +94,11 @@ class WeatherAgent:
         return response
 
 
-@pytest.mark.skipif(not os.getenv("ANTHROPIC_API_KEY"),reason="Anthropic API keys not set")
-def test_proxy_response():
+@pytest.mark.skipif(not os.getenv("ANTHROPIC_API_KEY"), reason="No ANTHROPIC_API_KEY set")
+def test_proxy_response(proxy_url):
     """Test the proxy response for the weather agent."""
     # Initialize agent with Anthropic API key
-    weather_agent = WeatherAgent()
+    weather_agent = WeatherAgent(proxy_url)
 
     # Example queries
     queries = [