mirror of
https://github.com/StanGirard/quivr.git
synced 2024-12-20 09:42:12 +03:00
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
from logger import get_logger
|
|
from models.chats import ChatMessage
|
|
from models.settings import CommonsDep
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
def create_chat(commons: CommonsDep, user_id, history, chat_name):
|
|
# Chat is created upon the user's first question asked
|
|
logger.info(f"New chat entry in chats table for user {user_id}")
|
|
|
|
# Insert a new row into the chats table
|
|
new_chat = {
|
|
"user_id": user_id,
|
|
"history": history, # Empty chat to start
|
|
"chat_name": chat_name,
|
|
}
|
|
insert_response = commons["supabase"].table("chats").insert(new_chat).execute()
|
|
logger.info(f"Insert response {insert_response.data}")
|
|
|
|
return insert_response
|
|
|
|
|
|
def update_chat(commons: CommonsDep, chat_id, history=None, chat_name=None):
|
|
if not chat_id:
|
|
logger.error("No chat_id provided")
|
|
return
|
|
|
|
updates = {}
|
|
|
|
if history is not None:
|
|
updates["history"] = history
|
|
|
|
if chat_name is not None:
|
|
updates["chat_name"] = chat_name
|
|
|
|
if updates:
|
|
commons["supabase"].table("chats").update(updates).match(
|
|
{"chat_id": chat_id}
|
|
).execute()
|
|
logger.info(f"Chat {chat_id} updated")
|
|
else:
|
|
logger.info(f"No updates to apply for chat {chat_id}")
|
|
|
|
|
|
def get_chat_name_from_first_question(chat_message: ChatMessage):
|
|
# Step 1: Get the summary of the first question
|
|
# first_question_summary = summarize_as_title(chat_message.question)
|
|
# Step 2: Process this summary to create a chat name by selecting the first three words
|
|
chat_name = " ".join(chat_message.question.split()[:3])
|
|
|
|
return chat_name
|