Spaces:
Sleeping
Sleeping
ASG Models
commited on
Rename chat_ai.py to genai_chat_ai.py
Browse files- chat_ai.py +0 -63
- 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 |
+
|