npcJensen commited on
Commit
1d3deae
·
verified ·
1 Parent(s): b723f75

Update chatbot_memory.py

Browse files
Files changed (1) hide show
  1. chatbot_memory.py +28 -80
chatbot_memory.py CHANGED
@@ -1,81 +1,29 @@
1
- '''from openai import OpenAI
2
- from dotenv import load_dotenv
3
- import os
4
- from prompts import ai_motivational_speaker
5
-
6
- load_dotenv()
7
- api_key = os.getenv("GEMINI_API_KEY")
8
- model="gemini-2.5-flash-lite"
9
- base_url="https://generativelanguage-googleapis.com/v1beta/openai/"
10
-
11
- client= OpenAI(base_url=base_url, api_key=api_key)
12
-
13
- ai_motivational_speaker = ai_motivational_speaker
14
- '''
15
- '''
16
- def get_response(message, history):
17
- messages=({"role":"system","content":ai_motivational_speaker})
18
- messages.extend(history)
19
- messages.append({"role":"user","content":message})
20
- response = client.chat.completions.create(
21
- model=model,
22
- messages = messages)
23
-
24
- ai_response = response.choices[0].message.content
25
- return ai_response
26
-
27
- print(get_response(" hh h hhh h ", []))*'''
28
- '''
29
- def get_response(message, history):
30
- # Start messages with the system prompt to set the AI's persona
31
- messages = [{"role": "system", "content": ai_motivational_speaker}]
32
-
33
- # Extend the messages list with the existing chat history
34
- messages.extend(history)
35
-
36
- # Add the current user's message to the conversation
37
- messages.append({"role": "user", "content": message})
38
-
39
- # Call the OpenAI API to get a completion from the Gemini model
40
- response = client.chat.completions.create(model="gemini-2.5-flash", messages=messages)
41
- # Extract the AI's response content
42
- ai_response = response.choices[0].message.content
43
-
44
- # Return the AI's generated response
45
- return ai_response
46
-
47
- # Main execution block to test the chatbot function
48
- if __name__ == "__main__":
49
- # Print a test conversation with the chatbot
50
- print(get_response("Hello, Caramel AI! Can you tell me what AI is?", []))'''
51
-
52
-
53
- from openai import OpenAI
54
- from dotenv import load_dotenv
55
- import os
56
- from prompts import ai_motivational_speaker
57
-
58
- load_dotenv()
59
- api_key = os.getenv("GEMINI_API_KEY")
60
- model ="gemini-2.5-flash-lite"
61
- base_url = "https://generativelanguage.googleapis.com/v1beta/openai/"
62
-
63
- client = OpenAI(api_key=api_key, base_url=base_url)
64
-
65
- ai_motivational_speaker = ai_motivational_speaker
66
-
67
-
68
- def get_response(message, history):
69
- messages = [{"role": "system", "content": ai_motivational_speaker}]
70
- messages.extend(history)
71
- messages.append({"role": "user", "content": message})
72
- response = client.chat.completions.create(
73
- model=model,
74
- messages=messages
75
- )
76
- Ai_response = response.choices[0].message.content
77
- return Ai_response
78
-
79
- #if __name__ == "__main__":
80
-
81
  # print(get_response("Hello, Caramel AI! Can you tell me what AI is?", []))
 
1
+ from openai import OpenAI
2
+ from dotenv import load_dotenv
3
+ import os
4
+ from prompts import ai_Travel_Assistant
5
+
6
+ load_dotenv()
7
+ api_key = os.getenv("GEMINI_API_KEY")
8
+ model ="gemini-2.5-flash-lite"
9
+ base_url = "https://generativelanguage.googleapis.com/v1beta/openai/"
10
+
11
+ client = OpenAI(api_key=api_key, base_url=base_url)
12
+
13
+ ai_Travel_Assistant = ai_Travel_Assistant
14
+
15
+
16
+ def get_response(message, history):
17
+ messages = [{"role": "system", "content": ai_Travel_Assistant}]
18
+ messages.extend(history)
19
+ messages.append({"role": "user", "content": message})
20
+ response = client.chat.completions.create(
21
+ model=model,
22
+ messages=messages
23
+ )
24
+ Ai_response = response.choices[0].message.content
25
+ return Ai_response
26
+
27
+ #if __name__ == "__main__":
28
+
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
  # print(get_response("Hello, Caramel AI! Can you tell me what AI is?", []))