Spaces:

hkunlp
/

Binder

Runtime error

tianbaoxiexxx commited on Apr 17, 2023

Commit

9da50da

1 Parent(s): 9a01777

Update nsql/qa_module/openai_qa.py

Files changed (1) hide show

nsql/qa_module/openai_qa.py CHANGED Viewed

@@ -34,7 +34,7 @@ class OpenAIQAModel(object):
         self.db_mapping_token = "\t"
     def call_openai_api_completion(self, prompt):
-        completion = self.generator._call_codex_api(engine="code-davinci-002",
                                                     prompt=prompt,
                                                     max_tokens=max_tokens,
                                                     temperature=0,

         self.db_mapping_token = "\t"
     def call_openai_api_completion(self, prompt):
+        completion = self.generator._call_codex_api(engine="text-davinci-002",
                                                     prompt=prompt,
                                                     max_tokens=max_tokens,
                                                     temperature=0,