OMGeeky/gpt-pilot
1
0
Fork 0
mirror of https://github.com/OMGeeky/gpt-pilot.git synced 2026-02-23 15:49:50 +01:00
Code Issues Packages Projects Releases Wiki Activity

fixed test for CI

Browse Source
This commit is contained in:
Nicholas Albion
2023-09-27 12:45:20 +10:00
parent 7a81b599c7
commit b0fea79c7b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pilot/utils/test_llm_connection.py
View File

@@ -101,8 +101,9 @@ class TestLlmConnection:
builtins.print, ipc_client_instance = get_custom_print({})
@patch('utils.llm_connection.requests.post')
def test_stream_gpt_completion(self, mock_post):
def test_stream_gpt_completion(self, mock_post, monkeypatch):
# Given streaming JSON response
monkeypatch.setenv('OPENAI_API_KEY', 'secret')
deltas = ['{', '\\n',
' \\"foo\\": \\"bar\\",', '\\n',
' \\"prompt\\": \\"Hello\\",', '\\n',