Update __init__.py
I changed the import statement from request import Session to import requests because in the original the Session class is the only class imported from the request library. In this case, it is more efficient to import the whole library with import requests and use requests.Session() instead of Session() and added various code comments Instead of having a separate dictionary, I created a headers dictionary and used the update() method to update the session header. The create() function returns the results of a request made to a website as a JSON dictionary using the .json() method. However, you don't need to import the json library explicitly because the json() function is a method of the Response object returned by session.post(). In other words, the .json() method does the json decoding for us, so we don't need to use the json module's json()
This commit is contained in:
parent
8c2e8a233f
commit
96eab8ed1d
1 changed files with 16 additions and 12 deletions
|
@ -1,27 +1,31 @@
|
||||||
from requests import Session
|
import requests
|
||||||
import json
|
|
||||||
|
|
||||||
class Completion:
|
class Completion:
|
||||||
def create(
|
def create(prompt="What is the square root of pi",
|
||||||
prompt: str = "What is the square root of pi",
|
system_prompt="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()
|
# Initialize a session
|
||||||
client.headers = {
|
session = requests.Session()
|
||||||
|
|
||||||
|
# Set headers for the request
|
||||||
|
headers = {
|
||||||
'Accept': '*/*',
|
'Accept': '*/*',
|
||||||
'Accept-Language': 'en-US,en;q=0.5',
|
'Accept-Language': 'en-US,en;q=0.5',
|
||||||
"origin" : "https://cocalc.com",
|
'Origin': 'https://cocalc.com',
|
||||||
"referer" : "https://cocalc.com/api/v2/openai/chatgpt",
|
'Referer': 'https://cocalc.com/api/v2/openai/chatgpt',
|
||||||
"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",
|
'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',
|
||||||
}
|
}
|
||||||
|
session.headers.update(headers)
|
||||||
|
|
||||||
|
# Set the data that will be submitted
|
||||||
payload = {
|
payload = {
|
||||||
"input": prompt,
|
"input": prompt,
|
||||||
"system": system_prompt,
|
"system": system_prompt,
|
||||||
"tag": "next:index"
|
"tag": "next:index"
|
||||||
}
|
}
|
||||||
|
|
||||||
response = client.post(f"https://cocalc.com/api/v2/openai/chatgpt", json=payload).json()
|
# Submit the request
|
||||||
|
response = session.post("https://cocalc.com/api/v2/openai/chatgpt", json=payload).json()
|
||||||
|
|
||||||
|
# Return the results
|
||||||
return response
|
return response
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue