Update session_id.py

session_id.py

@@ -1,21 +1,25 @@
 import os
 import requests
 
-url = "https://cloud.onyx.app/api/chat/create-chat-session"
+def generate_chat_session_id():
+    url = "https://cloud.onyx.app/api/chat/create-chat-session"
 
-payload = {
-    "persona_id": 0,
-    "description": "Session created via script",
-    "project_id": 123
-}
+    payload = {
+        "persona_id": 0,
+        "description": "auto session per response",
+        "project_id": 123
+    }
 
-headers = {
-    "Authorization": f"Bearer {os.getenv('PR_KEY')}",
-    "Content-Type": "application/json"
-}
+    headers = {
+        "Authorization": f"Bearer {os.getenv('PR_KEY')}",
+        "Content-Type": "application/json"
+    }
 
-response = requests.post(url, json=payload, headers=headers, timeout=60)
-response.raise_for_status()
+    response = requests.post(url, json=payload, headers=headers, timeout=60)
+    response.raise_for_status()
 
-data = response.json()
-print(data["chat_session_id"])
+    return response.json()["chat_session_id"]
+
+
+if __name__ == "__main__":
+    print(generate_chat_session_id())