DronA23 commited on
Commit
a2f3707
·
verified ·
1 Parent(s): d6ae916

Create helper.py

Browse files
Files changed (1) hide show
  1. helper.py +23 -0
helper.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import openai
3
+ from dotenv import load_dotenv
4
+
5
+ load_dotenv()
6
+ openai.api_key = os.getenv("OPENAI_API_KEY")
7
+
8
+ def generate_reply(message: str, history: list[tuple[str, str]]) -> tuple[str, list[tuple[str, str]]]:
9
+ """
10
+ Generic OpenAI chat reply.
11
+ """
12
+ messages = [{"role": "system", "content": "You are kind and caring."}]
13
+ for u, b in history:
14
+ messages.append({"role": "user", "content": u})
15
+ messages.append({"role": "assistant", "content": b})
16
+ messages.append({"role": "user", "content": message})
17
+
18
+ resp = openai.ChatCompletion.create(
19
+ model="gpt-4o-mini",
20
+ messages=messages
21
+ )
22
+ reply = resp.choices[0].message.content
23
+ return reply, history + [(message, reply)]