Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -18,10 +18,6 @@ import spaces
|
|
18 |
import threading
|
19 |
import sys
|
20 |
import torch
|
21 |
-
from llama_index.core.indices.query.query_transform.base import (
|
22 |
-
StepDecomposeQueryTransform,
|
23 |
-
)
|
24 |
-
from llama_index.core.query_engine import MultiStepQueryEngine
|
25 |
|
26 |
|
27 |
head = """
|
@@ -161,7 +157,7 @@ def main():
|
|
161 |
vector_store, storage_context=storage_context
|
162 |
)
|
163 |
|
164 |
-
retriever = VectorIndexRetriever(index=index, similarity_top_k=3, vector_store_query_mode="hybrid", embed_model=embed_model, alpha=0.
|
165 |
|
166 |
current_collection = "BANDI_SISTEMA_PUGLIA"
|
167 |
return "collezione settata"
|
@@ -204,7 +200,7 @@ def main():
|
|
204 |
vector_store, storage_context=storage_context
|
205 |
)
|
206 |
|
207 |
-
retriever = VectorIndexRetriever(index=index, similarity_top_k=3, vector_store_query_mode="hybrid", embed_model=embed_model, alpha=0.
|
208 |
else:
|
209 |
chat_engine_bandi.reset()
|
210 |
chat_engine_bandi_sistema_puglia.reset()
|
@@ -302,7 +298,7 @@ def main():
|
|
302 |
index = VectorStoreIndex.from_vector_store(
|
303 |
vector_store, storage_context=storage_context
|
304 |
)
|
305 |
-
retriever_bandi_sistema_puglia = VectorIndexRetriever(index=index, similarity_top_k=3, vector_store_query_mode="hybrid", embed_model=embed_model, alpha=0.
|
306 |
|
307 |
chat_engine_bandi_sistema_puglia = ContextChatEngine(retriever=retriever_bandi_sistema_puglia,
|
308 |
context_template="Sei un chatbot in grado di rispondere alle domande su bandi regionali e avvisi della regione Puglia. Hai accesso ai bandi della regione Puglia. Qui sotto le informazioni di contesto recuperate. \n"
|
@@ -477,14 +473,10 @@ def main():
|
|
477 |
else:
|
478 |
print("MODALITA STANDARD")
|
479 |
if(str(current_response_mode)=="tree_summarize"):
|
480 |
-
step_decompose_transform = StepDecomposeQueryTransform(llm, verbose=True)
|
481 |
# define response synthesizer
|
482 |
response_synthesizer = get_response_synthesizer(streaming=True,response_mode="tree_summarize",text_qa_template=text_qa_template)
|
483 |
query_engine = None
|
484 |
query_engine = RetrieverQueryEngine(retriever=retriever, response_synthesizer=response_synthesizer)
|
485 |
-
query_engine = MultiStepQueryEngine(
|
486 |
-
query_engine, query_transform=step_decompose_transform
|
487 |
-
)
|
488 |
stream_response = None
|
489 |
print(userMessage)
|
490 |
stream_response = query_engine.query(userMessage)
|
|
|
18 |
import threading
|
19 |
import sys
|
20 |
import torch
|
|
|
|
|
|
|
|
|
21 |
|
22 |
|
23 |
head = """
|
|
|
157 |
vector_store, storage_context=storage_context
|
158 |
)
|
159 |
|
160 |
+
retriever = VectorIndexRetriever(index=index, similarity_top_k=3, vector_store_query_mode="hybrid", embed_model=embed_model, alpha=0.5)
|
161 |
|
162 |
current_collection = "BANDI_SISTEMA_PUGLIA"
|
163 |
return "collezione settata"
|
|
|
200 |
vector_store, storage_context=storage_context
|
201 |
)
|
202 |
|
203 |
+
retriever = VectorIndexRetriever(index=index, similarity_top_k=3, vector_store_query_mode="hybrid", embed_model=embed_model, alpha=0.5)
|
204 |
else:
|
205 |
chat_engine_bandi.reset()
|
206 |
chat_engine_bandi_sistema_puglia.reset()
|
|
|
298 |
index = VectorStoreIndex.from_vector_store(
|
299 |
vector_store, storage_context=storage_context
|
300 |
)
|
301 |
+
retriever_bandi_sistema_puglia = VectorIndexRetriever(index=index, similarity_top_k=3, vector_store_query_mode="hybrid", embed_model=embed_model, alpha=0.5)
|
302 |
|
303 |
chat_engine_bandi_sistema_puglia = ContextChatEngine(retriever=retriever_bandi_sistema_puglia,
|
304 |
context_template="Sei un chatbot in grado di rispondere alle domande su bandi regionali e avvisi della regione Puglia. Hai accesso ai bandi della regione Puglia. Qui sotto le informazioni di contesto recuperate. \n"
|
|
|
473 |
else:
|
474 |
print("MODALITA STANDARD")
|
475 |
if(str(current_response_mode)=="tree_summarize"):
|
|
|
476 |
# define response synthesizer
|
477 |
response_synthesizer = get_response_synthesizer(streaming=True,response_mode="tree_summarize",text_qa_template=text_qa_template)
|
478 |
query_engine = None
|
479 |
query_engine = RetrieverQueryEngine(retriever=retriever, response_synthesizer=response_synthesizer)
|
|
|
|
|
|
|
480 |
stream_response = None
|
481 |
print(userMessage)
|
482 |
stream_response = query_engine.query(userMessage)
|