Fix database URL in db.py and update import path for schemas in llm_parser.py
This commit is contained in:
+1
-1
@@ -9,7 +9,7 @@ from sqlalchemy.orm import Session, sessionmaker
|
|||||||
Base = declarative_base()
|
Base = declarative_base()
|
||||||
|
|
||||||
# Database configuration
|
# Database configuration
|
||||||
DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///investors_2.db")
|
DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///investors.db")
|
||||||
|
|
||||||
# Create engine
|
# Create engine
|
||||||
engine = create_engine(DATABASE_URL, echo=False)
|
engine = create_engine(DATABASE_URL, echo=False)
|
||||||
|
|||||||
@@ -16,7 +16,6 @@ class InvestmentStage(enum.Enum):
|
|||||||
GROWTH = "GROWTH"
|
GROWTH = "GROWTH"
|
||||||
LATE_STAGE = "LATE_STAGE"
|
LATE_STAGE = "LATE_STAGE"
|
||||||
|
|
||||||
|
|
||||||
# Association table for many-to-many relationship between investors and companies
|
# Association table for many-to-many relationship between investors and companies
|
||||||
investor_company_association = Table(
|
investor_company_association = Table(
|
||||||
"investor_companies",
|
"investor_companies",
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ from dotenv import load_dotenv
|
|||||||
from openai import OpenAI
|
from openai import OpenAI
|
||||||
|
|
||||||
from db import get_session, init_database
|
from db import get_session, init_database
|
||||||
from schema import CSVRow, Investor
|
from py_schemas import CSVRow, Investor
|
||||||
|
|
||||||
# Load environment variables
|
# Load environment variables
|
||||||
load_dotenv()
|
load_dotenv()
|
||||||
|
|||||||
Reference in New Issue
Block a user