2 files changed, 15 insertions, 8 deletions

diff --git a/cocalc/__init__.py b/unfinished/cocalc/__init__.py
index a77934e4..fa565a45 100644
--- a/cocalc/__init__.py
+++ b/unfinished/cocalc/__init__.py
@@ -4,25 +4,24 @@ import json
 class Completion:
     def create(
         prompt: str = "What is the square root of pi",
-        system_prompt: str = "ASSUME I HAVE FULL ACCESS TO COCALC. ENCLOSE MATH IN $. INCLUDE THE LANGUAGE DIRECTLY AFTER THE TRIPLE BACKTICKS IN ALL MARKDOWN CODE BLOCKS. How can I do the following using CoCalc? "
-        ) -> str:
+        system_prompt: str = "ASSUME I HAVE FULL ACCESS TO COCALC. ENCLOSE MATH IN $. INCLUDE THE LANGUAGE DIRECTLY AFTER THE TRIPLE BACKTICKS IN ALL MARKDOWN CODE BLOCKS. How can I do the following using CoCalc? ") -> str:
 
-        client = Session(client_identifier="chrome_108")
+        client = Session()
         client.headers = {
-            'Accept': */*
-            'Accept-Language': en-US,en;q=0.5
+            'Accept': '*/*',
+            'Accept-Language': 'en-US,en;q=0.5',
             "origin"            : "https://cocalc.com",
             "referer"           : "https://cocalc.com/api/v2/openai/chatgpt",
-            'cookie'            : "Cookie: CC_ANA=c68b16f3-f74c-403f-8e18-1d89168b2d13; "
             "user-agent"        : "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36",
         }
+        
         payload = {
             "input": prompt,
             "system": system_prompt,
             "tag": "next:index"
         }
 
-        response = client.post(f"https://cocalc.com/api/v2/openai/chatgpt", json=payload)
+        response = client.post(f"https://cocalc.com/api/v2/openai/chatgpt", json=payload).json()
 
-        return json.loads(response)["output"]
+        return response
 
diff --git a/unfinished/cocalc/cocalc_test.py b/unfinished/cocalc/cocalc_test.py
new file mode 100644
index 00000000..0e1a7b3b
--- /dev/null
+++ b/unfinished/cocalc/cocalc_test.py
@@ -0,0 +1,8 @@
+import cocalc
+
+
+response = cocalc.Completion.create(
+    prompt = 'hello world'
+)
+
+print(response)
\ No newline at end of file