removed unused imports

Author: VinciGit00

scrapegraphai/builders/graph_builder.py

@@ -4,10 +4,9 @@
 
 from langchain_core.prompts import ChatPromptTemplate
 from langchain.chains import create_extraction_chain
-from ..models import OpenAI, Gemini
+from ..models import Gemini
 from ..helpers import nodes_metadata, graph_schema
-from ..models.ernie import Ernie
-
+from langchain_openai import ChatOpenAI
 
 class GraphBuilder:
     """
@@ -71,7 +70,7 @@ def _create_llm(self, llm_config: dict):
 
         # select the model based on the model name
         if "gpt-" in llm_params["model"]:
-            return OpenAI(llm_params)
+            return ChatOpenAI(llm_params)
         elif "gemini" in llm_params["model"]:
             return Gemini(llm_params)
         elif "ernie" in llm_params["model"]:

scrapegraphai/graphs/abstract_graph.py

@@ -135,7 +135,6 @@ def _create_llm(self, llm_config: dict) -> object:
                 raise KeyError("model_tokens not specified") from exc
             return llm_params["model_instance"]
 
-        # Instantiate the language model based on the model name (models that use the common interface)
         def handle_model(model_name, provider, token_key, default_token=8192):
             try:
                 self.model_token = models_tokens[provider][token_key]