Revert "Merge Updates & Dockerfile improvements" (#3)

This reverts commit 9763d885be.

backend/apps/rag/utils.py

@@ -1,11 +1,7 @@
 import re
-import logging
 from typing import List
 
-from config import SRC_LOG_LEVELS, CHROMA_CLIENT
-
-log = logging.getLogger(__name__)
-log.setLevel(SRC_LOG_LEVELS["RAG"])
+from config import CHROMA_CLIENT
 
 
 def query_doc(collection_name: str, query: str, k: int, embedding_function):
@@ -95,13 +91,14 @@ def query_collection(
 
 
 def rag_template(template: str, context: str, query: str):
-    template = template.replace("[context]", context)
-    template = template.replace("[query]", query)
+    template = re.sub(r"\[context\]", context, template)
+    template = re.sub(r"\[query\]", query, template)
+
     return template
 
 
 def rag_messages(docs, messages, template, k, embedding_function):
-    log.debug(f"docs: {docs}")
+    print(docs)
 
     last_user_message_idx = None
     for i in range(len(messages) - 1, -1, -1):
@@ -141,8 +138,6 @@ def rag_messages(docs, messages, template, k, embedding_function):
                     k=k,
                     embedding_function=embedding_function,
                 )
-            elif doc["type"] == "text":
-                context = doc["content"]
             else:
                 context = query_doc(
                     collection_name=doc["collection_name"],
@@ -151,13 +146,11 @@ def rag_messages(docs, messages, template, k, embedding_function):
                     embedding_function=embedding_function,
                 )
         except Exception as e:
-            log.exception(e)
+            print(e)
             context = None
 
         relevant_contexts.append(context)
 
-    log.debug(f"relevant_contexts: {relevant_contexts}")
-
     context_string = ""
     for context in relevant_contexts:
         if context: