tianbaoxiexxx commited on
Commit
9da50da
1 Parent(s): 9a01777

Update nsql/qa_module/openai_qa.py

Browse files
Files changed (1) hide show
  1. nsql/qa_module/openai_qa.py +1 -1
nsql/qa_module/openai_qa.py CHANGED
@@ -34,7 +34,7 @@ class OpenAIQAModel(object):
34
  self.db_mapping_token = "\t"
35
 
36
  def call_openai_api_completion(self, prompt):
37
- completion = self.generator._call_codex_api(engine="code-davinci-002",
38
  prompt=prompt,
39
  max_tokens=max_tokens,
40
  temperature=0,
 
34
  self.db_mapping_token = "\t"
35
 
36
  def call_openai_api_completion(self, prompt):
37
+ completion = self.generator._call_codex_api(engine="text-davinci-002",
38
  prompt=prompt,
39
  max_tokens=max_tokens,
40
  temperature=0,