Format Python code with psf/black push

github-actions 2 years ago
parent bea1ff903d
commit fed556a076

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

Loading…
Cancel
Save