Add time data (#51)

* Openai activeloop data (#37)

* adding openai and activeloop data

* fixing issues with names

* concurrency

* black

* black

* revert to gradio3.50 for concurrency

---------

Co-authored-by: Omar Solano <[email protected]>

* ensure gradio version for HF

* Updates to files

* Push to advanced rag course

* edits to prompt

* Add datetime to mongodb logs

* tests for history

* add history and history length to database

* formatting

---------

Co-authored-by: Omar Solano <[email protected]>

app.py

@@ -55,7 +55,7 @@ logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.INFO)
 
 
-def save_completion(completion: Completion, question: gr.Textbox):
+def save_completion(completion: Completion, history):
     collection = "completion_data-hf"
 
     # Convert completion to JSON and ignore certain columns
@@ -65,6 +65,8 @@ def save_completion(completion: Completion, question: gr.Textbox):
 
     # Add the current date and time to the JSON
     completion_json["timestamp"] = datetime.utcnow().isoformat()
+    completion_json["history"] = history
+    completion_json["history_len"] = len(history)
 
     try:
         cfg.mongo_db[collection].insert_one(completion_json)
@@ -159,7 +161,6 @@ def get_empty_source_completion(user_input):
 
 def get_answer(history, sources: Optional[list[str]] = None):
     user_input = history[-1][0]
-
     if len(sources) == 0:
         completion = get_empty_source_completion(user_input)
 
@@ -175,7 +176,6 @@ def get_answer(history, sources: Optional[list[str]] = None):
 
     for token in completion.answer_generator:
         history[-1][1] += token
-
         yield history, completion
 
 
@@ -204,7 +204,6 @@ with gr.Blocks(
     )
 
     chatbot = gr.Chatbot(elem_id="chatbot", show_copy_button=True)
-
     with gr.Row():
         question = gr.Textbox(
             label="What's your question?",
@@ -237,13 +236,13 @@ with gr.Blocks(
     submit.click(user, [question, chatbot], [question, chatbot], queue=False).then(
         get_answer, inputs=[chatbot, source_selection], outputs=[chatbot, completion]
     ).then(add_sources, inputs=[chatbot, completion], outputs=[chatbot]).then(
-        save_completion, inputs=[completion]
+        save_completion, inputs=[completion, chatbot]
     )
 
     question.submit(user, [question, chatbot], [question, chatbot], queue=False).then(
         get_answer, inputs=[chatbot, source_selection], outputs=[chatbot, completion]
     ).then(add_sources, inputs=[chatbot, completion], outputs=[chatbot]).then(
-        save_completion, inputs=[completion]
+        save_completion, inputs=[completion, chatbot]
     )
 
     chatbot.like(log_likes, completion)