@@ -437,7 +437,7 @@ def _run_crewai(self, config, topic, tools_dict):
437437 llm_model = details .get ('llm' )
438438 if llm_model :
439439 llm = PraisonAIModel (
440- model = llm_model .get ("model" , os .environ .get ("MODEL_NAME" , "openai/gpt-4o" )) ,
440+ model = llm_model .get ("model" ) or os .environ .get ("MODEL_NAME" ) or "openai/gpt-4o" ,
441441 ).get_model ()
442442 else :
443443 llm = PraisonAIModel ().get_model ()
@@ -446,7 +446,7 @@ def _run_crewai(self, config, topic, tools_dict):
446446 function_calling_llm_model = details .get ('function_calling_llm' )
447447 if function_calling_llm_model :
448448 function_calling_llm = PraisonAIModel (
449- model = function_calling_llm_model .get ("model" , os .environ .get ("MODEL_NAME" , "openai/gpt-4o" )) ,
449+ model = function_calling_llm_model .get ("model" ) or os .environ .get ("MODEL_NAME" ) or "openai/gpt-4o" ,
450450 ).get_model ()
451451 else :
452452 function_calling_llm = PraisonAIModel ().get_model ()
@@ -557,8 +557,8 @@ def _run_praisonai(self, config, topic, tools_dict):
557557 backstory = backstory_filled ,
558558 tools = tools_list , # Pass the entire tools list to the agent
559559 allow_delegation = details .get ('allow_delegation' , False ),
560- llm = details .get ('llm' , {}).get ("model" , os .environ .get ("MODEL_NAME" , " gpt-4o")) ,
561- function_calling_llm = details .get ('function_calling_llm' , {}).get ("model" , os .environ .get ("MODEL_NAME" , " gpt-4o")) ,
560+ llm = details .get ('llm' , {}).get ("model" ) or os .environ .get ("MODEL_NAME" ) or "openai/ gpt-4o" ,
561+ function_calling_llm = details .get ('function_calling_llm' , {}).get ("model" ) or os .environ .get ("MODEL_NAME" ) or "openai/ gpt-4o" ,
562562 max_iter = details .get ('max_iter' , 15 ),
563563 max_rpm = details .get ('max_rpm' ),
564564 max_execution_time = details .get ('max_execution_time' ),
@@ -567,7 +567,7 @@ def _run_praisonai(self, config, topic, tools_dict):
567567 system_template = details .get ('system_template' ),
568568 prompt_template = details .get ('prompt_template' ),
569569 response_template = details .get ('response_template' ),
570- reflect_llm = details .get ('reflect_llm' , {}).get ("model" , os .environ .get ("MODEL_NAME" , " gpt-4o")) ,
570+ reflect_llm = details .get ('reflect_llm' , {}).get ("model" ) or os .environ .get ("MODEL_NAME" ) or "openai/ gpt-4o" ,
571571 min_reflect = details .get ('min_reflect' , 1 ),
572572 max_reflect = details .get ('max_reflect' , 3 ),
573573 )
@@ -623,7 +623,7 @@ def _run_praisonai(self, config, topic, tools_dict):
623623 tasks = tasks ,
624624 verbose = True ,
625625 process = "hierarchical" ,
626- manager_llm = config .get ('manager_llm' , ' gpt-4o' ) ,
626+ manager_llm = config .get ('manager_llm' ) or os . environ . get ( "MODEL_NAME" ) or "openai/ gpt-4o" ,
627627 memory = memory
628628 )
629629 else :
0 commit comments