add sentry integration

app/config.py

@@ -24,3 +24,4 @@ def __init__(self):
  "HUGGINGFACE_API_TOKEN", "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"
  ),
  }
+ self.sentry_dsn = os.environ.get("SENTRY_DSN", "")

app/handlers/products.py

@@ -81,7 +81,6 @@ def index(self, request: ProductIndexRequest):
  docs = self.product_indexer.index(request.catalog_id, request.product)
  return ProductIndexResponse(catalog_id=request.catalog_id, documents=docs)
  except Exception as e:
- logger.error(msg=str(e))
  raise HTTPException(status_code=500, detail=[{"msg": str(e)}])
 
  def batch_index(self, request: ProductBatchIndexRequest):
@@ -91,7 +90,6 @@ def batch_index(self, request: ProductBatchIndexRequest):
  )
  return ProductIndexResponse(catalog_id=request.catalog_id, documents=docs)
  except Exception as e:
- logger.error(msg=str(e))
  raise HTTPException(status_code=500, detail=[{"msg": str(e)}])
 
  def search(self, request: ProductSearchRequest):
@@ -101,7 +99,6 @@ def search(self, request: ProductSearchRequest):
  )
  return ProductSearchResponse(products=matched_products)
  except Exception as e:
- logger.error(msg=str(e))
  raise HTTPException(status_code=500, detail=[{"msg": str(e)}])
 
  def delete(
@@ -117,7 +114,6 @@ def delete(
  docs = self.product_indexer.delete(catalog_id, product_retailer_id)
  return ProductIndexResponse(catalog_id=catalog_id, documents=docs)
  except Exception as e:
- logger.error(msg=str(e))
  raise HTTPException(status_code=500, detail=[{"msg": str(e)}])
 
  def delete_batch(self, request: ProductDeleteRequest):
@@ -127,5 +123,4 @@ def delete_batch(self, request: ProductDeleteRequest):
  )
  return ProductIndexResponse(catalog_id=request.catalog_id, documents=docs)
  except Exception as e:
- logger.error(msg=str(e))
  raise HTTPException(status_code=500, detail=[{"msg": str(e)}])

app/main.py

@@ -1,3 +1,4 @@
+import sentry_sdk
 from fastapi import FastAPI
 from langchain.embeddings import SagemakerEndpointEmbeddings, HuggingFaceHubEmbeddings
 from langchain.embeddings.base import Embeddings
@@ -38,6 +39,11 @@ def __init__(self, config: AppConfig):
  content_handler=content_handler,
  )
 
+ if config.sentry_dsn != "":
+ sentry_sdk.init(
+ dsn=config.sentry_dsn,
+ )
+
  self.api = FastAPI()
  self.vectorstore = ElasticVectorSearch(
  elasticsearch_url=config.es_url,

pyproject.toml

@@ -14,6 +14,7 @@ boto3 = "^1.28.41"
 fastapi = "^0.103.1"
 uvicorn = "^0.23.2"
 huggingface-hub = "^0.16.4"
+sentry-sdk = {extras = ["fastapi"], version = "^1.35.0"}
 
 
 [tool.poetry.group.dev.dependencies]