From d5c098de7393efd5527a0da7f7d989e56055ccd2 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Tue, 17 Oct 2023 10:21:05 +0300 Subject: [PATCH] another protection --- pr_agent/tools/pr_similar_issue.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/pr_agent/tools/pr_similar_issue.py b/pr_agent/tools/pr_similar_issue.py index 0a624f85..9ccd73a6 100644 --- a/pr_agent/tools/pr_similar_issue.py +++ b/pr_agent/tools/pr_similar_issue.py @@ -1,3 +1,4 @@ +import time from enum import Enum from typing import List @@ -46,6 +47,13 @@ class PRSimilarIssue: # check if index exists, and if repo is already indexed run_from_scratch = False + if run_from_scratch: + if not index_name in pinecone.list_indexes(): + get_logger().info('Removing index...') + pinecone.init(api_key=api_key, environment=environment) + pinecone.delete_index(index_name) + get_logger().info('Done') + upsert = True pinecone.init(api_key=api_key, environment=environment) if not index_name in pinecone.list_indexes(): @@ -246,6 +254,7 @@ class PRSimilarIssue: if not upsert: get_logger().info('Creating index from scratch...') ds.to_pinecone_index(self.index_name, api_key=api_key, environment=environment) + time.sleep(15) # wait for pinecone to finalize indexing before querying else: get_logger().info('Upserting index...') namespace = "" @@ -253,6 +262,7 @@ class PRSimilarIssue: concurrency: int = 10 pinecone.init(api_key=api_key, environment=environment) ds._upsert_to_index(self.index_name, namespace, batch_size, concurrency) + time.sleep(5) # wait for pinecone to finalize upserting before querying get_logger().info('Done')