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

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -29,17 +29,17 @@ import datetime
29
 
30
  os.environ["TOKENIZERS_PARALLELISM"] = os.environ["TOKENIZERS_PARALLELISM"]
31
  os.environ['ANTHROPIC_API_KEY'] = os.environ['ANTHROPIC_API_KEY']
32
- index_name = os.environ['PINECONE_INDEX_NAME']
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
 
29
 
30
  os.environ["TOKENIZERS_PARALLELISM"] = os.environ["TOKENIZERS_PARALLELISM"]
31
  os.environ['ANTHROPIC_API_KEY'] = os.environ['ANTHROPIC_API_KEY']
32
+ #index_name = os.environ['PINECONE_INDEX_NAME']
33
  embeddings = HuggingFaceEmbeddings()
34
  pinecone.init(
35
  api_key=os.environ['PINECONE_API_KEY'],
36
  environment="us-west4-gcp-free"
37
  )
38
+ index_name = pinecone.Index("all-jdlp")
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