Spaces:
Running
Running
kudo1026
commited on
Commit
•
05de806
1
Parent(s):
4259577
make api key part of config
Browse files- app.py +1 -0
- gpt_dialogue.py +3 -3
app.py
CHANGED
@@ -85,6 +85,7 @@ def process_instruction_callback(inp_api_key, instruction, llm_name):
|
|
85 |
|
86 |
# get oepnai config
|
87 |
openai_config = get_openai_config(llm_name)
|
|
|
88 |
|
89 |
# get LLM response
|
90 |
|
|
|
85 |
|
86 |
# get oepnai config
|
87 |
openai_config = get_openai_config(llm_name)
|
88 |
+
openai_config['api_key'] = inp_api_key
|
89 |
|
90 |
# get LLM response
|
91 |
|
gpt_dialogue.py
CHANGED
@@ -4,7 +4,6 @@ import datetime
|
|
4 |
# import openai
|
5 |
# openai.api_key = os.getenv("OPENAI_API_KEY")
|
6 |
from openai import OpenAI
|
7 |
-
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
|
8 |
|
9 |
|
10 |
# HUGGINGFACE_MODELS = {
|
@@ -19,7 +18,8 @@ client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
|
|
19 |
|
20 |
|
21 |
class Dialogue:
|
22 |
-
def __init__(self, model='gpt-4', temperature=0, top_p=0.0, max_tokens=10, system_message='', load_path=None, save_path='chats', debug=False):
|
|
|
23 |
self.model = model
|
24 |
self.temperature = temperature
|
25 |
self.top_p = top_p
|
@@ -98,7 +98,7 @@ class Dialogue:
|
|
98 |
messages = self.pretext + user_message
|
99 |
# print('messages: ', messages)
|
100 |
if 'gpt' in self.model:
|
101 |
-
completion = client.chat.completions.create(
|
102 |
model=self.model,
|
103 |
messages=self.pretext + user_message,
|
104 |
temperature=self.temperature,
|
|
|
4 |
# import openai
|
5 |
# openai.api_key = os.getenv("OPENAI_API_KEY")
|
6 |
from openai import OpenAI
|
|
|
7 |
|
8 |
|
9 |
# HUGGINGFACE_MODELS = {
|
|
|
18 |
|
19 |
|
20 |
class Dialogue:
|
21 |
+
def __init__(self, api_key, model='gpt-4', temperature=0, top_p=0.0, max_tokens=10, system_message='', load_path=None, save_path='chats', debug=False):
|
22 |
+
self.client = OpenAI(api_key=api_key)
|
23 |
self.model = model
|
24 |
self.temperature = temperature
|
25 |
self.top_p = top_p
|
|
|
98 |
messages = self.pretext + user_message
|
99 |
# print('messages: ', messages)
|
100 |
if 'gpt' in self.model:
|
101 |
+
completion = self.client.chat.completions.create(
|
102 |
model=self.model,
|
103 |
messages=self.pretext + user_message,
|
104 |
temperature=self.temperature,
|