LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-30 00:53:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: user logging api operations level

Browse Source
This commit is contained in:
Alex
2024-09-10 01:30:47 +01:00
parent ac930d5504
commit 90309d5552
5 changed files with 78 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
application/api/answer/routes.py
View File

@@ -23,6 +23,7 @@ conversations_collection = db["conversations"]
vectors_collection = db["vectors"]
prompts_collection = db["prompts"]
api_key_collection = db["api_keys"]
user_logs_collection = db["user_logs"]
answer = Blueprint("answer", __name__)
gpt_model = ""
@@ -202,6 +203,19 @@ def complete_stream(
# send data.type = "end" to indicate that the stream has ended as json
data = json.dumps({"type": "id", "id": str(conversation_id)})
yield f"data: {data}\n\n"
retriever_params = retriever.get_params()
user_logs_collection.insert_one({
"action": "stream_answer",
"level": "info",
"user": "local",
"api_key": user_api_key,
"date": datetime.datetime.utcnow(),
"question": question,
"response": response_full,
"sources": source_log_docs,
"retriever_params": retriever_params
})
data = json.dumps({"type": "end"})
yield f"data: {data}\n\n"
@@ -405,6 +419,18 @@ def api_answer():
result["conversation_id"] = save_conversation(
conversation_id, question, response_full, source_log_docs, llm
)
retriever_params = retriever.get_params()
user_logs_collection.insert_one({
"action": "api_answer",
"level": "info",
"user": "local",
"api_key": user_api_key,
"date": datetime.datetime.utcnow(),
"question": question,
"response": response_full,
"sources": source_log_docs,
"retriever_params": retriever_params
})
return result
except Exception as e:
@@ -460,6 +486,18 @@ def api_search():
)
docs = retriever.search()
retriever_params = retriever.get_params()
user_logs_collection.insert_one({
"action": "api_search",
"level": "info",
"user": "local",
"api_key": user_api_key,
"date": datetime.datetime.utcnow(),
"question": question,
"sources": docs,
"retriever_params": retriever_params
})
if data.get("isNoneDoc"):
for doc in docs:
doc["source"] = "None"