refactor(gpt4free/cocalc/__init__.py): Refactored init file
このコミットが含まれているのは:
コミット
dfc9e01889
|
@ -101,6 +101,10 @@ Run
|
|||
```
|
||||
docker run -p 8501:8501 gpt4free:latest
|
||||
```
|
||||
Another way - docker-compose (no docker build/run needed)
|
||||
```
|
||||
docker-compose up -d
|
||||
```
|
||||
|
||||
## Deploy using docker-compose
|
||||
|
||||
|
|
|
@ -0,0 +1,9 @@
|
|||
version: "3.9"
|
||||
|
||||
services:
|
||||
gpt4free:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: Dockerfile
|
||||
ports:
|
||||
- "8501:8501"
|
|
@ -9,29 +9,52 @@ class CoCalcResponse(BaseModel):
|
|||
|
||||
|
||||
class Completion:
|
||||
"""A class for generating text completions using CoCalc's GPT-based chatbot."""
|
||||
|
||||
API_ENDPOINT = "https://cocalc.com/api/v2/openai/chatgpt"
|
||||
DEFAULT_SYSTEM_PROMPT = "ASSUME I HAVE FULL ACCESS TO COCALC. "
|
||||
|
||||
@staticmethod
|
||||
def create(prompt: str, cookie_input: str) -> CoCalcResponse:
|
||||
"""
|
||||
Generate a text completion for the given prompt using CoCalc's GPT-based chatbot.
|
||||
|
||||
Args:
|
||||
prompt: The text prompt to complete.
|
||||
cookie_input: The cookie required to authenticate the chatbot API request.
|
||||
|
||||
Returns:
|
||||
A CoCalcResponse object containing the text completion and a boolean indicating
|
||||
whether the request was successful.
|
||||
"""
|
||||
|
||||
# Initialize a session with custom headers
|
||||
session = Completion._initialize_session(cookie_input)
|
||||
|
||||
# Set the data that will be submitted
|
||||
payload = Completion._create_payload(prompt, 'ASSUME I HAVE FULL ACCESS TO COCALC. ')
|
||||
payload = Completion._create_payload(prompt, Completion.DEFAULT_SYSTEM_PROMPT)
|
||||
|
||||
# Submit the request and return the results
|
||||
return Completion._submit_request(session, payload)
|
||||
try:
|
||||
# Submit the request and return the results
|
||||
response = session.post(Completion.API_ENDPOINT, json=payload).json()
|
||||
return CoCalcResponse(text=response['output'], status=response['success'])
|
||||
except requests.exceptions.RequestException as e:
|
||||
# Handle exceptions that may occur during the request
|
||||
print(f"Error: {e}")
|
||||
return CoCalcResponse(text="", status=False)
|
||||
|
||||
@classmethod
|
||||
def _initialize_session(cls, conversation_cookie) -> requests.Session:
|
||||
def _initialize_session(cls, conversation_cookie: str) -> requests.Session:
|
||||
"""Initialize a session with custom headers for the request."""
|
||||
|
||||
session = requests.Session()
|
||||
headers = {
|
||||
'Accept': '*/*',
|
||||
'Accept-Language': 'en-US,en;q=0.5',
|
||||
'Origin': 'https://cocalc.com',
|
||||
'Referer': 'https://cocalc.com/api/v2/openai/chatgpt',
|
||||
'Cookie': conversation_cookie,
|
||||
'User-Agent': UserAgent().random,
|
||||
"Accept": "*/*",
|
||||
"Accept-Language": "en-US,en;q=0.5",
|
||||
"Origin": "https://cocalc.com",
|
||||
"Referer": "https://cocalc.com/api/v2/openai/chatgpt",
|
||||
"Cookie": conversation_cookie,
|
||||
"User-Agent": UserAgent().random,
|
||||
}
|
||||
session.headers.update(headers)
|
||||
|
||||
|
@ -39,9 +62,6 @@ class Completion:
|
|||
|
||||
@staticmethod
|
||||
def _create_payload(prompt: str, system_prompt: str) -> dict:
|
||||
return {'input': prompt, 'system': system_prompt, 'tag': 'next:index'}
|
||||
"""Create the payload for the API request."""
|
||||
|
||||
@staticmethod
|
||||
def _submit_request(session: requests.Session, payload: dict) -> CoCalcResponse:
|
||||
response = session.post('https://cocalc.com/api/v2/openai/chatgpt', json=payload).json()
|
||||
return CoCalcResponse(text=response['output'], status=response['success'])
|
||||
return {"input": prompt, "system": system_prompt, "tag": "next:index"}
|
||||
|
|
読み込み中…
新しいイシューから参照