fix: circular `db` import

app.py

@@ -11,6 +11,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from gradio import ChatMessage
 import textwrap
 from tools import *
+from db.connection import db
 load_dotenv()
 os.environ['LANGCHAIN_PROJECT'] = 'gradio-test'
 

db/connection.py

@@ -0,0 +1,3 @@
+from langchain_community.utilities import SQLDatabase
+
+db = SQLDatabase.from_uri("sqlite:///db/Bahrain_2023_Q.db")

tools/__init__.py

@@ -1,4 +1,3 @@
-from langchain_community.utilities import SQLDatabase
 from rich.console import Console
 from .driver_performance import GetDriverPerformance
 from .event_performance import GetEventPerformance
@@ -8,8 +7,6 @@ from .weather_impact import GetWeatherImpact
 
 console = Console(style="chartreuse1 on grey7")
 
-db = SQLDatabase.from_uri("sqlite:///db/Bahrain_2023_Q.db")
-
 __all__ = [
     "GetDriverPerformance",
     "GetEventPerformance",

tools/driver_performance.py

@@ -1,6 +1,7 @@
 from pydantic import BaseModel, Field
 from langchain_core.tools import BaseTool
-from . import db, console
+from db.connection import db
+from . import console
 
 
 class GetDriverPerformanceOutput(BaseModel):

tools/event_performance.py

@@ -1,7 +1,8 @@
 from pydantic import BaseModel, Field
 from typing import Type
 from langchain_core.tools import BaseTool
-from . import db, console
+from db.connection import db
+from . import console
 
 
 class GetEventPerformanceOutput(BaseModel):

tools/telemetry_analysis.py

@@ -1,7 +1,8 @@
 from pydantic import BaseModel, Field
 from typing import Type
 from langchain_core.tools import BaseTool
-from . import db, console
+from db.connection import db
+from . import console
 
 
 class GetTelemetryAndWeatherInput(BaseModel):

tools/tyre_performance.py

@@ -1,7 +1,8 @@
 from pydantic import BaseModel, Field
 from typing import Type
 from langchain_core.tools import BaseTool
-from . import db, console
+from db.connection import db
+from . import console
 
 
 class GetTyrePerformanceInput(BaseModel):

tools/weather_impact.py

@@ -1,10 +1,8 @@
 from pydantic import BaseModel, Field
 from typing import Type
 from langchain_core.tools import BaseTool
-from rich.console import Console
-from . import db
-
-console = Console(style="chartreuse1 on grey7")
+from db.connection import db
+from . import console
 
 
 class GetWeatherImpactInput(BaseModel):