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 |
|