luanpoppe commited on
Commit
fb553fb
·
1 Parent(s): 869c9f9
Files changed (1) hide show
  1. langchain_backend/utils.py +3 -2
langchain_backend/utils.py CHANGED
@@ -10,8 +10,6 @@ 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,6 +34,9 @@ def create_prompt_llm_chain(system_prompt, modelParam="gpt-4o-mini"):
36
  if modelParam == "gpt-4o-mini":
37
  model = ChatOpenAI(model=modelParam)
38
  else:
 
 
 
39
 
40
  model = HuggingFaceEndpoint(
41
  repo_id=modelParam,
 
10
 
11
  os.environ.get("OPENAI_API_KEY")
12
  os.environ.get("HUGGINGFACEHUB_API_TOKEN")
 
 
13
 
14
  def getPDF(file_path="./nike.pdf"):
15
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
 
34
  if modelParam == "gpt-4o-mini":
35
  model = ChatOpenAI(model=modelParam)
36
  else:
37
+
38
+ print('\n\n\n')
39
+ print(os.environ.get("HUGGINGFACEHUB_API_TOKEN"))
40
 
41
  model = HuggingFaceEndpoint(
42
  repo_id=modelParam,