hueso/Alpaca
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed chat title generation and image recognition (+ added support for jpg)

Browse Source
This commit is contained in:
jeffser
2024-06-30 14:43:12 -06:00
parent c6110e8cde
commit b6a180080f
3 changed files with 14 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/connection_handler.py
View File

@@ -4,17 +4,17 @@ import json, requests
url = None
bearer_token = None
def get_headers() -> dict:
headers = {
"Content-Type": "application/json"
}
def get_headers(include_json:bool) -> dict:
headers = {}
if include_json:
headers["Content-Type"] = "application/json"
if bearer_token:
headers["Authorization"] = "Bearer " + bearer_token
return headers
headers["Authorization"] = "Bearer {}".format(bearer_token)
return headers if len(headers.keys()) > 0 else None
def simple_get(connection_url:str) -> dict:
try:
response = requests.get(connection_url, headers=get_headers())
response = requests.get(connection_url, headers=get_headers(False))
if response.status_code == 200:
return {"status": "ok", "text": response.text, "status_code": response.status_code}
else:
@@ -24,7 +24,7 @@ def simple_get(connection_url:str) -> dict:
def simple_post(connection_url:str, data) -> dict:
try:
response = requests.post(connection_url, headers=get_headers(), data=data, stream=False)
response = requests.post(connection_url, headers=get_headers(True), data=data, stream=False)
if response.status_code == 200:
return {"status": "ok", "text": response.text, "status_code": response.status_code}
else:
@@ -34,7 +34,7 @@ def simple_post(connection_url:str, data) -> dict:
def simple_delete(connection_url:str, data) -> dict:
try:
response = requests.delete(connection_url, headers=get_headers(), json=data)
response = requests.delete(connection_url, headers=get_headers(False), json=data)
if response.status_code == 200:
return {"status": "ok", "status_code": response.status_code}
else:
@@ -44,7 +44,7 @@ def simple_delete(connection_url:str, data) -> dict:
def stream_post(connection_url:str, data, callback:callable) -> dict:
try:
response = requests.post(connection_url, headers=get_headers(), data=data, stream=True)
response = requests.post(connection_url, headers=get_headers(True), data=data, stream=True)
if response.status_code == 200:
for line in response.iter_lines():
if line: