### What problem does this PR solve?

### Type of change
- [x] Refactoring
This commit is contained in:
Kevin Hu
2024-12-31 14:31:31 +08:00
committed by GitHub
parent b52b0f68fc
commit 4ba4f622a5
6 changed files with 11 additions and 15 deletions

View File

@ -152,8 +152,8 @@ def load_user(web_request):
return user[0]
else:
return None
except Exception:
logging.exception("load_user got exception")
except Exception as e:
logging.warning(f"load_user got exception {e}")
return None
else:
return None

View File

@ -65,10 +65,6 @@ def set_conversation():
"message": [{"role": "assistant", "content": dia.prompt_config["prologue"]}]
}
ConversationService.save(**conv)
e, conv = ConversationService.get_by_id(conv["id"])
if not e:
return get_data_error_result(message="Fail to new a conversation!")
conv = conv.to_dict()
return get_json_result(data=conv)
except Exception as e:
return server_error_response(e)

View File

@ -96,14 +96,11 @@ class DocumentService(CommonService):
def insert(cls, doc):
if not cls.save(**doc):
raise RuntimeError("Database error (Document)!")
e, doc = cls.get_by_id(doc["id"])
if not e:
raise RuntimeError("Database error (Document retrieval)!")
e, kb = KnowledgebaseService.get_by_id(doc.kb_id)
e, kb = KnowledgebaseService.get_by_id(doc["kb_id"])
if not KnowledgebaseService.update_by_id(
kb.id, {"doc_num": kb.doc_num + 1}):
raise RuntimeError("Database error (Knowledgebase)!")
return doc
return Document(**doc)
@classmethod
@DB.connection_context()

View File

@ -98,6 +98,7 @@ def get_exponential_backoff_interval(retries, full_jitter=False):
def get_data_error_result(code=settings.RetCode.DATA_ERROR,
message='Sorry! Data missing!'):
logging.exception(Exception(message))
result_dict = {
"code": code,
"message": message}