karthikvarunn commited on
Commit
09a1990
·
verified ·
1 Parent(s): 93c7674

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -27,7 +27,7 @@ voyage_api_key = os.getenv("VOYAGE_API_KEY")
27
  pc = Pinecone(api_key=pinecone_api_key)
28
 
29
  # Redis caching for reranking
30
- redis_client = redis.Redis(host='localhost', port=6379, db=0)
31
 
32
  # Initialize embeddings
33
  embeddings = VoyageAIEmbeddings(voyage_api_key=voyage_api_key, model="voyage-law-2")
@@ -135,9 +135,9 @@ def rerank_results(query, search_results):
135
  return search_results
136
 
137
  cache_key = f"rerank:{query}"
138
- cached_result = redis_client.get(cache_key)
139
- if cached_result:
140
- return json.loads(cached_result)
141
 
142
  # Prepare input pairs for reranking
143
  pairs = [(query, doc["text"]) for doc in search_results]
@@ -149,7 +149,7 @@ def rerank_results(query, search_results):
149
 
150
  sorted_results = sorted(search_results, key=lambda x: x["rerank_score"], reverse=True)
151
 
152
- redis_client.setex(cache_key, 600, json.dumps(sorted_results)) # Cache for 10 min
153
  return sorted_results
154
 
155
 
 
27
  pc = Pinecone(api_key=pinecone_api_key)
28
 
29
  # Redis caching for reranking
30
+ # redis_client = redis.Redis(host='localhost', port=6379, db=0)
31
 
32
  # Initialize embeddings
33
  embeddings = VoyageAIEmbeddings(voyage_api_key=voyage_api_key, model="voyage-law-2")
 
135
  return search_results
136
 
137
  cache_key = f"rerank:{query}"
138
+ # cached_result = redis_client.get(cache_key)
139
+ # if cached_result:
140
+ # return json.loads(cached_result)
141
 
142
  # Prepare input pairs for reranking
143
  pairs = [(query, doc["text"]) for doc in search_results]
 
149
 
150
  sorted_results = sorted(search_results, key=lambda x: x["rerank_score"], reverse=True)
151
 
152
+ # redis_client.setex(cache_key, 600, json.dumps(sorted_results)) # Cache for 10 min
153
  return sorted_results
154
 
155