Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|
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
|
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 |
|