llmware

1

2
"""This example script shows how to set the os.environ variables with API keys for supported models"""
3

4
import os
5

6

7
def user_managed_secrets_setup(key, value):
8

9
    user_managed_access_key_list = ["USER_MANAGED_OPENAI_API_KEY",
10
                                    "USER_MANAGED_COHERE_API_KEY",
11
                                    "USER_MANAGED_ANTHROPIC_API_KEY",
12
                                    "USER_MANAGED_AI21_API_KEY",
13
                                    "USER_MANAGED_GOOGLE_API_KEY",
14
                                    "USER_MANAGED_PINECONE_API_KEY",
15
                                    "USER_MANAGED_PINECONE_ENVIRONMENT",
16
                                    "USER_MANAGED_AWS_ACCESS_KEY",
17
                                    "USER_MANAGED_AWS_SECRET_KEY"]
18

19
    if key in user_managed_access_key_list:
20
        os.environ[key] = value
21

22
    # set environ variables- and will be automatically passed to corresponding model when invoked
23

24
    # os.environ["USER_MANAGED_OPENAI_API_KEY"] = "{INSERT-YOUR-OPENAI-KEY}"
25
    # os.environ["USER_MANAGED_COHERE_API_KEY"] = "{INSERT_YOUR-COHERE-API-KEY}"
26
    # os.environ["USER_MANAGED_ANTHROPIC_API_KEY"] = "{INSERT_YOUR_ANTHROPIC_API_KEY}"
27
    # os.environ["USER_MANAGED_AI21_API_KEY"] = "{INSERT_YOUR_AI21_API_KEY}"
28
    # os.environ["USER_MANAGED_GOOGLE_API_KEY"] = "{INSERT_YOUR_GOOGLE_API_KEY}"
29
    # os.environ["USER_MANAGED_PINECONE_API_KEY"] = "{INSERT_YOUR_PINECONE_API_KEY}"
30
    # os.environ["USER_MANAGED_PINECONE_ENVIRONMENT"] = "{INSERT_YOUR_PINECONE_ENVIRONMENT_KEY}"
31
    # os.environ["USER_MANAGED_AWS_ACCESS_KEY"] = "{INSERT_YOUR_AWS_ACCESS_KEY}"
32
    # os.environ["USER_MANAGED_AWS_SECRET_KEY"] = "{INSERT_YOUR_AWS_SECRET_KEY}"
33

34
    return 0
35

36
#   Example Usage
37
#       os.environ["USER_MANAGED_OPENAI_API_KEY"] = "..."
38
#       prompter = Prompt().load_model("gpt-4")
39
#       --> prompter will look to the environ variable and pass the API key
40

41