joakimp/python_test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Snyggade till en loggutskrivt

Browse Source
This commit is contained in:
Joakim Persson
2024-08-06 23:37:27 +02:00
parent 942f9f78c9
commit 97ee179b29
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
smartassist/src/backend.py
+11 -6
View File
@@ -177,7 +177,7 @@ def chat() -> dict[str, any]:
url = url_server url = url_server
headers = get_auth_headers(url) headers = get_auth_headers(url)
logger.debug(f"Sending request to:\n\turl:\t{url}\nmodel:\n\t{model}") logger.debug(f"Sending request to:\n\turl:\t{url}\n\tmodel:\t{model}")
try: try:
url = url + "/api/generate" url = url + "/api/generate"
logger.debug(f"url: {url} headers: {headers}") logger.debug(f"url: {url} headers: {headers}")
@@ -239,11 +239,16 @@ def select_endpoint_llm() -> Response:
if len(endpoints) != 1: if len(endpoints) != 1:
raise ValueError(f"Expected exactly one endpoint with title '{title}', found {len(endpoints)}") raise ValueError(f"Expected exactly one endpoint with title '{title}', found {len(endpoints)}")
global_state.set_host_url(endpoints[0]['url']) # Reset the session
global_state.set_llm(llm) if (title != global_state.get_host_title()) or (llm != global_state.get_llm()): # A change in setting
logger.debug(f"Updated to host url {endpoints[0]['url']} and LLM {llm}") session.clear()
logger.debug('Session cleared due to changed endpoint or changed LLM')
return jsonify({'message': 'Endpoint and LLM selected successfully'}) global_state.set_host_url(endpoints[0]['url'])
global_state.set_llm(llm)
logger.debug(f"Updated to host url {endpoints[0]['url']} and LLM {llm}")
return jsonify({'message': 'New endpoint and/or LLM detected, settings were changed successfully'})
else:
return jsonify({'message': 'Endpoint and LLM are untouched'})
@app.route('/smartassist', methods=["POST"]) @app.route('/smartassist', methods=["POST"])