ciyidogan commited on
Commit
6b8ff51
·
verified ·
1 Parent(s): 2c7b721

Update llm_factory.py

Browse files
Files changed (1) hide show
  1. llm_factory.py +3 -3
llm_factory.py CHANGED
@@ -37,7 +37,7 @@ class LLMFactory:
37
  return LLMFactory._create_spark_provider(llm_config, api_key, provider_def)
38
  elif provider_name == "spark_cloud":
39
  return LLMFactory._create_spark_provider(llm_config, api_key, provider_def)
40
- elif provider_name in ["gpt4o", "gpt4o-mini"]:
41
  return LLMFactory._create_gpt_provider(llm_config, api_key, provider_def)
42
  else:
43
  raise ValueError(f"Unsupported LLM provider: {provider_name}")
@@ -83,8 +83,8 @@ class LLMFactory:
83
  # Then try environment variables
84
  env_mappings = {
85
  "spark": "SPARK_TOKEN",
86
- "gpt4o": "OPENAI_API_KEY",
87
- "gpt4o-mini": "OPENAI_API_KEY"
88
  }
89
 
90
  env_var = env_mappings.get(provider_name)
 
37
  return LLMFactory._create_spark_provider(llm_config, api_key, provider_def)
38
  elif provider_name == "spark_cloud":
39
  return LLMFactory._create_spark_provider(llm_config, api_key, provider_def)
40
+ elif provider_name in ["gpt-4o", "gpt-4o-mini"]:
41
  return LLMFactory._create_gpt_provider(llm_config, api_key, provider_def)
42
  else:
43
  raise ValueError(f"Unsupported LLM provider: {provider_name}")
 
83
  # Then try environment variables
84
  env_mappings = {
85
  "spark": "SPARK_TOKEN",
86
+ "gpt-4o": "OPENAI_API_KEY",
87
+ "gpt-4o-mini": "OPENAI_API_KEY"
88
  }
89
 
90
  env_var = env_mappings.get(provider_name)