euler314 commited on
Commit
f2ac9f8
·
verified ·
1 Parent(s): e0df0e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -83,9 +83,12 @@ def prepare_api_params(messages, model_name):
83
  }
84
 
85
  # Add the appropriate token parameter based on model
86
- if config["param_name"] == "max_completion_tokens":
87
- api_params["max_completion_tokens"] = config["max_tokens"]
 
 
88
  else:
 
89
  api_params["max_tokens"] = config["max_tokens"]
90
 
91
  return api_params, config
 
83
  }
84
 
85
  # Add the appropriate token parameter based on model
86
+ param_name = config["param_name"]
87
+ if param_name == "max_completion_tokens":
88
+ # Use the max_completion_tokens value from config
89
+ api_params["max_completion_tokens"] = config["max_completion_tokens"]
90
  else:
91
+ # Use the max_tokens value for other models
92
  api_params["max_tokens"] = config["max_tokens"]
93
 
94
  return api_params, config