sharjeel1477 commited on
Commit
cb80c15
·
1 Parent(s): fc97274

update ask

Browse files
Files changed (1) hide show
  1. ask.py +8 -3
ask.py CHANGED
@@ -10,7 +10,8 @@ import os
10
 
11
  def askQuestion(brain, question, temperature, maxTokens):
12
  print(brain, question, temperature)
13
- Brain_Name = brain
 
14
  pinecone.init(api_key=os.environ['PINECONE_KEY'],
15
  environment="us-west4-gcp")
16
  pineconeindex = pinecone.Index(Brain_Name)
@@ -37,9 +38,13 @@ def askQuestion(brain, question, temperature, maxTokens):
37
  return response
38
 
39
 
40
- def getBrains():
41
  pinecone.init(api_key=os.environ['PINECONE_KEY'],
42
  environment="us-west4-gcp")
43
  active_indexes = pinecone.list_indexes()
44
  print(active_indexes)
45
- return active_indexes
 
 
 
 
 
10
 
11
  def askQuestion(brain, question, temperature, maxTokens):
12
  print(brain, question, temperature)
13
+ Brain_Name = brain.lower()
14
+ print(Brain_Name)
15
  pinecone.init(api_key=os.environ['PINECONE_KEY'],
16
  environment="us-west4-gcp")
17
  pineconeindex = pinecone.Index(Brain_Name)
 
38
  return response
39
 
40
 
41
+ def getBrains(name):
42
  pinecone.init(api_key=os.environ['PINECONE_KEY'],
43
  environment="us-west4-gcp")
44
  active_indexes = pinecone.list_indexes()
45
  print(active_indexes)
46
+ name = name.lower()
47
+ if name in active_indexes:
48
+ return True
49
+ else:
50
+ return False