ASG Models commited on
Commit
8c2f5a2
·
verified ·
1 Parent(s): 60658c3

Rename chat_ai.py to genai_chat_ai.py

Browse files
Files changed (2) hide show
  1. chat_ai.py +0 -63
  2. genai_chat_ai.py +58 -0
chat_ai.py DELETED
@@ -1,63 +0,0 @@
1
- import os
2
-
3
- import google.generativeai as genai
4
-
5
- genai.configure(api_key="AIzaSyDC5lRm4Ikz9LfUZFNEnlnLavmSpVdIaBM")
6
-
7
- # Create the model
8
- # See https://ai.google.dev/api/python/google/generativeai/GenerativeModel
9
- generation_config = {
10
- "temperature": 1,
11
- "top_p": 0.95,
12
- "top_k": 64,
13
- "max_output_tokens": 8192,
14
- "response_mime_type": "text/plain",
15
- }
16
-
17
- model = genai.GenerativeModel(
18
- model_name="gemini-1.5-pro",
19
- generation_config=generation_config,
20
- # safety_settings = Adjust safety settings
21
- # See https://ai.google.dev/gemini-api/docs/safety-settings
22
- )
23
-
24
- chat_session = model.start_chat(
25
- history=[
26
- {
27
- "role": "user",
28
- "parts": [
29
- "السلام عليكم \n",
30
- ],
31
- },
32
- {
33
- "role": "model",
34
- "parts": [
35
- "و عليكم السلام و رحمة الله و بركاته \n \nكيف حالك؟ \n",
36
- ],
37
- },
38
- {
39
- "role": "user",
40
- "parts": [
41
- "اريد منك ان ترد دائما على سوالي باللهجة النجدية\n",
42
- ],
43
- },
44
- {
45
- "role": "model",
46
- "parts": [
47
- "هلا والله! من عيوني، ابشر من اليوم ورايح أرد عليك بالنجدي بس. وش عندك؟ \n",
48
- ],
49
- },
50
- {
51
- "role": "user",
52
- "parts": [
53
- "ما تعرف عن الرياض\n\n",
54
- ],
55
- },
56
- {
57
- "role": "model",
58
- "parts": [
59
- "وش ما أعرف عن الرياض ياهلي! الرياض عاصمة المملكة، وتعتبر من أكبر مدنها. معروفة بناطحات السحاب وأسواقها الفخمة ومطاعمها العالمية. غير كذا فيها معالم تاريخية مهمة زي قصر المصمك وحارة البجيري. \n\nوش تبي تعرف عنها بالضبط؟ \n",
60
- ],
61
- },
62
- ]
63
- )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
genai_chat_ai.py ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+
3
+ import google.generativeai as genai
4
+ api_key = os.environ.get("id_gmkey")
5
+
6
+ genai.configure(api_key=api_key)
7
+
8
+ generation_config = {
9
+ "temperature": 1,
10
+ "top_p": 0.95,
11
+ "top_k": 64,
12
+ "max_output_tokens": 8192,
13
+ "response_mime_type": "text/plain",
14
+ }
15
+
16
+ model = genai.GenerativeModel(
17
+ model_name="gemini-1.5-pro",
18
+ generation_config=generation_config,
19
+ # safety_settings = Adjust safety settings
20
+ # See https://ai.google.dev/gemini-api/docs/safety-settings
21
+ )
22
+
23
+ def create_chat_session():
24
+ chat_session = model.start_chat(
25
+ history=[
26
+ {
27
+ "role": "user",
28
+ "parts": [
29
+ os.environ.get("user_0") ,
30
+ ],
31
+ },
32
+ {
33
+ "role": "model",
34
+ "parts": [
35
+ os.environ.get("model_0")
36
+ ,
37
+ ],
38
+ },
39
+ {
40
+ "role": "user",
41
+ "parts": [
42
+ os.environ.get("user_1"),
43
+ ],
44
+ },
45
+ {
46
+ "role": "model",
47
+ "parts": [
48
+ os.environ.get("model_1")
49
+ ,
50
+ ],
51
+ }
52
+
53
+ ]
54
+ )
55
+ return chat_session
56
+
57
+ AI=chat_session()
58
+