prasadnu commited on
Commit
dac1d4e
·
1 Parent(s): 51f22e0

RAG changes

Browse files
Files changed (1) hide show
  1. RAG/colpali.py +3 -3
RAG/colpali.py CHANGED
@@ -43,11 +43,11 @@ model_name = (
43
  # model_name
44
  # )
45
 
46
- credentials = boto3.Session().get_credentials()
47
- auth = AWSV4SignerAuth(credentials, 'us-west-2', 'es')
48
  aos_client = OpenSearch(
49
  hosts = [{'host': 'search-opensearchservi-shjckef2t7wo-iyv6rajdgxg6jas25aupuxev6i.us-west-2.es.amazonaws.com', 'port': 443}],
50
- http_auth = auth,
51
  use_ssl = True,
52
  verify_certs = True,
53
  connection_class = RequestsHttpConnection,
 
43
  # model_name
44
  # )
45
 
46
+ awsauth = HTTPBasicAuth('master',st.secrets['ml_search_demo_api_access'])
47
+ headers = {"Content-Type": "application/json"}
48
  aos_client = OpenSearch(
49
  hosts = [{'host': 'search-opensearchservi-shjckef2t7wo-iyv6rajdgxg6jas25aupuxev6i.us-west-2.es.amazonaws.com', 'port': 443}],
50
+ http_auth = awsauth,
51
  use_ssl = True,
52
  verify_certs = True,
53
  connection_class = RequestsHttpConnection,