Spaces:
Running
Running
[email protected]
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 |
}
|