Prakhar Bhandari commited on
Commit
ae26cc4
·
1 Parent(s): 05b774f

Added updated kg_builder

Browse files
kg_builder/src/knowledge_graph_builder.py CHANGED
@@ -65,6 +65,9 @@ def map_to_base_relationship(rel: Relationship) -> BaseRelationship:
65
  )
66
 
67
  import os
 
 
 
68
  from langchain.chains.openai_functions import (
69
  create_openai_fn_chain,
70
  create_structured_output_runnable,
@@ -74,7 +77,7 @@ from langchain_openai import ChatOpenAI
74
  from langchain.prompts import ChatPromptTemplate
75
 
76
  # Setting the OpenAI API key for usage in LLM calls
77
- os.environ["OPENAI_API_KEY"] = "sk-proj-hceIL56CC2zfjAvAlMjbT3BlbkFJyHKX2wbiQxsG9yy8dGJN"
78
  llm = ChatOpenAI(model="gpt-3.5-turbo-16k", temperature=0)
79
 
80
  def get_extraction_chain(
 
65
  )
66
 
67
  import os
68
+ from dotenv import load_dotenv
69
+ load_dotenv() # This loads the variables from .env into os.environ
70
+
71
  from langchain.chains.openai_functions import (
72
  create_openai_fn_chain,
73
  create_structured_output_runnable,
 
77
  from langchain.prompts import ChatPromptTemplate
78
 
79
  # Setting the OpenAI API key for usage in LLM calls
80
+ os.environ["OPENAI_API_KEY"]
81
  llm = ChatOpenAI(model="gpt-3.5-turbo-16k", temperature=0)
82
 
83
  def get_extraction_chain(