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

model = os.getenv('MODEL_NAME', 'gpt-4') - which makes sense as CI doesn't have my env

Browse Source
This commit is contained in:
Nicholas Albion
2023-10-04 15:50:45 +11:00
parent 366e88dc4d
commit 9251c1831e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pilot/utils/llm_connection.py
View File

@@ -261,12 +261,12 @@ def stream_gpt_completion(data, req_type, project):
# print(yellow("Stream response from OpenAI:"))
# Configure for the selected ENDPOINT
model = os.getenv('MODEL_NAME')
model = os.getenv('MODEL_NAME', 'gpt-4')
endpoint = os.getenv('ENDPOINT')
logger.info(f'> Request model: {model} ({data["model"]}) messages: {data["messages"]}')
logger.info(f'##### build endpoint...')
logger.info(f'##### build endpoint for {endpoint}')
if endpoint == 'AZURE':
# If yes, get the AZURE_ENDPOINT from .ENV file
@@ -292,7 +292,7 @@ def stream_gpt_completion(data, req_type, project):
'Authorization': 'Bearer ' + os.getenv('OPENAI_API_KEY')
}
logger.info(f'##### 0.1, endpoint: {endpoint}, headers: {headers}')
logger.info(f'##### 0.1, endpoint_url: {endpoint_url}, headers: {headers}')
response = requests.post(
endpoint_url,