Spaces:
Building
Building
Update llm_factory.py
Browse files- 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 ["
|
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 |
-
"
|
87 |
-
"
|
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)
|