✍️ Add sources option in `bot`

README.md

@@ -83,6 +83,7 @@ The `bot` function should serve as the starting point for creating and customisi
 | model            | The name of the model to use for the bot. You can specify a different model by providing its name, like "text-davinci-003". Supported models: `gpt-3.5-turbo` (default),`text-davinci-003` More comming soon.                                                                              |
 | prompt_template  | A string template for the prompt, which defines the format of the question and context passed to the model. The template should include placeholders for the variables specified in `prompt_variables`.                                                                                    |
 | prompt_variables | A list of variables to be used in the prompt template. These variables are replaced with actual values when the bot processes a query.                                                                                                                                                     |
+| sources          | When `sources` is `True` the bot will also include sources in the response. A known [issue](https://github.com/hwchase17/langchain/issues/2858) exists, where if you pass a custom prompt with sources the code breaks.                                                                    |
 
 ### How QnA bot works
 

megabots/__init__.py

@@ -37,13 +37,14 @@ class Bot:
 
         # Load the question-answering chain for the selected model
         self.chain = self.create_chain(
-            prompt_template, prompt_variables, verbose=verbose
+            prompt_template, prompt_variables, sources=sources, verbose=verbose
         )
 
     def create_chain(
         self,
         prompt_template: str | None = None,
         prompt_variables: list[str] | None = None,
+        sources: bool | None = False,
         verbose: bool = False,
     ):
         prompt = (
@@ -51,6 +52,12 @@ class Bot:
             if prompt_template is not None and prompt_variables is not None
             else QA_PROMPT
         )
+        # TODO: Changing the prompt here is not working. Leave it as is for now.
+        # Reference: https://github.com/hwchase17/langchain/issues/2858
+        if sources:
+            return load_qa_with_sources_chain(
+                self.llm, chain_type="stuff", verbose=verbose
+            )
         return load_qa_chain(
             self.llm, chain_type="stuff", verbose=verbose, prompt=prompt
         )