Refactoring to deal with circularity

1 files changed, 3 insertions, 6 deletions

diff --git a/continuedev/src/continuedev/server/notebook.py b/continuedev/src/continuedev/server/notebook.py
index c9d4edc5..5eb151d7 100644
--- a/continuedev/src/continuedev/server/notebook.py
+++ b/continuedev/src/continuedev/server/notebook.py
@@ -6,20 +6,17 @@ from uvicorn.main import Server
 
 from ..models.filesystem_edit import FileEditWithFullContents
 from ..libs.policy import DemoPolicy
-from ..libs.core import Agent, FullState, History, Step
+from ..libs.core import FullState, History, Step
+from ..libs.agent import Agent
 from ..libs.steps.nate import ImplementAbstractMethodStep
 from ..libs.observation import Observation
-from dotenv import load_dotenv
 from ..libs.llm.openai import OpenAI
 from .ide_protocol import AbstractIdeProtocolServer
-import os
+from ..libs.env import openai_api_key
 import asyncio
 import nest_asyncio
 nest_asyncio.apply()
 
-load_dotenv()
-openai_api_key = os.getenv("OPENAI_API_KEY")
-
 router = APIRouter(prefix="/notebook", tags=["notebook"])
 
 # Graceful shutdown by closing websockets