Merge branch 'main' into patch-2

aiwaves-cn · Jun 17, 2023 · e32c9d5 · e32c9d5
2 parents c65c7b1 + dbeea84
commit e32c9d5
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/utils.py b/utils.py
@@ -7,6 +7,7 @@
 )
 
 def get_api_response(content: str, max_tokens=None):
+
     chat = ChatOpenAI(
         openai_api_key=OPENAI_API_KEY,
         #model='gpt-3',
@@ -18,6 +19,18 @@ def get_api_response(content: str, max_tokens=None):
         #model='gpt-4-0613',
         #model='gpt-4-32k-0613',
         temperature=0.5,
+
+
+    response = openai.ChatCompletion.create(
+        model='gpt-3.5-turbo-16k-0613',
+        messages=[{
+            'role': 'system',
+            'content': 'You are a helpful and creative assistant for writing novel.'
+        }, {
+            'role': 'user',
+            'content': content,
+        }],
+        temperature=0.5,  
         max_tokens=max_tokens
     )
     response = None