danicafisher commited on
Commit
d8a71d0
1 Parent(s): 7bd859c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,7 +6,7 @@ from aimakerspace.openai_utils.prompts import (
6
  )
7
  from aimakerspace.vectordatabase import VectorDatabase
8
  from aimakerspace.openai_utils.chatmodel import ChatOpenAI
9
- from aimakerspace.openai_utils.embeddings import OpenAIEmbeddings
10
  import chainlit as cl
11
  import nest_asyncio
12
  nest_asyncio.apply()
@@ -72,9 +72,9 @@ async def start_chat():
72
  }
73
  cl.user_session.set("settings", settings)
74
 
75
- embeddings = OpenAIEmbeddings(model="text-embedding-3-small")
76
  # Create a vector store
77
- vector_db = VectorDatabase(embedding_model = embeddings)
78
  vector_db = await vector_db.abuild_from_list(split_documents_NIST)
79
  vector_db = await vector_db.abuild_from_list(split_documents_Blueprint)
80
 
 
6
  )
7
  from aimakerspace.vectordatabase import VectorDatabase
8
  from aimakerspace.openai_utils.chatmodel import ChatOpenAI
9
+ from langchain_community.embeddings import OpenAIEmbeddings
10
  import chainlit as cl
11
  import nest_asyncio
12
  nest_asyncio.apply()
 
72
  }
73
  cl.user_session.set("settings", settings)
74
 
75
+ embeddings = OpenAIEmbeddings()
76
  # Create a vector store
77
+ vector_db = VectorDatabase()
78
  vector_db = await vector_db.abuild_from_list(split_documents_NIST)
79
  vector_db = await vector_db.abuild_from_list(split_documents_Blueprint)
80