OMGeeky/gpt-pilot
1
0
Fork 0
mirror of https://github.com/OMGeeky/gpt-pilot.git synced 2025-12-30 16:13:32 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #39 from coolxeo/patch-1

Browse Source 
Add OPENAI_MODEL env var
This commit is contained in:
zvone187
2023-09-05 22:33:20 +02:00
committed by GitHub
parent 38d56277ff af6a972cba
commit dc0533f67b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/utils/llm_connection.py
View File

@@ -94,7 +94,7 @@ def create_gpt_chat_completion(messages: List[dict], req_type, min_tokens=MIN_TO
raise ValueError(f'Too many tokens in messages: {tokens_in_messages}. Please try a different test.')
gpt_data = {
'model': 'gpt-4',
'model': os.getenv('OPENAI_MODEL', 'gpt-4'),
'n': 1,
'max_tokens': min(4096, MAX_GPT_MODEL_TOKENS - tokens_in_messages),
'temperature': 1,