ai_chatbot 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bde3d724e066978fd481402f7aaa899041d85044b8b3ecc10574410ee79946f2
-  data.tar.gz: 8cc52ee8ec6b7e061fa6c009fdac5886b6b8e1a6e785d50abf70146bf04135ab
+  metadata.gz: 44d4375b7617e626f0ece5d48d9951db6cbb29d40fc039ed432f57f95214ff61
+  data.tar.gz: b47d32642a89b0a8ee430d210d57fa310616828938e431cf5b9094b1720e9d92
 SHA512:
-  metadata.gz: 115664a820b9b5c3da45d3fbb382785bd24dc0a840ad707fd8d1f0e7612d498c377ed12e86e7f997191cc83585a4245cfedda79cc22ee8ecfd44e01c131bb10b
-  data.tar.gz: f94763913a5cf902b9f940855c9f5a608a47495f776ddd3ab59e223ebad03be2bb0320c9d57317db2435a2f0662e08aca18f3143e56265a3fb7d7e86e018a6ca
+  metadata.gz: 2b85bdabf2fcf579a8355ef6b847251b6f1a715ef648a70bdbff1d4807f3615b425c50b80ce54f653477ae4c4caa163c8614e9563364a75145ff346b369077c3
+  data.tar.gz: 1fd7b59dde9c05e0ebca46fb472365786e7dbeb6440d2b30c7a0239246343efb8c3cb617aad4a83c48af2c3d18daab3726aaf6f672ea20b33bfbaa012995b94c

data/lib/ai_chatbot/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AiChatbot
-  VERSION = "0.1.0"
+  VERSION = "0.1.2"
 end

data/lib/ml_model.py

@@ -34,6 +34,10 @@ def main(action, query=None, new_answer=None):
         return get_prediction(query)
     elif action == "train":
         return train_model(query, new_answer)
+    elif action == "update":
+        return update_answer(query, new_answer)
+    elif action == "list":
+        return list_questions()
 
 # Function to predict the response with confidence check
 def get_prediction(query):
@@ -44,7 +48,7 @@ def get_prediction(query):
     similarities = cosine_similarity(query_vec, question_vecs)
     max_similarity = similarities.max()
 
-    threshold = 0.85
+    threshold = 0.70
     if max_similarity < threshold:
         return "No good match found. Please provide the correct answer."
     else:
@@ -68,6 +72,27 @@ def train_model(new_question, new_answer):
 
     return f"Model retrained with the new question: '{new_question}' and answer: '{new_answer}'"
 
+def update_answer(existing_question, new_answer):
+    global questions, answers
+
+    if existing_question in questions:
+        # Find the index of the existing question
+        index = questions.index(existing_question)
+        # Update the answer
+        answers[index] = new_answer
+        # Retrain the model with updated data
+        model.fit(questions, answers)
+        # Save the updated model and data
+        with open("qa_model.pkl", "wb") as f:
+            pickle.dump({"questions": questions, "answers": answers}, f)
+        return f"Answer updated for the question: '{existing_question}'"
+    else:
+        return "Question not found. Please provide a valid question."    
+
+def list_questions():
+    global questions
+    return questions
+
 if __name__ == "__main__":
     # Expecting action (predict/train), question, and answer (if training)
     action = sys.argv[1]

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ai_chatbot
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Sanet
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-09-14 00:00:00.000000000 Z
+date: 2024-09-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: open3