aoutler/aidgaf-server
aoutler/aidgaf-server
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

GAF to handle various server failures

Browse Source
This commit is contained in:
adamoutler 2023-02-08 18:31:48 +00:00
parent f7564146be
commit 5e0442a366
3 changed files with 31 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -1,6 +1,6 @@
# docker build -t aidgaf .
#for test
# docker run --rm -eSERVERPORT=8087 -eAPIKEY=sk-AaKV.........................HZ4v2ipzFm6 -p8087:8087 -it aidgaf
# docker run --rm -eAPIKEY=sk-AaKV.........................HZ4v2ipzFm6 -p8087:8087 -it aidgaf
# curl -X PATCH http://127.0.0.1:8087 -d "{\"message\":{\"command\":\"aidgaf\",\"data\":{\"username\":\"AdamOutler\"}}}"
#for deployment

6
src/aidgaf/aidgaf-server/idgaf.py
View File

@ -69,7 +69,11 @@ def parse_idgaf_request(command):
the_data = get_prompt(command)
gpt_response = requests.post(URL, json=the_data, headers=request_headers)
print(gpt_response)
try:
response_text = gpt_response.json()['choices'][0]['text'].strip()
except (KeyError):
response_text=gpt_response.text
obj = get_response_base_object(response_text)
obj['hash']=get_message_hash(json.dumps(obj))
json_result = json.dumps(obj)
@ -94,8 +98,10 @@ def get_message_hash(json_command) -> bytes:
strip1 = re.sub('.*\"message\":', "", json_command, 1)
strip2 = re.sub(',\"hash\":.*', '', strip1)
json_value = bytes(strip2, "utf-8")
if (settings.HASHKEY is not None):
hash_value = hash_calculator.calculate_hash(json_value, settings.HASHKEY)
return hash_value
return ""
def verify_message_time(json_command) -> bool:

8
src/aidgaf/aidgaf-server/settings.py
View File

@ -2,13 +2,17 @@ import os
# The hostname used by this app
HOSTNAME: str = os.getenv('HOSTNAME') # localhost or some name
# The port to broadcast the server
SERVERPORT:int = int(os.getenv('SERVERPORT')) #8087 or the port you want to run on. pass in with docker -e command.
# 8087 or the port you want to run on. pass in with docker -e command.
SERVERPORT: int = 8087
# The API key for OpenAI
APIKEY: str = os.getenv('APIKEY') # secret key from OpenAPI website
if APIKEY is None:
raise Exception("APIKEY Environmental Variable must be set")
# The hash key
HASHKEY:str = bytes(os.getenv('HASHKEY'),'utf-8') #shared secret for hmac of message
HASHKEY: str = None
hashKey = os.getenv('HASHKEY') # shared secret for hmac of message
if (hashKey is not None and hashKey.replace(" ", "") != ""):
HASKHEY = bytes(hashKey, "utf-8")
# The prompts used for OpenAI.
PROMPTS = ["Say \"USERNAME does not give a fuck\" in a thoughtful and clever paragraph of 5 sentences.",
"Say \"USERNAME does not give a fuck\" in a Dr Suess poem.",