Update ki_gen/data_retriever.py

ki_gen/data_retriever.py

@@ -13,7 +13,8 @@ from langchain_community.graphs import Neo4jGraph
 from langchain_community.chains.graph_qa.cypher_utils import CypherQueryCorrector, Schema
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import ChatPromptTemplate
-from langchain_core.pydantic_v1 import BaseModel, Field
+from langchain_core.pydantic_v1 import Field
+from pydantic import BaseModel
 from langchain_groq import ChatGroq
 
 from langgraph.graph import StateGraph