Skip to content

Commit 2d69509

Browse files
committed
minor fix
1 parent 1aa6abd commit 2d69509

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/integration/llms/test_API_Perplexity.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@ def test_api_key(self):
1717
def test_ask_using_messages__async_mode(self):
1818
model = "mistral-7b-instruct"
1919
system_prompt = "act like a calculator, just reply with the answer"
20-
user_prompt = "40+2"
21-
expected_response = ['', '42', '']
20+
user_prompt = "2+1"
21+
#expected_response = ['', '42', '']
2222
async_mode = True
2323
# expected_response = ['Sure', '! ', '40', ' + ', '2 =', ' 4', '2.', ''] for #'llama-2-70b-chat' #
2424
messages = [ { "role" : "system" ,
@@ -27,7 +27,7 @@ def test_ask_using_messages__async_mode(self):
2727
"content" : user_prompt}]
2828
generator = self.api_perplexity.ask_using_messages(messages, model=model, async_mode=async_mode)
2929

30-
assert '42' in list(generator)
30+
assert '3' in list(generator)
3131

3232

3333
def test_ask_using_messages__sync_mode(self):

0 commit comments

Comments
 (0)