devve1 commited on
Commit
69d7f54
1 Parent(s): b256ccf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -124,7 +124,14 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
124
  metadatas = [doc['metadata'] for doc in docs]
125
 
126
  context = "\n".join(contents)
127
- result_metadatas = "\n".join([f'{key}: {value}' for metadata in metadatas for key, value in metadata.items()])
 
 
 
 
 
 
 
128
 
129
  response = llm.create_chat_completion(
130
  messages = [
@@ -161,7 +168,7 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
161
  ], stop=["</s>"], temperature=0.7, max_tokens=3000)
162
 
163
  answer = output['choices'][0]['message']['content']
164
- answer_with_metadatas = f"{answer}\n\n\n\nSources :\n\n{result_metadatas}"
165
 
166
  print(f'OUTPUT: {output}')
167
  return answer, answer_with_metadatas
@@ -401,7 +408,7 @@ if __name__ == '__main__':
401
  )
402
 
403
  texts, metadatas = [], []
404
- for doc in docs:
405
  texts.append(elem.text)
406
  metadatas.append(elem.metadata.to_dict())
407
 
 
124
  metadatas = [doc['metadata'] for doc in docs]
125
 
126
  context = "\n".join(contents)
127
+
128
+ seen_values = set()
129
+ result_metadatas = "\n".join(
130
+ f'{key}: {value}'
131
+ for metadata in metadatas
132
+ for key, value in metadata.items()
133
+ if (value not in seen_values and not seen_values.add(value))
134
+ )
135
 
136
  response = llm.create_chat_completion(
137
  messages = [
 
168
  ], stop=["</s>"], temperature=0.7, max_tokens=3000)
169
 
170
  answer = output['choices'][0]['message']['content']
171
+ answer_with_metadatas = f"{answer}\n\n{result_metadatas}"
172
 
173
  print(f'OUTPUT: {output}')
174
  return answer, answer_with_metadatas
 
408
  )
409
 
410
  texts, metadatas = [], []
411
+ for elem in elements:
412
  texts.append(elem.text)
413
  metadatas.append(elem.metadata.to_dict())
414