gufett0 commited on
Commit
b7c9620
·
1 Parent(s): e8f0dba

added memory

Browse files
Files changed (1) hide show
  1. backend.py +17 -2
backend.py CHANGED
@@ -88,6 +88,17 @@ def handle_query(query_str, chat_history) -> Iterator[str]:
88
  verbose=True,
89
  )"""
90
 
 
 
 
 
 
 
 
 
 
 
 
91
  chat_engine = index.as_chat_engine(
92
  chat_mode="condense_plus_context",
93
  memory=memory,
@@ -97,14 +108,18 @@ def handle_query(query_str, chat_history) -> Iterator[str]:
97
  context_prompt = (
98
  "Sei un assistente Q&A italiano di nome Odi, che risponde solo alle domande o richieste pertinenti in modo preciso."
99
  " Quando un utente ti chiede informazioni su di te o sul tuo creatore puoi dire che sei un assistente ricercatore creato dagli Osservatori Digitali e fornire gli argomenti di cui sei esperto."
100
- "Ecco i documenti rilevanti per il contesto:\n"
101
  "{context_str}"
102
- "\nIstruzione: Usa la cronologia delle chat precedenti, o il contesto sopra, per interagire e aiutare l'utente."
103
  ),
104
 
 
 
105
  )
106
 
107
 
 
 
108
  outputs = []
109
  response = chat_engine.stream_chat(query_str)
110
  #response = chat_engine.chat(query_str)
 
88
  verbose=True,
89
  )"""
90
 
91
+
92
+ conversation = []
93
+ for user, assistant in chat_history:
94
+ conversation.extend(
95
+ [
96
+ {"role": "user", "content": user},
97
+ {"role": "assistant", "content": assistant},
98
+ ]
99
+ )
100
+ conversation.append({"role": "user", "content": query_str})
101
+
102
  chat_engine = index.as_chat_engine(
103
  chat_mode="condense_plus_context",
104
  memory=memory,
 
108
  context_prompt = (
109
  "Sei un assistente Q&A italiano di nome Odi, che risponde solo alle domande o richieste pertinenti in modo preciso."
110
  " Quando un utente ti chiede informazioni su di te o sul tuo creatore puoi dire che sei un assistente ricercatore creato dagli Osservatori Digitali e fornire gli argomenti di cui sei esperto."
111
+ " Ecco i documenti rilevanti per il contesto:\n"
112
  "{context_str}"
113
+ "\nIstruzione: Usa la cronologia delle chat precedenti, o il contesto sopra, per interagire e aiutare l'utente a rispondere alla sua domanda: {query_str}"
114
  ),
115
 
116
+ verbose=False,
117
+
118
  )
119
 
120
 
121
+
122
+
123
  outputs = []
124
  response = chat_engine.stream_chat(query_str)
125
  #response = chat_engine.chat(query_str)