Update ai_generate.py

ai_generate.py

@@ -42,8 +42,7 @@ def generate_openai(text, model, openai_client):
         max_tokens=800,
         frequency_penalty=0.0
     )
-    print(response.choices[0].message)
-    return response.choices[0].message
+    return response.choices[0].message.content
 
 def generate(text, model, api):
     if model == "Llama 3":