leksval
commited on
Commit
·
9a9f096
1
Parent(s):
7d76d0d
update db for hf space
Browse files- database.py +7 -5
database.py
CHANGED
@@ -19,13 +19,15 @@ class DatabaseManager:
|
|
19 |
Connects to the existing langfuse-db PostgreSQL instance
|
20 |
"""
|
21 |
|
|
|
|
|
22 |
def __init__(self):
|
23 |
self.db_config = {
|
24 |
-
'host': 'langfuse-db',
|
25 |
-
'port': 5432,
|
26 |
-
'database': 'langfuse',
|
27 |
-
'user': 'langfuse',
|
28 |
-
'password': 'langfuse'
|
29 |
}
|
30 |
self.init_database()
|
31 |
|
|
|
19 |
Connects to the existing langfuse-db PostgreSQL instance
|
20 |
"""
|
21 |
|
22 |
+
import os
|
23 |
+
|
24 |
def __init__(self):
|
25 |
self.db_config = {
|
26 |
+
'host': os.getenv('DB_HOST', 'langfuse-db'),
|
27 |
+
'port': int(os.getenv('DB_PORT', 5432)),
|
28 |
+
'database': os.getenv('DB_NAME', 'langfuse'),
|
29 |
+
'user': os.getenv('DB_USER', 'langfuse'),
|
30 |
+
'password': os.getenv('DB_PASSWORD', 'langfuse')
|
31 |
}
|
32 |
self.init_database()
|
33 |
|