Merge branch 'main' into feature/elastic-apm

weni-ai · Jun 13, 2024 · c5cee2e · c5cee2e
2 parents c80b265 + 54df6bb
commit c5cee2e
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 7 deletions.
diff --git a/app/config.py b/app/config.py
@@ -58,4 +58,5 @@ def __init__(self):
             "INDEX_CONTENTBASES_NAME", "content_bases"
         )
         self.sentry_dsn = os.environ.get("SENTRY_DSN", "")
+        self.environment = os.environ.get("ENVIRONMENT", "local")
         self.es_timeout = os.environ.get("ELASTICSEARCH_TIMEOUT", "30")
diff --git a/app/main.py b/app/main.py
@@ -6,6 +6,7 @@
 from langchain.embeddings import HuggingFaceHubEmbeddings, CohereEmbeddings
 from langchain.embeddings.base import Embeddings
 from langchain.vectorstores import ElasticVectorSearch, VectorStore, ElasticsearchStore
+from sentry_sdk.integrations.fastapi import FastApiIntegration
 
 from app.handlers import IDocumentHandler
 from app.handlers.products import ProductsHandler
@@ -60,6 +61,8 @@ def __init__(self, config: AppConfig):
         if config.sentry_dsn != "":
             sentry_sdk.init(
                 dsn=config.sentry_dsn,
+                integrations=[FastApiIntegration()],
+                environment=config.environment,
             )
 
         self.api = FastAPI()

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,7 +26,7 @@ python-dotenv = "^1.0.0"
 pydantic = "2.3.0"
 celery = "^5.3.6"
 redis = "^5.0.1"
-sentry-sdk = {extras = ["fastapi"], version = "^1.35.0"}
+sentry-sdk = {extras = ["fastapi"], version = "^2.5.1"}
 xlrd = "^2.0.1"
 pdfminer-six = "^20231228"
 cohere = "^4.37"