wehub-resource-sync
41 lines
1.9 KiB
Python
41 lines
1.9 KiB
Python
|
|
"""This example script shows how to set the os.environ variables with API keys for supported models"""
|
|
|
|
import os
|
|
|
|
|
|
def user_managed_secrets_setup(key, value):
|
|
|
|
user_managed_access_key_list = ["USER_MANAGED_OPENAI_API_KEY",
|
|
"USER_MANAGED_COHERE_API_KEY",
|
|
"USER_MANAGED_ANTHROPIC_API_KEY",
|
|
"USER_MANAGED_AI21_API_KEY",
|
|
"USER_MANAGED_GOOGLE_API_KEY",
|
|
"USER_MANAGED_PINECONE_API_KEY",
|
|
"USER_MANAGED_PINECONE_ENVIRONMENT",
|
|
"USER_MANAGED_AWS_ACCESS_KEY",
|
|
"USER_MANAGED_AWS_SECRET_KEY"]
|
|
|
|
if key in user_managed_access_key_list:
|
|
os.environ[key] = value
|
|
|
|
# set environ variables- and will be automatically passed to corresponding model when invoked
|
|
|
|
# os.environ["USER_MANAGED_OPENAI_API_KEY"] = "{INSERT-YOUR-OPENAI-KEY}"
|
|
# os.environ["USER_MANAGED_COHERE_API_KEY"] = "{INSERT_YOUR-COHERE-API-KEY}"
|
|
# os.environ["USER_MANAGED_ANTHROPIC_API_KEY"] = "{INSERT_YOUR_ANTHROPIC_API_KEY}"
|
|
# os.environ["USER_MANAGED_AI21_API_KEY"] = "{INSERT_YOUR_AI21_API_KEY}"
|
|
# os.environ["USER_MANAGED_GOOGLE_API_KEY"] = "{INSERT_YOUR_GOOGLE_API_KEY}"
|
|
# os.environ["USER_MANAGED_PINECONE_API_KEY"] = "{INSERT_YOUR_PINECONE_API_KEY}"
|
|
# os.environ["USER_MANAGED_PINECONE_ENVIRONMENT"] = "{INSERT_YOUR_PINECONE_ENVIRONMENT_KEY}"
|
|
# os.environ["USER_MANAGED_AWS_ACCESS_KEY"] = "{INSERT_YOUR_AWS_ACCESS_KEY}"
|
|
# os.environ["USER_MANAGED_AWS_SECRET_KEY"] = "{INSERT_YOUR_AWS_SECRET_KEY}"
|
|
|
|
return 0
|
|
|
|
# Example Usage
|
|
# os.environ["USER_MANAGED_OPENAI_API_KEY"] = "..."
|
|
# prompter = Prompt().load_model("gpt-4")
|
|
# --> prompter will look to the environ variable and pass the API key
|
|
|