Skip to content

Commit b031e0c

Browse files
committed
Local II
1 parent a9f3897 commit b031e0c

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/test_interpreter.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,8 @@ def test_generator():
2727
Sends two messages, makes sure everything is correct with display both on and off.
2828
"""
2929

30+
interpreter.llm.model = "gpt-4"
31+
3032
for tests in [
3133
{"query": "What's 38023*40334? Use Python", "display": True},
3234
{"query": "What's 2334*34335555? Use Python", "display": True},
@@ -129,7 +131,7 @@ async def test_fastapi_server():
129131
# Sending POST request
130132
post_url = "http://localhost:8000/settings"
131133
settings = {
132-
"model": "gpt-4-turbo",
134+
"model": "gpt-4o",
133135
"messages": [
134136
{
135137
"role": "user",
@@ -183,7 +185,7 @@ async def test_fastapi_server():
183185
# Send another POST request
184186
post_url = "http://localhost:8000/settings"
185187
settings = {
186-
"model": "gpt-4-turbo",
188+
"model": "gpt-4o",
187189
"messages": [
188190
{
189191
"role": "user",

0 commit comments

Comments
 (0)