tomascufaro commited on
Commit
15bcf0e
·
1 Parent(s): 64fd6d4

addin apikey

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,6 +9,7 @@ from langchain import hub
9
  from langchain_openai import ChatOpenAI, OpenAIEmbeddings
10
  from langchain_core.runnables import RunnablePassthrough
11
  from langchain_core.output_parsers import StrOutputParser
 
12
  import gradio as gr
13
 
14
 
@@ -60,7 +61,7 @@ def wrap_all(file, input_prompt:str):
60
  db = doc_to_embeddings(data)
61
  retriever = db.as_retriever()
62
  prompt = hub.pull("rlm/rag-prompt")
63
- llm = ChatOpenAI(model_name="gpt-4", temperature=0)
64
  rag_chain = (
65
  {"context": retriever | format_docs, "question": RunnablePassthrough()}
66
  | prompt
 
9
  from langchain_openai import ChatOpenAI, OpenAIEmbeddings
10
  from langchain_core.runnables import RunnablePassthrough
11
  from langchain_core.output_parsers import StrOutputParser
12
+ import os
13
  import gradio as gr
14
 
15
 
 
61
  db = doc_to_embeddings(data)
62
  retriever = db.as_retriever()
63
  prompt = hub.pull("rlm/rag-prompt")
64
+ llm = ChatOpenAI(model_name="gpt-4",openai_api_key=os.environ['OpenAI_APIKEY'], temperature=0)
65
  rag_chain = (
66
  {"context": retriever | format_docs, "question": RunnablePassthrough()}
67
  | prompt