Update code & pass test

2023-09-23 13:19:18 +08:00
parent 85259ca408
commit 2be2ef3f60
5 changed files with 20 additions and 14 deletions
@@ -106,14 +106,14 @@ class OpenAI_ComputeNode(ComputeNode):
                    logger.info(f"call openai {mode_name} prompts: {prompts}")
                    resp = openai.ChatCompletion.create(model=mode_name,
                                                    messages=prompts,
-                                                    max_tokens=result_token,
+                                                    #max_tokens=result_token,
                                                    temperature=0.7)
                else:
                    logger.info(f"call openai {mode_name} prompts: {prompts} functions: {json.dumps(llm_inner_functions)}")
                    resp = openai.ChatCompletion.create(model=mode_name,
                                                        messages=prompts,
                                                        functions=llm_inner_functions,
-                                                        max_tokens=result_token,
+                                                        #max_tokens=result_token,
                                                        temperature=0.7) # TODO: add temperature to task params?