wassname/stampy-chat
1
0
Fork 0
mirror of https://github.com/wassname/stampy-chat.git synced 2026-09-12 13:00:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

Handle both old and new pinecode sturctures

Browse Source
This commit is contained in:
Daniel O'Connell
2023-09-29 14:04:20 +02:00
parent fd0cbfc1b9
commit 677f3ba264
1 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/src/stampy_chat/get_blocks.py
+10 -9
View File
@@ -93,8 +93,9 @@ def get_top_k_blocks(index, user_query: str, k: int) -> List[Block]:
)
blocks = []
for match in query_response['matches']:
metadata = match['metadata']
date = match['metadata']['date_published']
date = metadata.get('date_published') or metadata.get('date')
if isinstance(date, datetime.date):
date = date.isoformat()
@@ -103,18 +104,18 @@ def get_top_k_blocks(index, user_query: str, k: int) -> List[Block]:
elif isinstance(date, float):
date = datetime.datetime.fromtimestamp(date).date().isoformat()
authors = match['metadata'].get('authors')
if not authors and match['metadata'].get('author'):
authors = [match['metadata'].get('author')]
authors = metadata.get('authors')
if not authors and metadata.get('author'):
authors = [metadata.get('author')]
blocks.append(Block(
id = match['metadata']['hash_id'],
title = match['metadata']['title'],
id = metadata.get('hash_id'),
title = metadata['title'],
authors = authors,
date = date,
url = match['metadata']['url'],
tags = match['metadata'].get('tags'),
text = strip_block(match['metadata']['text'])
url = metadata['url'],
tags = metadata.get('tags'),
text = strip_block(metadata['text'])
))
t2 = time.time()