From 6a87248a0a02debf9c5ebee19b147400c25dac3d Mon Sep 17 00:00:00 2001 From: github-actions <${GITHUB_ACTOR}@users.noreply.github.com> Date: Sun, 5 Feb 2023 23:00:13 +0000 Subject: [PATCH] Format Python code with psf/black push --- models/search_model.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/models/search_model.py b/models/search_model.py index ac6c756..7d4ede3 100644 --- a/models/search_model.py +++ b/models/search_model.py @@ -56,7 +56,9 @@ class Search: if response.status == 200: data = await response.json() # Return a list of the top 5 links - return [item["link"] for item in data["items"][:search_scope]], [item["link"] for item in data["items"]] + return [item["link"] for item in data["items"][:search_scope]], [ + item["link"] for item in data["items"] + ] else: return "An error occurred while searching." @@ -121,7 +123,9 @@ class Search: embedding_model = OpenAIEmbedding() index = GPTSimpleVectorIndex(documents, embed_model=embedding_model) - await self.usage_service.update_usage(embedding_model.last_token_usage, embeddings=True) + await self.usage_service.update_usage( + embedding_model.last_token_usage, embeddings=True + ) llm_predictor = LLMPredictor(llm=OpenAI(model_name="text-davinci-003")) # Now we can search the index for a query: