melhorias de referência

.vscode/launch.json

@@ -0,0 +1,18 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Start app",
+            "type": "debugpy",
+            "request": "launch",
+            "module": "streamlit",
+            "args": [
+                "run",
+                "app.py"
+            ]
+        }
+    ]
+}

app.py

@@ -44,7 +44,7 @@ input = st.chat_input("Digite sua pergunta:")
 
 if input:
     st.chat_message("user").markdown(input)
-    st.session_state.messages.append({"role": "assistant", "content": input})
+    st.session_state.messages.append({"role": "user", "content": input})
     documentos = vector_store.similarity_search(input, k=3)
 
     documentos_json = []
@@ -58,7 +58,7 @@ if input:
         documentos_json.append(documento_json)
 
     prompt = base_prompt.format(context=documentos_json, question=input)
-    print(prompt)
+
     st.session_state.messages_for_ia.append({"role": "assistant", "content": prompt})
 
     resposta = model.invoke(st.session_state.messages_for_ia)

prompts/base.md

@@ -3,8 +3,8 @@
 Responda à pergunta abaixo com base no contexto fornecido em formato JSON. O contexto contém três referências principais para sua resposta:
 
 * "source": Indica o caminho e o nome do arquivo PDF onde a informação foi encontrada.
-* "page": Representa o número da página no arquivo PDF onde o conteúdo está localizado.
-* "page_label": Corresponde à numeração da página conforme exibida no documento original.
+* "page": Corresponde à numeração da página conforme exibida no documento original.
+* "page_label":   Representa o número da página no arquivo PDF onde o conteúdo está localizado.
 * "page_content": Contém o trecho extraído do arquivo PDF.
 
 Caso existam outras chaves no JSON, ignore-as.
@@ -13,7 +13,7 @@ Sempre que utilizar uma referência do contexto, inclua a fonte correspondente n
 
 # Exemplo de resposta esperada:
 
-Se a resposta for baseada em um trecho do arquivo nome_do_documento.pdf (source: ./data/articles/nome_do_documento.pdf) na página 12 do arquivo (page: 12) e página numerada 10 no documento (page_label: 10), cite a fonte da seguinte forma:
+Se a resposta for baseada em um trecho do arquivo nome_do_documento.pdf (source: ./data/articles/nome_do_documento.pdf) na página 12 do arquivo (page_label: 12) e página numerada 10 no documento (page: 10), cite a fonte da seguinte forma:
 
 (Fonte: nome_do_documento.pdf, página 12 [Página no documento: 10])