Ludovicollin commited on
Commit
f6586a4
·
1 Parent(s): 6396156

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -33,13 +33,13 @@ index_name = os.environ['PINECONE_INDEX_NAME']
33
  embeddings = HuggingFaceEmbeddings()
34
  pinecone.init(
35
  api_key=os.environ['PINECONE_API_KEY'],
36
- environment=os.environ['PINECONE_ENVIRONMENT']
37
  )
38
 
39
  @cl.cache
40
  def retriever_to_cache():
41
  vectorstore = Pinecone.from_existing_index(
42
- index_name=index_name, embedding=embeddings
43
  )
44
  retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 30,"filter": {'categorie': {'$eq': 'OF'}}})
45
  return retriever
 
33
  embeddings = HuggingFaceEmbeddings()
34
  pinecone.init(
35
  api_key=os.environ['PINECONE_API_KEY'],
36
+ environment="us-west4-gcp-free"
37
  )
38
 
39
  @cl.cache
40
  def retriever_to_cache():
41
  vectorstore = Pinecone.from_existing_index(
42
+ index_name="all-jdlp", embedding=embeddings
43
  )
44
  retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 30,"filter": {'categorie': {'$eq': 'OF'}}})
45
  return retriever