ciyidogan commited on
Commit
841f42e
·
verified ·
1 Parent(s): 26855c2

Update llm_factory.py

Browse files
Files changed (1) hide show
  1. llm_factory.py +2 -0
llm_factory.py CHANGED
@@ -35,6 +35,8 @@ class LLMFactory:
35
  # Create provider based on name
36
  if provider_name == "spark":
37
  return LLMFactory._create_spark_provider(llm_config, api_key, provider_def)
 
 
38
  elif provider_name in ["gpt4o", "gpt4o-mini"]:
39
  return LLMFactory._create_gpt_provider(llm_config, api_key, provider_def)
40
  else:
 
35
  # Create provider based on name
36
  if provider_name == "spark":
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: