Skip to content

Commit

Permalink
minor additions to text
Browse files Browse the repository at this point in the history
  • Loading branch information
Your Name authored and Your Name committed Sep 11, 2024
1 parent 2e46e5b commit 213da21
Showing 1 changed file with 8 additions and 9 deletions.
17 changes: 8 additions & 9 deletions src/config_wizard.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,10 +37,10 @@ def run_config_wizard(args, config, config_file_path):

# Ask the user to select the AI provider
provider_choice = input("Choose the AI provider to use:\n"
"1. OpenAI\n"
"2. Azure OpenAI\n"
"3. Google Vertex AI\n"
"4. Offline (use pre-existing templates)\n"
"1. OpenAI (Working)\n"
"2. Azure OpenAI (Work In Progress)\n"
"3. Google Vertex AI (Work In Progress)\n"
"4. Offline (Use pre-existing templates)\n"
"Enter the number of your choice: ")

ai_service = None # Initialize the ai_service variable
Expand All @@ -64,7 +64,7 @@ def run_config_wizard(args, config, config_file_path):
# Initialize AIService for querying later (generic service class)
ai_service = AIService(api_key=openai_key, debug_mode=args.debug) # Removed 'provider'

if provider_choice == '2':
elif provider_choice == '2':
provider = 'azure'
azure_key = input("Enter your Azure OpenAI API key: ")
azure_endpoint = input("Enter your Azure OpenAI endpoint: ")
Expand Down Expand Up @@ -102,8 +102,8 @@ def run_config_wizard(args, config, config_file_path):
config.set('gcp', 'location', gcp_location)
config.set('gcp', 'model_id', gcp_model_id)

# Initialize AIService for querying later
exit ('Oh no!\nSnap!\nGoogle is charging us to develop integration with them!\nOther provider is recommended.\nIf you dont have other procider access try the offline mode!\nxoxo :-*')
# Exit with a message for unsupported integration
exit('Oh no!\nSnap!\nGoogle is charging us to develop integration with them!\nOther provider is recommended.\nIf you don’t have other provider access, try the offline mode!\nXOXO :-*')

elif provider_choice == '4':
provider = 'offline'
Expand Down Expand Up @@ -204,5 +204,4 @@ def run_config_wizard(args, config, config_file_path):

# Query the AI service for responses if not in offline mode
if provider != 'offline' and ai_service:
query_ai_service_for_responses(technology, segment, domain, anonymous_access, args.debug, ai_service)

query_ai_service_for_responses(technology, segment, domain, anonymous_access, args.debug, ai_service)

0 comments on commit 213da21

Please sign in to comment.