Penality commited on
Commit
053606e
·
verified ·
1 Parent(s): 08655c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -2
app.py CHANGED
@@ -77,9 +77,29 @@ def retrieve_document(query):
77
  embeddings_ = requests.get(API_URL_EMBEDDINGS)
78
  metadata_ = requests.get(API_URL_METADATA)
79
 
80
- embeddings_file = embeddings_.json()
81
- metadata_file = metadata_.json()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
82
  print(embeddings_file, metadata_file)
 
83
  # Generate query embedding
84
  query_embedding = embedding_model.encode([query]).astype(np.float32)
85
 
 
77
  embeddings_ = requests.get(API_URL_EMBEDDINGS)
78
  metadata_ = requests.get(API_URL_METADATA)
79
 
80
+ # Check for errors before parsing JSON
81
+ if embeddings_.status_code != 200:
82
+ print(f"Error fetching embeddings: {embeddings_.status_code} - {embeddings_.text}")
83
+ return None
84
+
85
+ if metadata_.status_code != 200:
86
+ print(f"Error fetching metadata: {metadata_.status_code} - {metadata_.text}")
87
+ return None
88
+
89
+ try:
90
+ embeddings_file = embeddings_.json()
91
+ except requests.exceptions.JSONDecodeError as e:
92
+ print(f"Error decoding embeddings JSON: {e}")
93
+ return None
94
+
95
+ try:
96
+ metadata_file = metadata_.json()
97
+ except requests.exceptions.JSONDecodeError as e:
98
+ print(f"Error decoding metadata JSON: {e}")
99
+ return None
100
+
101
  print(embeddings_file, metadata_file)
102
+
103
  # Generate query embedding
104
  query_embedding = embedding_model.encode([query]).astype(np.float32)
105