giannantonio15 commited on
Commit
4b2f203
·
verified ·
1 Parent(s): 65d6f57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -11
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.2)
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.4)
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.4)
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)