added eqe manual

InnovationHub/llm/chain.py

@@ -19,7 +19,6 @@ class VehicleManualChatbot:
         self.embeddings = embeddings
         self.chatgpt_chain = self._create_chatgpt_chain()
 
-
     def _create_chatgpt_chain(self, temperature=0.5):
         template = """
         {history}
@@ -38,7 +37,6 @@ class VehicleManualChatbot:
             memory=ConversationalBufferWindowMemory(k=2),
         )
 
-
     def _get_prompt(self, question, vehicle, k=4):
         prompt = f"""
         I need information from my {vehicle} manual.
@@ -57,7 +55,6 @@ class VehicleManualChatbot:
             '\n'.join(context[:k]) + '\n' + 'Question:\n' + question
         return user_input
 
-
     def _ask_question(self, question, vehicle, k=2):
         index = FAISS.load_local(
             folder_path=self.db_paths[vehicle], embeddings=self.embeddings)
@@ -67,27 +64,28 @@ class VehicleManualChatbot:
         response = self.chatgpt_chain.predict(human_input=prompt)
         return response
 
-
     def chat(self, question, vehicle, k=2, temperature=0.5):
-        self.chatgpt_chain = self._create_chatgpt_chain(temperature=temperature)
+        self.chatgpt_chain = self._create_chatgpt_chain(
+            temperature=temperature)
         response = self._ask_question(question=question, vehicle=vehicle,
-                                    k=k)
+                                      k=k)
         return response
 
 
-
 db_paths = {
     "S-Class": "data/s-class-manual",
-    "EQS": "data/eqs-manual"
+    "EQS": "data/eqs-manual",
+    "EQE": "data/eqe-manual",
 }
 
 embeddings = HuggingFaceEmbeddings()
 
-vehicle_options = ["S-Class", "EQS"]
+vehicle_options = ["S-Class", "EQS", "EQE"]
 chatbot = VehicleManualChatbot(db_paths=db_paths,
                                vehicle_options=vehicle_options,
                                embeddings=embeddings)
 
+
 def start_ui():
     chatbot_interface = gradio.Interface(
         fn=chatbot.chat,
@@ -95,7 +93,7 @@ def start_ui():
                 gradio.inputs.Dropdown(
                     vehicle_options, label="Select Vehicle Model"),
                 gradio.inputs.Slider(minimum=1, maximum=10, step=1, label="k")
-        ],
+                ],
         outputs="text",
         title="Owner's Manual",
         description="Ask your vehicle manual and get a response.",
@@ -103,7 +101,7 @@ def start_ui():
                   ["What is flacon?", "S-Class", 3],
                   ["What is hyperscreen?", "EQS", 2],
                   ["Where can I find my vin?", "EQS", 3]
-        ]
+                  ]
     )
 
-    chatbot_interface.launch()
+    chatbot_interface.launch()