ChatCompletion

This commit is contained in:
钱忱 2023-11-08 15:15:01 +08:00
parent 74f94da64a
commit 59965ab0bd

View File

@ -69,7 +69,7 @@ class OpenAIModel(ModelBackend):
num_max_completion_tokens = num_max_token - num_prompt_tokens
self.model_config_dict['max_tokens'] = num_max_completion_tokens
print("using model: {}".format(self.model_type.value))
response = openai.chat.completions.create(*args, **kwargs,
response = openai.ChatCompletion.create(*args, **kwargs,
model=self.model_type.value,
**self.model_config_dict)
cost = prompt_cost(