fixed add new row issue

pradeepgudipati · Apr 22, 2024 · 55a809e · 55a809e
1 parent 5e539e0
commit 55a809e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -132,3 +132,4 @@ cython_debug/
 *.jsonl
 *.csv
 *.ann
+/chroma_db
diff --git a/README.md b/README.md
@@ -136,8 +136,8 @@ This python code converts the jsonl data to a sqllite database with a table call
 The messages table has 3 rows
 
 1. id - Primary key
-2. User - Contains the Question
-3. Assistant - Contains the answer to the Question
+2. Question - Contains the Question
+3. Answer - Contains the answer to the Question
 
 ### 2. DB to JSONL 
  - SQLite to JSONL Converter code - [db_to_jsonl.py](jsonl_data_to_db%2Fdb_to_jsonl.py)

diff --git a/src/app.py b/src/app.py
@@ -175,7 +175,7 @@ def add_new_qa():
     :return: response
     """
     data = request.json
-    new_item = LLMDataModel(user=data["question"], assistant=data["answer"])
+    new_item = LLMDataModel(question=data["question"], answer=data["answer"])
     db.session.add(new_item)
     db.session.commit()
     return jsonify(status="success", message=f"New item added."), 200