Update app.py

app.py

@@ -109,15 +109,23 @@ with st.sidebar:
                     query_engine = index.as_query_engine(streaming=True)
 
                     # ====== Customise prompt template ======
-                    qa_prompt_tmpl_str = (
-                        "Context information is below.\n"
-                        "---------------------\n"
-                        "{context_str}\n"
-                        "---------------------\n"
-                        "Given the context information above I want you to think step by step to answer the query in a highly precise and crisp manner focused on the final answer, in case you don't know the answer say 'I don't know!'.\n"
-                        "Query: {query_str}\n"
-                        "Answer: "
-                    )
+                    qa_prompt_tmpl_str = """
+                    You are an AI assistant specialized in analyzing GitHub repositories.
+
+                    Repository structure:
+                    {tree}
+                    ---------------------
+
+                    Context information from the repository:
+                    {context_str}
+                    ---------------------
+
+                    Given the repository structure and context above, provide a clear and precise answer to the query. 
+                    Focus on the repository's content, code structure, and implementation details. 
+                    If the information is not available in the context, respond with 'I don't have enough information about that aspect of the repository.'
+
+                    Query: {query_str}
+                    Answer: """
                     qa_prompt_tmpl = PromptTemplate(qa_prompt_tmpl_str)
 
                     query_engine.update_prompts(