Merge branch 'develop' into main

Author: seominjae1

chat/__init__.py

@@ -1,25 +1,28 @@
 from flask import Blueprint, request, jsonify
+from typing import Literal
 from utils import confirm_request
 from server.logger import logger
 from .bot import Bot
 
 chat_bp = Blueprint('chat', __name__, url_prefix='/chatbot')
 
-@chat_bp.route('/<chat_id>', methods=['GET'])
-def chat_with_watson():
-    data = request.json
+@chat_bp.route('/<int:user_id>', methods=['GET'])
+def chat_with_watson(user_id):
+    data = request.args
     if response_for_invalid_request := confirm_request(data, {
-        'question': str
+        'question': str,
+        'request': Literal["ask", "reset"]
     }):
         return response_for_invalid_request
 
-
-    bot = Bot()
-
-    try:
-        return jsonify({"answer": bot.ask(data['question'])}), 200
-    except Exception as e:
-        logger.error(e)
-        return jsonify({"answer": f"죄송합니다. 에러가 발생했습니다. 시스템, 또는 AI를 제공하는 외부 API의 문제일 수 있습니다."}), 500
+    if data['request'] == "ask":
+        try:
+            return jsonify({"answer": Bot(user_id).ask(data['question'])}), 200
+        except Exception as e:
+            logger.error(e)
+            return jsonify({"answer": f"죄송합니다. 에러가 발생했습니다. 시스템, 또는 AI를 제공하는 외부 API의 문제일 수 있습니다."}), 500
+    else:
+        Bot(user_id).clear_message_history() # TODO: 버그 발생중. 확인 필요
+        return jsonify({"message": "success"}), 200
 
 

chat/bot.py

@@ -1,7 +1,46 @@
-class Bot:
-    def __init__(self):
-        pass
+import sqlite3
+import threading
+from os.path import join, dirname, abspath
+from typing import Union
 
-    def ask(self, question:str) -> str:
-        return "챗봇의 응답입니다."
+from langchain_community.vectorstores import FAISS
+from langchain_openai import OpenAIEmbeddings
+from langgraph.checkpoint.sqlite import SqliteSaver
+from langgraph.graph.state import CompiledStateGraph
+
+from .graph import LangGraphMethods
+from .tools import Tools
+from .vectorstore import VectorStoreMethods
+
+
+class Bot(LangGraphMethods, VectorStoreMethods, Tools):
+    _instances: dict[int, 'Bot'] = {}
+    _lock: threading.Lock = threading.Lock()
+    SQLITE_CONNECTION_STRING: str = join(dirname(abspath(__file__)),
+                                         f"chats.db")  # base.py와 같은 경로에 SQLITE memory file 생성
+    embeddings = OpenAIEmbeddings()
+    vectorstore: FAISS
+
+    def __new__(cls, *args, **kwargs):
+        """
+            싱글톤 객체의 변형 구현.
+            입력받은 user id에 대응하는 챗봇이 없을 경우에 한해,
+            고유한 Bot 객체를 새로 만들고 반환하는 동시에 _instances에 내부적으로 저장한다.
+
+            만약 해당 user id에 대응하는 챗봇이 이미 생성되었을 경우,
+            그 챗봇을 반환한다.
+        """
+        with cls._lock: # 챗봇이 호출될 때마다 Bot 클래스 속성인 vectorstore를 업데이트
+            cls.vectorstore = getattr(cls, "vectorstore", None) or cls.initialize_vectorstore()
+            cls.update_vectorstore()
+            cls.retriever_tool = cls.create_retriever_tool()
+            cls.retriever_tool_node = cls.create_retriever_tool_node()
+        return super().__new__(cls)
+
+    def __init__(self, user_id):
+        # 메모리 저장소 생성 (그래프에 사용되기 때문에, 반드시 그래프 생성 이전에 선행되어야 함)
+        super(Tools).__init__()
+        self.memory = SqliteSaver(sqlite3.connect(self.SQLITE_CONNECTION_STRING, check_same_thread=False))
+        self.graph: CompiledStateGraph = self.build_graph()
+        self.id = user_id # TODO: 사용자별로 고유한 bot_id 필요
 

chat/constants.py

@@ -0,0 +1,90 @@
+from os.path import dirname
+from .local_vectorstore import __file__ as f
+vectorstore_dir = dirname(f)
+vectorstore_index_name:str = "activity"
+
+openai_dimension_size:int = 1536
+
+indication_for_information_request = """
+You are an intelligent assistant designed to answer user questions based on the provided context retrieved from the web.
+The user will likely ask about recent global issues or topics related to current worldwide events. 
+Your primary mission is to answer questions based on provided context or chat history.
+Provided context consists of search results from the internet.
+Ensure your response is concise and directly addresses the question.
+
+###
+
+Your final answer should be written concisely, followed by the source of the information.
+
+# Steps
+
+1. Carefully read and understand the context provided and the chat history.
+2. Identify the key information related to the question within the context.
+3. Formulate a answer based on the relevant information.
+4. Ensure your final answer directly addresses the question.
+5. List the source of the answer in bullet points, which must be a url of the document, followed by brief part of the context. Omit if the source cannot be found.
+
+# Output Format:
+
+Your final answer here.
+
+**Source**(Optional)
+- (Source of the answer, must be a url of the source information, followed by brief part of the context. Omit if you can't find the source of the answer.)
+- (list more if there are multiple sources)
+- ...
+
+###
+
+Remember:
+- It's crucial to base your answer solely on the **PROVIDED CONTEXT**. 
+- DO NOT use any external knowledge or information not present in the given materials.
+- If you can't find the source of the answer, you should answer that you don't know.
+
+###
+
+# Here is the CONTEXT that you should use to answer the question:
+{context}
+"""
+
+indication_for_recommendation_request = """
+You are an intelligent assistant designed to answer user questions based on the provided context retrieved from the database.
+The user will likely ask you to recommend extracurricular or external activities. 
+The provided context contains information about activities that have already been selected as suitable for the user. 
+Based on this context, generate a response that recommends these activities to the user in a helpful and engaging manner.
+
+###
+
+Your final answer should be written concisely, followed by the source of the information.
+
+# Steps
+
+1. Carefully read and understand the context provided and the chat history.
+2. Identify the key information related to the question within the context.
+3. Summarize or highlight the most relevant aspects of each activity (e.g., purpose, eligibility, benefits, start and end date).
+4. Recommend these activities to the user.
+5. List the source of the answer in bullet points, which must be a url of the document, followed by brief part of the context. Omit if the source cannot be found.
+6. If the context is empty or does not contain any activity information, politely explain that no suitable recommendations are available at the moment.
+
+# Output Format:
+
+Your final answer here.
+
+**Source**(Optional)
+- (Source of the answer, must be a url of the source information, followed by brief part of the context and source site name. Omit if you can't find the source of the answer.)
+- (list more if there are multiple sources)
+- ...
+
+###
+
+Remember:
+- It's crucial to base your answer solely on the **PROVIDED CONTEXT**. 
+- DO NOT use any external knowledge or information not present in the given materials.
+- If you can't find the source of the answer, you should answer that you don't know.
+
+###
+
+# Here is the CONTEXT that you should use to answer the question:
+{context}
+"""
+
+