luanpoppe commited on
Commit
a37a365
·
1 Parent(s): d0443c0
Files changed (1) hide show
  1. langchain_backend/utils.py +3 -1
langchain_backend/utils.py CHANGED
@@ -9,6 +9,9 @@ from langchain_core.prompts import ChatPromptTemplate
9
  from langchain_huggingface import HuggingFaceEndpoint
10
 
11
  os.environ.get("OPENAI_API_KEY")
 
 
 
12
 
13
  def getPDF(file_path="./nike.pdf"):
14
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
@@ -33,7 +36,6 @@ def create_prompt_llm_chain(system_prompt, modelParam="gpt-4o-mini"):
33
  if modelParam == "gpt-4o-mini":
34
  model = ChatOpenAI(model=modelParam)
35
  else:
36
- os.environ.get("HUGGINGFACEHUB_API_TOKEN")
37
 
38
  llm = HuggingFaceEndpoint(
39
  repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
 
9
  from langchain_huggingface import HuggingFaceEndpoint
10
 
11
  os.environ.get("OPENAI_API_KEY")
12
+ os.environ.get("HUGGINGFACEHUB_API_TOKEN")
13
+ print('\n\n\n')
14
+ print(os.environ.get("HUGGINGFACEHUB_API_TOKEN"))
15
 
16
  def getPDF(file_path="./nike.pdf"):
17
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
 
36
  if modelParam == "gpt-4o-mini":
37
  model = ChatOpenAI(model=modelParam)
38
  else:
 
39
 
40
  llm = HuggingFaceEndpoint(
41
  repo_id="meta-llama/Meta-Llama-3-8B-Instruct",