Update bin_public/app/chat_func.py

bin_public/app/chat_func.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 
 import urllib3
 
+import openai
 from tqdm import tqdm
 from duckduckgo_search import ddg
 from llama_func import *
@@ -460,4 +461,19 @@ def reduce_token_size(
         yield chatbot, history, construct_token_message(
             sum(token_count), stream=stream
         ), token_count
-    logging.info("减少token数量完毕")
+    logging.info("减少token数量完毕")
+
+
+def predict_davinci(api_key, input, temperature, history=None):
+    if history is None:
+        history = []
+    s = list(sum(history, ()))
+    s.append(input)
+    openai.api_key = api_key
+    response = openai.Completion.create(
+            engine="text-davinci-003",
+            prompt=s,
+            temperature=temperature,
+            max_tokens=2048,
+    )
+    return response.choices[0].text