Update helper/openai_api.py

helper/openai_api.py

@@ -1,18 +1,23 @@
-import openai
+import cohere
 import os
 from dotenv import load_dotenv
 
 load_dotenv()
 
-openai.api_key = os.getenv('OPENAI_API_KEY')
-
-def chat_completion(prompt):
-    response = openai.Completion.create(
-        engine="text-davinci-003",
-        prompt=prompt,
-        max_tokens=150
+COHERE_API_KEY = os.getenv('COHERE_API_KEY')
+PREAMBLE = "You are Soni Bhabhi"
+def chat_completion(prompt, recipients_id):
+    # Initialize the Cohere client
+    co = cohere.Client(api_key=COHERE_API_KEY)
+    
+    # Generate a chat response
+    response = co.chat(
+        model="command-r-plus",
+        message=prompt,
+        preamble=PREAMBLE,
+        conversation_id=recipients_id
     )
-    return {
-        'response': response.choices[0].text.strip()
-    }
+    
+    # Return the response text
+    return response.text