shukdevdatta123 commited on
Commit
e021db4
·
verified ·
1 Parent(s): 908fcb9

Update generate_answer.py

Browse files
Files changed (1) hide show
  1. generate_answer.py +4 -3
generate_answer.py CHANGED
@@ -22,12 +22,13 @@ openai.api_key = api_key
22
 
23
  def base_model_chatbot(messages):
24
  system_message = [
25
- {"role": "system", "content": "You are an helpful AI chatbot, that answers questions asked by User."}
26
  ]
27
  messages = system_message + messages
28
  response = openai.ChatCompletion.create(
29
- model="gpt-3.5-turbo", # Ensure the model is specified correctly
30
- messages=messages
 
31
  )
32
  return response.choices[0].message['content']
33
 
 
22
 
23
  def base_model_chatbot(messages):
24
  system_message = [
25
+ {"role": "system", "content": "You are a helpful AI chatbot, that answers questions asked by User."}
26
  ]
27
  messages = system_message + messages
28
  response = openai.ChatCompletion.create(
29
+ model="gpt-3.5-turbo",
30
+ messages=messages,
31
+ max_tokens=2048 # Increase max_tokens limit (or adjust based on your needs)
32
  )
33
  return response.choices[0].message['content']
34