Add funcionality to add metadata to vector stores and vectore store files
This commit is contained in:
parent
70194de64c
commit
74e940c45d
22
modify_vector_store.py
Normal file
22
modify_vector_store.py
Normal file
|
@ -0,0 +1,22 @@
|
|||
from config import *
|
||||
import requests
|
||||
import json
|
||||
import sys
|
||||
|
||||
def modify_vector_store(vector_store_id,metadata):
|
||||
url = base_url+"vector_stores/"+vector_store_id
|
||||
|
||||
payload = json.dumps({
|
||||
"metadata": json.loads(metadata)
|
||||
})
|
||||
headers = {
|
||||
'Content-Type': 'application/json',
|
||||
'Authorization': 'Bearer '+secret_key,
|
||||
}
|
||||
|
||||
response = requests.request("POST", url, headers=headers, data=payload)
|
||||
|
||||
print(response.text)
|
||||
|
||||
if __name__ == '__main__':
|
||||
modify_vector_store(str(sys.argv[1]),str(sys.argv[2]))
|
22
update_vector_store_file.py
Normal file
22
update_vector_store_file.py
Normal file
|
@ -0,0 +1,22 @@
|
|||
from config import *
|
||||
import requests
|
||||
import json
|
||||
import sys
|
||||
|
||||
def update_vector_store_file(vector_store_id,file_id,attributes):
|
||||
url = base_url+"vector_stores/"+vector_store_id+"/files/"+file_id
|
||||
|
||||
payload = json.dumps({
|
||||
"attributes": json.loads(attributes)
|
||||
})
|
||||
headers = {
|
||||
'Content-Type': 'application/json',
|
||||
'Authorization': 'Bearer '+secret_key,
|
||||
}
|
||||
|
||||
response = requests.request("POST", url, headers=headers, data=payload)
|
||||
|
||||
print(response.text)
|
||||
|
||||
if __name__ == '__main__':
|
||||
update_vector_store_file(str(sys.argv[1]),str(sys.argv[2]),str(sys.argv[3]))
|
Loading…
Reference in a new issue