Spaces:
Sleeping
Sleeping
cd@bziiit.com commited on
Commit ·
425940e
1
Parent(s): 6b04ecd
Remove vectore store references
Browse files- app.py +1 -6
- config.yaml +0 -1
- rag.py +0 -3
app.py
CHANGED
|
@@ -25,12 +25,9 @@ def init_app():
|
|
| 25 |
|
| 26 |
|
| 27 |
if len(st.session_state) == 0:
|
| 28 |
-
# Define Vectore store strategy
|
| 29 |
-
pinecone_connector = PineconeConnector()
|
| 30 |
-
vs_manager = VectoreStoreManager(pinecone_connector)
|
| 31 |
|
| 32 |
st.session_state["messages"] = []
|
| 33 |
-
st.session_state["assistant"] = Rag(
|
| 34 |
st.session_state["data_dict"] = config['variables']
|
| 35 |
st.session_state["prompt_system"] = config['prompt_system']
|
| 36 |
|
|
@@ -57,7 +54,6 @@ def main():
|
|
| 57 |
|
| 58 |
st.write(f"**User ID:** {userId}")
|
| 59 |
|
| 60 |
-
saved_documents = st.Page("pages/persistent_documents.py", title="Communs", icon="🗃️")
|
| 61 |
documents = st.Page("pages/documents.py", title="Vos documents", icon="📂")
|
| 62 |
prompt_system = st.Page("pages/prompt_system.py", title="Prompt système", icon="🖊️", default=True)
|
| 63 |
form = st.Page("pages/form.py", title="Paramètres", icon="📋")
|
|
@@ -66,7 +62,6 @@ def main():
|
|
| 66 |
pg = st.navigation(
|
| 67 |
{
|
| 68 |
"Documents": [
|
| 69 |
-
saved_documents,
|
| 70 |
documents,
|
| 71 |
],
|
| 72 |
"Configurations": [
|
|
|
|
| 25 |
|
| 26 |
|
| 27 |
if len(st.session_state) == 0:
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
st.session_state["messages"] = []
|
| 30 |
+
st.session_state["assistant"] = Rag()
|
| 31 |
st.session_state["data_dict"] = config['variables']
|
| 32 |
st.session_state["prompt_system"] = config['prompt_system']
|
| 33 |
|
|
|
|
| 54 |
|
| 55 |
st.write(f"**User ID:** {userId}")
|
| 56 |
|
|
|
|
| 57 |
documents = st.Page("pages/documents.py", title="Vos documents", icon="📂")
|
| 58 |
prompt_system = st.Page("pages/prompt_system.py", title="Prompt système", icon="🖊️", default=True)
|
| 59 |
form = st.Page("pages/form.py", title="Paramètres", icon="📋")
|
|
|
|
| 62 |
pg = st.navigation(
|
| 63 |
{
|
| 64 |
"Documents": [
|
|
|
|
| 65 |
documents,
|
| 66 |
],
|
| 67 |
"Configurations": [
|
config.yaml
CHANGED
|
@@ -121,7 +121,6 @@ Paramètre 2 : {param2}
|
|
| 121 |
Paramètre 3 : {param3}
|
| 122 |
Paramètre 4 : {param4}
|
| 123 |
|
| 124 |
-
Documents partagées : {commonContext}
|
| 125 |
Document de référence : {documentContext}
|
| 126 |
|
| 127 |
{prompt_system}
|
|
|
|
| 121 |
Paramètre 3 : {param3}
|
| 122 |
Paramètre 4 : {param4}
|
| 123 |
|
|
|
|
| 124 |
Document de référence : {documentContext}
|
| 125 |
|
| 126 |
{prompt_system}
|
rag.py
CHANGED
|
@@ -86,14 +86,11 @@ class Rag:
|
|
| 86 |
else:
|
| 87 |
documentContext = self.retriever.invoke(query)
|
| 88 |
|
| 89 |
-
# Retrieve the VectoreStore
|
| 90 |
-
contextCommon = self.vector_store.retriever(query, self.embedding)
|
| 91 |
|
| 92 |
# Dictionnaire de base avec les variables principales
|
| 93 |
chain_input = {
|
| 94 |
"query": query,
|
| 95 |
"documentContext": documentContext,
|
| 96 |
-
"commonContext": contextCommon,
|
| 97 |
"prompt_system": prompt_system,
|
| 98 |
"messages": messages
|
| 99 |
}
|
|
|
|
| 86 |
else:
|
| 87 |
documentContext = self.retriever.invoke(query)
|
| 88 |
|
|
|
|
|
|
|
| 89 |
|
| 90 |
# Dictionnaire de base avec les variables principales
|
| 91 |
chain_input = {
|
| 92 |
"query": query,
|
| 93 |
"documentContext": documentContext,
|
|
|
|
| 94 |
"prompt_system": prompt_system,
|
| 95 |
"messages": messages
|
| 96 |
}
|