FilesExpand file tree

 main

/

config.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

60 lines (49 loc) · 1.28 KB

 main

/

config.py

Copy path

Top

File metadata and controls

• Code
• Blame

60 lines (49 loc) · 1.28 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

from os import getenv

import openai

import pinecone

from dotenv import load_dotenv

from sqlalchemy import create_engine

import sentry_sdk

from sentry_sdk.integrations.flask import FlaskIntegration

load_dotenv()

ENV = getenv("ENVIRONMENT") or "unknown"

DB_URL = getenv("DB_URL")

OPENAI_KEY = getenv("OPENAI_KEY")

PINECONE_KEY = getenv("PINECONE_KEY")

PINECONE_ENV = getenv("PINECONE_ENV")

EVENTS_URL = getenv("EVENTS_URL")

SENTRY_URL = getenv("SENTRY_URL")

if SENTRY_URL:

sentry_sdk.init(

dsn=SENTRY_URL,

environment=ENV,

integrations=[

FlaskIntegration(),

],

traces_sample_rate=1.0

)

openai.api_key = OPENAI_KEY

class FlaskAppConfig:

CORS_HEADERS = "Content-Type"

SQLALCHEMY_DATABASE_URI = DB_URL

SQLALCHEMY_ENGINE_OPTIONS = {"pool_pre_ping": True}

TIMEOUT = 60

if DB_URL:

ENGINE = create_engine(DB_URL)

dialect_mapping = {

"postgresql": "PostgreSQL 15.2",

"mysql": "MySQL",

}

DIALECT = dialect_mapping.get(ENGINE.dialect.name)

else:

print('DB_URL not found, please check your environment')

exit(1)

if EVENTS_URL:

EVENTS_ENGINE = create_engine(EVENTS_URL)

else:

EVENTS_ENGINE = None

if PINECONE_KEY and PINECONE_ENV:

pinecone.init(

api_key=PINECONE_KEY,

environment=PINECONE_ENV,

)