Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
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 |
|