Format Python code with psf/black push

github-actions 1 year ago
parent bea1ff903d
commit fed556a076

@ -149,7 +149,8 @@ class Index_handler:
def index_youtube_transcript(self, link, embed_model):
documents = YoutubeTranscriptReader().load_data(ytlinks=[link])
index = GPTSimpleVectorIndex(
documents, embed_model=embed_model,
documents,
embed_model=embed_model,
)
return index
@ -162,7 +163,8 @@ class Index_handler:
def index_discord(self, document, embed_model) -> GPTSimpleVectorIndex:
index = GPTSimpleVectorIndex(
document, embed_model=embed_model,
document,
embed_model=embed_model,
)
return index
@ -219,7 +221,9 @@ class Index_handler:
index = await self.loop.run_in_executor(
None, partial(self.index_file, temp_path, embedding_model)
)
await self.usage_service.update_usage(embedding_model.last_token_usage)
await self.usage_service.update_usage(
embedding_model.last_token_usage
)
file_name = file.filename
self.index_storage[ctx.user.id].add_index(index, ctx.user.id, file_name)

Loading…
Cancel
Save