config-sample.py

@@ -1,4 +1,4 @@
 base_url = "https://api.openai.com/v1/"
 secret_key = "YOUR-OPENAI-SECRET-KEY-HERE"
-vector_store = ["YOUR-VECTOR-STORE-IDs-HERE","SEPARATED-BY-COMMAS"]
+vector_store = "YOUR-VECTOR-STORE-ID-HERE"
 

create_model_response.py

@@ -10,7 +10,7 @@ def create_model_response(conversation,model,message):
       "model": model,
       "tools": [
       {"type": "web_search"},
-      {"type": "file_search", "vector_store_ids" :vector_store}
+      {"type": "file_search", "vector_store_ids" :[vector_store]}
       ],
       "input": message,
       "conversation": conversation
@@ -23,7 +23,5 @@ def create_model_response(conversation,model,message):
     response = requests.request("POST", url, headers=headers, data=payload)
 
     print(response.text)
-    print(response.json()["output"][-1]["content"][0]["text"])
-
 if __name__ == '__main__':
     create_model_response(str(sys.argv[1]),str(sys.argv[2]),str(sys.argv[3]))

retrieve_model_response.py

@@ -13,7 +13,6 @@ def retrieve_model_response(response):
     response = requests.request("GET", url, headers=headers, data=payload)
 
     print(response.text)
-    print(response.json()["output"][-1]["content"][0]["text"])
 
 if __name__ == '__main__':
     retrieve_model_response(str(sys.argv[1]))