andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

clean up prints

Browse Source
This commit is contained in:
Brandon Hancock
2024-10-30 16:11:50 -04:00
parent 6b3c5d28e2
commit 7c7c3f88ad
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/crewai/agent.py
View File

@@ -123,9 +123,6 @@ class Agent(BaseAgent):
def post_init_setup(self): def post_init_setup(self):
self.agent_ops_agent_name = self.role self.agent_ops_agent_name = self.role
print("IN POST INIT SETUP")
print("self.llm:", self.llm)
# Handle different cases for self.llm # Handle different cases for self.llm
if isinstance(self.llm, str): if isinstance(self.llm, str):
# If it's a string, create an LLM instance # If it's a string, create an LLM instance
@@ -134,7 +131,6 @@ class Agent(BaseAgent):
# If it's already an LLM instance, keep it as is # If it's already an LLM instance, keep it as is
pass pass
elif self.llm is None: elif self.llm is None:
print("No LLM provided")
# Determine the model name from environment variables or use default # Determine the model name from environment variables or use default
model_name = ( model_name = (
os.environ.get("OPENAI_MODEL_NAME") os.environ.get("OPENAI_MODEL_NAME")
@@ -162,6 +158,18 @@ class Agent(BaseAgent):
if "API_KEY" in env_var["key_name"] if "API_KEY" in env_var["key_name"]
else env_var["key_name"] else env_var["key_name"]
) )
# Map key names containing "API_BASE" to "api_base"
key_name = (
"api_base"
if "API_BASE" in env_var["key_name"]
else key_name
)
# Map key names containing "API_VERSION" to "api_version"
key_name = (
"api_version"
if "API_VERSION" in env_var["key_name"]
else key_name
)
llm_params[key_name] = env_value llm_params[key_name] = env_value
# Check for default values if the environment variable is not set # Check for default values if the environment variable is not set
elif env_var.get("default", False): elif env_var.get("default", False):
@@ -171,10 +179,8 @@ class Agent(BaseAgent):
if key in os.environ: if key in os.environ:
llm_params[key] = value llm_params[key] = value
print("LLM PARAMS:", llm_params)
self.llm = LLM(**llm_params) self.llm = LLM(**llm_params)
else: else:
print("IN ELSE")
# For any other type, attempt to extract relevant attributes # For any other type, attempt to extract relevant attributes
llm_params = { llm_params = {
"model": getattr(self.llm, "model_name", None) "model": getattr(self.llm, "model_name", None)