Update openai_requests.py
Browse files- openai_requests.py +1 -1
openai_requests.py
CHANGED
@@ -4,7 +4,7 @@ import requests
|
|
4 |
import json
|
5 |
import math
|
6 |
|
7 |
-
def get_completion_from_openai(prompt, max_tokens = None
|
8 |
url = os.getenv('OPENAI_COMPLETION_URL')
|
9 |
|
10 |
headers = {
|
|
|
4 |
import json
|
5 |
import math
|
6 |
|
7 |
+
def get_completion_from_openai(prompt,model, max_tokens = None):
|
8 |
url = os.getenv('OPENAI_COMPLETION_URL')
|
9 |
|
10 |
headers = {
|