Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,8 @@ from langchain.llms import OpenAI
|
|
11 |
|
12 |
from langchain.embeddings import OpenAIEmbeddings
|
13 |
|
|
|
|
|
14 |
|
15 |
from langchain.vectorstores import Chroma
|
16 |
|
@@ -20,8 +22,16 @@ def loading_pdf():
|
|
20 |
return "Loading..."
|
21 |
|
22 |
def pdf_changes(pdf_doc, open_ai_key):
|
23 |
-
if
|
24 |
os.environ['OPENAI_API_KEY'] = open_ai_key
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
loader = OnlinePDFLoader(pdf_doc.name)
|
26 |
documents = loader.load()
|
27 |
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
|
@@ -31,7 +41,7 @@ def pdf_changes(pdf_doc, open_ai_key):
|
|
31 |
retriever = db.as_retriever()
|
32 |
global qa
|
33 |
qa = ConversationalRetrievalChain.from_llm(
|
34 |
-
llm=
|
35 |
retriever=retriever,
|
36 |
return_source_documents=False)
|
37 |
return "Ready"
|
|
|
11 |
|
12 |
from langchain.embeddings import OpenAIEmbeddings
|
13 |
|
14 |
+
from langchain.llms import AzureOpenAI
|
15 |
+
import openai
|
16 |
|
17 |
from langchain.vectorstores import Chroma
|
18 |
|
|
|
22 |
return "Loading..."
|
23 |
|
24 |
def pdf_changes(pdf_doc, open_ai_key):
|
25 |
+
if True:
|
26 |
os.environ['OPENAI_API_KEY'] = open_ai_key
|
27 |
+
os.environ['OPENAI_API_TYPE'] = 'azure'
|
28 |
+
# API version to use (Azure has several)
|
29 |
+
os.environ['OPENAI_API_VERSION'] = '0301'
|
30 |
+
# base URL for your Azure OpenAI resource
|
31 |
+
os.environ['OPENAI_API_BASE'] = 'https://nand3.openai.azure.com'
|
32 |
+
|
33 |
+
llm = AzureOpenAI(deployment_name="Nand", model_name="gpt-35-turbo")
|
34 |
+
|
35 |
loader = OnlinePDFLoader(pdf_doc.name)
|
36 |
documents = loader.load()
|
37 |
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
|
|
|
41 |
retriever = db.as_retriever()
|
42 |
global qa
|
43 |
qa = ConversationalRetrievalChain.from_llm(
|
44 |
+
llm=AzureOpenAI(deployment_name="Nand", model_name="gpt-35-turbo"),
|
45 |
retriever=retriever,
|
46 |
return_source_documents=False)
|
47 |
return "Ready"
|