small refactoring

bkocis · Oct 19, 2023 · 995aa67 · 995aa67
1 parent bd49cdc
commit 995aa67
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 3 deletions.
diff --git a/chatgptApp/__init__.py b/chatgptApp/__init__.py
diff --git a/chatgptApp/main.py b/chatgptApp/main.py
@@ -9,7 +9,7 @@
 from langchain.chat_models import ChatOpenAI
 from langchain.prompts import HumanMessagePromptTemplate
 from langchain.schema import AIMessage, BaseMessage, HumanMessage, SystemMessage
-from callback import QueueCallback
+from utils.callback import QueueCallback
 
 
 logging.basicConfig(
@@ -20,6 +20,8 @@
 
 
 def insert_into_db(message, content):
+    conn = sqlite3.connect(path_to_db, check_same_thread=False)
+    c = conn.cursor()
     c.execute("INSERT INTO chat_session_01 (question, answer) VALUES (?, ?)", (message, content))
     conn.commit()
     c.execute("select * from chat_session_01")
@@ -183,8 +185,10 @@ def main(human_message_prompt_template):
             ) as demo:
         messages_gen = gr.State([SystemMessage(content=Path("prompts/system.prompt.general").read_text())])
         messages_py = gr.State([SystemMessage(content=Path("prompts/system.prompt.python").read_text())])
-        messages_4pyanalyzer = gr.State([SystemMessage(content=Path("prompts/system.prompt.python-analyser").read_text())])
-        messages_unit_test = gr.State([SystemMessage(content=Path("prompts/system.prompt.python-unittest-suggertion").read_text())])
+        messages_4pyanalyzer = gr.State([SystemMessage(content=Path("prompts/system.prompt.python-analyser").
+                                                       read_text())])
+        messages_unit_test = gr.State([SystemMessage(content=Path("prompts/system.prompt.python-unittest-suggestion").
+                                                     read_text())])
         kwargs = {
             "show_label": False,
             "height": 750

diff --git a/.../system.prompt.python-unittest-suggertion → .../system.prompt.python-unittest-suggestion b/.../system.prompt.python-unittest-suggertion → .../system.prompt.python-unittest-suggestion
diff --git a/chatgptApp/utils/__init__.py b/chatgptApp/utils/__init__.py
diff --git a/chatgptApp/callback.py → chatgptApp/utils/callback.py b/chatgptApp/callback.py → chatgptApp/utils/callback.py