leksval commited on
Commit
9a9f096
·
1 Parent(s): 7d76d0d

update db for hf space

Browse files
Files changed (1) hide show
  1. 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