Commit graph

14 commits

Author SHA1 Message Date
Ishaan Jaff
fd6366d041 fix test google secret manager 2024-09-03 14:01:03 -07:00
Ishaan Jaff
04456ab19e test secret manager 2024-09-03 11:22:06 -07:00
Ishaan Jaff
3c898e23ea refactor secret managers 2024-09-03 10:58:02 -07:00
Ishaan Jaff
b0178a85cf refactor get_secret 2024-09-03 10:42:12 -07:00
Krish Dholakia
37f9705d6e
Bedrock Embeddings refactor + model support (#5462)
* refactor(bedrock): initial commit to refactor bedrock to a folder

Improve code readability + maintainability

* refactor: more refactor work

* fix: fix imports

* feat(bedrock/embeddings.py): support translating embedding into amazon embedding formats

* fix: fix linting errors

* test: skip test on end of life model

* fix(cohere/embed.py): fix linting error

* fix(cohere/embed.py): fix typing

* fix(cohere/embed.py): fix post-call logging for cohere embedding call

* test(test_embeddings.py): fix error message assertion in test
2024-09-01 13:29:58 -07:00
David Manouchehri
11668c31c1
(oidc): Add support for loading tokens via a file, environment variable, and from a file path set in an env var. 2024-08-16 20:13:07 +00:00
David Manouchehri
ffbe1845ef
(bedrock_httpx.py) - Add support for custom STS endpoints, e.g. for FIPS. 2024-07-31 16:51:52 +00:00
David Manouchehri
c364a3129e
(test_secret_manager.py) - Improve and add CircleCI v1 test with Amazon. 2024-07-24 16:05:58 +00:00
David Manouchehri
af3a900b4d
(tests) - Try azure AD auth directly. 2024-07-24 15:42:40 +00:00
David Manouchehri
eb57b2c6d7
Fix OIDC tests. This should always work, even without IAM set up . 2024-05-31 13:48:52 +00:00
Ishaan Jaff
2e9f4ff23a fix oidc tests 2024-05-11 16:31:38 -07:00
David Manouchehri
4b655d8b33
feat(util.py): Add OIDC support. 2024-05-07 15:46:48 +00:00
Krrish Dholakia
c3e3ca9722 test(test_secret_manager.py): skip test - aws account suspended 2024-04-04 10:10:21 -07:00
Krrish Dholakia
bc66ef9d5c fix(utils.py): fix aws secret manager + support key_management_settings
fixes the aws secret manager implementation and allows the user to set which keys they want to check thr
ough it
2024-03-16 16:47:50 -07:00