Skip to content

Commit a1a869b

Browse files
committed
fix; api path
1 parent bea6acf commit a1a869b

File tree

4 files changed

+8
-3
lines changed

4 files changed

+8
-3
lines changed

tests/models/test_groq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ class MockGroq:
9797
completions: MockChatCompletion | Sequence[MockChatCompletion] | None = None
9898
stream: Sequence[MockChatCompletionChunk] | Sequence[Sequence[MockChatCompletionChunk]] | None = None
9999
index: int = 0
100-
base_url: str = 'https://api.groq.com/openai/v1'
100+
base_url: str = 'https://api.groq.com'
101101

102102
@cached_property
103103
def chat(self) -> Any:

tests/models/test_huggingface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ class MockHuggingFace:
8080
stream: Sequence[MockStreamEvent] | Sequence[Sequence[MockStreamEvent]] | None = None
8181
index: int = 0
8282
chat_completion_kwargs: list[dict[str, Any]] = field(default_factory=list)
83-
model: str = 'hf-model'
83+
model: str = 'https://api-inference.huggingface.co'
8484

8585
@cached_property
8686
def chat(self) -> Any:

tests/models/test_mistral.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@
7272
@dataclass
7373
class MockSdkConfiguration:
7474
def get_server_details(self) -> tuple[str, ...]:
75-
return ('https://api.mistral.ai/v1',)
75+
return ('https://api.mistral.ai',)
7676

7777

7878
@dataclass

tests/test_temporal.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1856,6 +1856,7 @@ async def test_temporal_agent_with_hitl_tool(allow_model_requests: None, client:
18561856
model_name=IsStr(),
18571857
timestamp=IsDatetime(),
18581858
provider_name='openai',
1859+
provider_url='https://api.openai.com/v1/',
18591860
provider_details={'finish_reason': 'tool_calls'},
18601861
provider_response_id=IsStr(),
18611862
finish_reason='tool_call',
@@ -1898,6 +1899,7 @@ async def test_temporal_agent_with_hitl_tool(allow_model_requests: None, client:
18981899
model_name='gpt-4o-2024-08-06',
18991900
timestamp=IsDatetime(),
19001901
provider_name='openai',
1902+
provider_url='https://api.openai.com/v1/',
19011903
provider_details={'finish_reason': 'stop'},
19021904
provider_response_id=IsStr(),
19031905
finish_reason='stop',
@@ -1975,6 +1977,7 @@ async def test_temporal_agent_with_model_retry(allow_model_requests: None, clien
19751977
model_name='gpt-4o-2024-08-06',
19761978
timestamp=IsDatetime(),
19771979
provider_name='openai',
1980+
provider_url='https://api.openai.com/v1/',
19781981
provider_details={'finish_reason': 'tool_calls'},
19791982
provider_response_id=IsStr(),
19801983
finish_reason='tool_call',
@@ -2012,6 +2015,7 @@ async def test_temporal_agent_with_model_retry(allow_model_requests: None, clien
20122015
model_name='gpt-4o-2024-08-06',
20132016
timestamp=IsDatetime(),
20142017
provider_name='openai',
2018+
provider_url='https://api.openai.com/v1/',
20152019
provider_details={'finish_reason': 'tool_calls'},
20162020
provider_response_id=IsStr(),
20172021
finish_reason='tool_call',
@@ -2043,6 +2047,7 @@ async def test_temporal_agent_with_model_retry(allow_model_requests: None, clien
20432047
model_name='gpt-4o-2024-08-06',
20442048
timestamp=IsDatetime(),
20452049
provider_name='openai',
2050+
provider_url='https://api.openai.com/v1/',
20462051
provider_details={'finish_reason': 'stop'},
20472052
provider_response_id=IsStr(),
20482053
finish_reason='stop',

0 commit comments

Comments
 (0)