Upload rag.py

rag.py

@@ -95,12 +95,16 @@ class Rag:
                         If the pieces of text don't enough information about the question to create the news, just say that you need more sources of information, nothing more.
                         The news should have a title.
                         The news should be written in a formal language.
-                        The source of the news should not be in the news, just the text.
+                        The source written in the pieces of text should not be shown in the news.
                         The news should have between {min_words} and {max_words} words.
+                        The total of words should no be shown in the news.
                         After write the news, confirm if it is written in Portuguese language.
+                        Answer the title and the news, nothing more.
                         The news should be about the following context: <{context}>
                         Question: <{question}>
-                        Answer here:"""
+                        Answer here: 
+                        Title: 
+                        News:"""
             self.prompt = PromptTemplate(template=prompt_template,
                                          input_variables=["context", "question"],
                                          partial_variables={"min_words": min_words, "max_words": max_words})