Spaces:
Runtime error
Runtime error
code shuffle
Browse files
app.py
CHANGED
|
@@ -12,6 +12,16 @@ from langchain.chains import ConversationChain
|
|
| 12 |
from langchain.retrievers.self_query.base import SelfQueryRetriever
|
| 13 |
from langchain.chains.query_constructor.base import AttributeInfo
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
metadata_field_info=[
|
| 16 |
AttributeInfo(
|
| 17 |
name="author",
|
|
@@ -30,18 +40,33 @@ metadata_field_info=[
|
|
| 30 |
),
|
| 31 |
]
|
| 32 |
|
| 33 |
-
document_content_description = "Excerpt's from Reid Hoffman's book Impromptu"
|
| 34 |
-
embeddings = OpenAIEmbeddings()
|
| 35 |
-
|
| 36 |
-
pinecone.init(
|
| 37 |
-
api_key=str(os.environ['PINECONE_API_KEY']),
|
| 38 |
-
environment=str(os.environ['PINECONE_ENV']))
|
| 39 |
|
| 40 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
|
| 42 |
def load_chain():
|
| 43 |
-
vectorstore = Pinecone.from_existing_index(index_name, embeddings)
|
| 44 |
-
llm = OpenAI(temperature=0)
|
| 45 |
retriever = SelfQueryRetriever.from_llm(
|
| 46 |
llm,
|
| 47 |
vectorstore,
|
|
|
|
| 12 |
from langchain.retrievers.self_query.base import SelfQueryRetriever
|
| 13 |
from langchain.chains.query_constructor.base import AttributeInfo
|
| 14 |
|
| 15 |
+
pinecone.init(
|
| 16 |
+
api_key=str(os.environ['PINECONE_API_KEY']),
|
| 17 |
+
environment=str(os.environ['PINECONE_ENV']))
|
| 18 |
+
|
| 19 |
+
llm = OpenAI(temperature=0)
|
| 20 |
+
embeddings = OpenAIEmbeddings()
|
| 21 |
+
vectorstore = Pinecone.from_existing_index("impromptu", embeddings)
|
| 22 |
+
document_content_description = ("Excerpts from the book Impromptu, "
|
| 23 |
+
"jointly authored by Reid Hoffman, "
|
| 24 |
+
"GPT-3, and GPT 4.")
|
| 25 |
metadata_field_info=[
|
| 26 |
AttributeInfo(
|
| 27 |
name="author",
|
|
|
|
| 40 |
),
|
| 41 |
]
|
| 42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
|
| 44 |
+
# metadata_field_info=[
|
| 45 |
+
# AttributeInfo(
|
| 46 |
+
# name="author",
|
| 47 |
+
# description="The author of the excerpt",
|
| 48 |
+
# type="string or list[string]",
|
| 49 |
+
# ),
|
| 50 |
+
# AttributeInfo(
|
| 51 |
+
# name="chapter_number",
|
| 52 |
+
# description="The chapter number of excerpt",
|
| 53 |
+
# type="integer",
|
| 54 |
+
# ),
|
| 55 |
+
# AttributeInfo(
|
| 56 |
+
# name="chapter_name",
|
| 57 |
+
# description="The chapter name of the excerpt",
|
| 58 |
+
# type="string",
|
| 59 |
+
# ),
|
| 60 |
+
# ]
|
| 61 |
+
|
| 62 |
+
# document_content_description = "Excerpt's from Reid Hoffman's book Impromptu"
|
| 63 |
+
# embeddings = OpenAIEmbeddings()
|
| 64 |
+
|
| 65 |
+
# index_name = str(os.environ['PINECONE_INDEX_NAME'])
|
| 66 |
|
| 67 |
def load_chain():
|
| 68 |
+
# vectorstore = Pinecone.from_existing_index(index_name, embeddings)
|
| 69 |
+
# llm = OpenAI(temperature=0)
|
| 70 |
retriever = SelfQueryRetriever.from_llm(
|
| 71 |
llm,
|
| 72 |
vectorstore,
|