[email protected] commited on
Commit
425940e
·
1 Parent(s): 6b04ecd

Remove vectore store references

Browse files
Files changed (3) hide show
  1. app.py +1 -6
  2. config.yaml +0 -1
  3. 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(vectore_store=vs_manager)
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
  }