Update app.py

app.py

@@ -1,15 +1,12 @@
 import os
-import pinecone
 from functools import partial
-from langchain.vectorstores import Pinecone
-from langchain.embeddings.openai import OpenAIEmbeddings
-from langchain.chat_models import ChatOpenAI
+from langchain_openai import ChatOpenAI, OpenAIEmbeddings
+from langchain_pinecone import PineconeVectorStore
 import gradio as gr
 from utils import *
 
-
-# embedding model
-embedding_model_name = "text-embedding-ada-002"
+# set up OpenAI embedding endpoint
+embedding_model_name = "text-embedding-3-small"
 
 embedding_model = OpenAIEmbeddings(
     model=embedding_model_name,
@@ -17,25 +14,26 @@ embedding_model = OpenAIEmbeddings(
 )
 
 
-# llm
+# set up OpenAI chat-completion endpoint
 llm = ChatOpenAI(
     openai_api_key=os.getenv("OPENAI_API_KEY"),
-    model_name="gpt-4o",
-    temperature=0.0
+    model="gpt-4o",
+    temperature=0,
+    max_tokens=None
 )
 
 
-# vectorstore
+# set up Pinecone vector store
 index_name = "workouts"
+text_key = "text"
 
-pinecone.init(
-    api_key=os.getenv("PINECONE_API_KEY"),
-    environment=os.getenv("PINECONE_ENVIRONMENT")
+docsearch = PineconeVectorStore(
+    pinecone_api_key=os.getenv("PINECONE_API_KEY"),
+    index_name=index_name,
+    embedding=embedding_model,
+    text_key=text_key
 )
 
-text_field = "text"
-index = pinecone.Index(index_name)
-vectorstore = Pinecone(index, embedding_model, text_field)
 
 
 # prompt
@@ -105,7 +103,7 @@ with gr.Blocks(theme=gr.themes.Monochrome(radius_size=gr.themes.sizes.radius_sm)
             generation = gr.Markdown(value="")
 
     generate_button.click(
-        partial(run, vectorstore=vectorstore, system_prompt=system_prompt, llm=llm),
+        partial(run, vectorstore=docsearch, system_prompt=system_prompt, llm=llm),
         inputs=[gender, level, muscle_group, equipment, duration],
         outputs=generation
     )