Skip to content

Commit daf6908

Browse files
test: Update coverage improvement test for tests/graphs/abstract_graph_test.py
1 parent f1abfe6 commit daf6908

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/graphs/abstract_graph_test.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -67,14 +67,11 @@ class TestAbstractGraph:
6767
),
6868
({"model": "ollama/llama2"}, ChatOllama),
6969
({"model": "oneapi/qwen-turbo", "api_key": "oneapi-api-key"}, OneApi),
70-
(
71-
{"model": "deepseek/deepseek-coder", "api_key": "deepseek-api-key"},
72-
DeepSeek,
73-
),
7470
(
7571
{
7672
"model": "bedrock/anthropic.claude-3-sonnet-20240229-v1:0",
7773
"region_name": "IDK",
74+
"temperature": 0.7
7875
},
7976
ChatBedrock,
8077
),
@@ -134,6 +131,7 @@ def test_create_llm_unknown_provider(self):
134131
"model": "bedrock/anthropic.claude-3-sonnet-20240229-v1:0",
135132
"region_name": "IDK",
136133
"rate_limit": {"requests_per_second": 1},
134+
"temperature": 0.7
137135
},
138136
ChatBedrock,
139137
),

0 commit comments

Comments
 (0)