Dagfinn1962 commited on
Commit
5d73a34
1 Parent(s): c25307a

Update src/llm_boilers.py

Browse files
Files changed (1) hide show
  1. src/llm_boilers.py +2 -4
src/llm_boilers.py CHANGED
@@ -74,7 +74,7 @@ def gpt(
74
  temperature (float, optional): The value used to modulate the next token probabilities.
75
  Defaults to 1.0
76
  """
77
- conversation = prompt.split("")
78
 
79
  messages = []
80
  for turn in conversation:
@@ -115,11 +115,9 @@ def gpt(
115
  return chat_completion
116
 
117
 
118
- MODEL_FUNCTIONS.append(gpt)
119
-
120
  # Define the model and its parameters
121
  model_id = "dfurman/chat-gpt-3.5-turbo"
122
- openai_key = "<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>"
123
 
124
  model = llm_boiler(model_id, openai_key)
125
 
 
74
  temperature (float, optional): The value used to modulate the next token probabilities.
75
  Defaults to 1.0
76
  """
77
+ conversation = prompt.split("\n")
78
 
79
  messages = []
80
  for turn in conversation:
 
115
  return chat_completion
116
 
117
 
 
 
118
  # Define the model and its parameters
119
  model_id = "dfurman/chat-gpt-3.5-turbo"
120
+ openai_key = os.getenv("API_KEY")
121
 
122
  model = llm_boiler(model_id, openai_key)
123