UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-21 12:29:29 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-29 01:42:42 -07:00
parent 5cf44ac7da
commit 7f8be4a709
1 changed files with 19 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
backend/main.py
View File

@ -484,14 +484,13 @@ async def get_pipeline_valves(pipeline_id: str, user=Depends(get_admin_user)):
url = openai_app.state.config.OPENAI_API_BASE_URLS[urlIdx] url = openai_app.state.config.OPENAI_API_BASE_URLS[urlIdx]
key = openai_app.state.config.OPENAI_API_KEYS[urlIdx] key = openai_app.state.config.OPENAI_API_KEYS[urlIdx]
if key != "": headers = {"Authorization": f"Bearer {key}"}
headers = {"Authorization": f"Bearer {key}"} r = requests.get(f"{url}/{pipeline['id']}/valves", headers=headers)
r = requests.get(f"{url}/{pipeline['id']}/valves", headers=headers)
r.raise_for_status() r.raise_for_status()
data = r.json() data = r.json()
return {**data} return {**data}
except Exception as e: except Exception as e:
# Handle connection error here # Handle connection error here
print(f"Connection error: {e}") print(f"Connection error: {e}")
@ -533,14 +532,13 @@ async def get_pipeline_valves_spec(pipeline_id: str, user=Depends(get_admin_user
url = openai_app.state.config.OPENAI_API_BASE_URLS[urlIdx] url = openai_app.state.config.OPENAI_API_BASE_URLS[urlIdx]
key = openai_app.state.config.OPENAI_API_KEYS[urlIdx] key = openai_app.state.config.OPENAI_API_KEYS[urlIdx]
if key != "": headers = {"Authorization": f"Bearer {key}"}
headers = {"Authorization": f"Bearer {key}"} r = requests.get(f"{url}/{pipeline['id']}/valves/spec", headers=headers)
r = requests.get(f"{url}/{pipeline['id']}/valves/spec", headers=headers)
r.raise_for_status() r.raise_for_status()
data = r.json() data = r.json()
return {**data} return {**data}
except Exception as e: except Exception as e:
# Handle connection error here # Handle connection error here
print(f"Connection error: {e}") print(f"Connection error: {e}")
@ -583,18 +581,17 @@ async def update_pipeline_valves(
url = openai_app.state.config.OPENAI_API_BASE_URLS[urlIdx] url = openai_app.state.config.OPENAI_API_BASE_URLS[urlIdx]
key = openai_app.state.config.OPENAI_API_KEYS[urlIdx] key = openai_app.state.config.OPENAI_API_KEYS[urlIdx]
if key != "": headers = {"Authorization": f"Bearer {key}"}
headers = {"Authorization": f"Bearer {key}"} r = requests.post(
r = requests.post( f"{url}/{pipeline['id']}/valves/update",
f"{url}/{pipeline['id']}/valves/update", headers=headers,
headers=headers, json={**form_data},
json={**form_data}, )
)
r.raise_for_status() r.raise_for_status()
data = r.json() data = r.json()
return {**data} return {**data}
except Exception as e: except Exception as e:
# Handle connection error here # Handle connection error here
print(f"Connection error: {e}") print(f"Connection error: {e}")