INNER CODE UNIT · Python
setup
instavm/clickclickclick · main.py:85
def setup():
"""Setup command to configure planner and finder"""
planner = click.prompt("Choose planner model ('gemini', '4o', or 'ollama')", type=str)
if planner.lower() == "gemini":
gemini_api_key = click.prompt("Enter your Gemini API key", hide_input=True)
os.environ["GEMINI_API_KEY"] = gemini_api_key
elif planner.lower() == "4o":
version = click.prompt("Is it OpenAI or Azure version?", type=str)
if version.lower() == "openai":
openai_api_key = click.prompt("Enter your OpenAI API key", hide_input=True)
os.environ["AZURE_OPENAI_API_KEY"] = openai_api_key
os.environ["OPENAI_API_TYPE"] = "openai"
elif version.lower() == "azure":
azure_api_key = click.prompt("Enter your Azure API key", hide_input=True)
os.environ["AZURE_OPENAI_API_KEY"] = azure_api_key
azure_model_name = click.prompt("Enter your Azure model name", type=str)
os.environ["AZURE_OPENAI_MODEL_NAME"] = azure_model_name