|
21 | 21 | logger.disabled = True |
22 | 22 | est_timezone = pytz.timezone('US/Eastern') |
23 | 23 |
|
| 24 | +# Global default logging level for all IkaBaseAgent instances. |
| 25 | +# Override by passing logging_level into Agent.__init__. |
| 26 | +DEFAULT_AGENT_LOGGING_LEVEL = 1 |
| 27 | + |
24 | 28 | def enable_base_agent_logging(): |
25 | 29 | try: |
26 | 30 | if os.getenv("FOG_DEBUG") == "1": |
@@ -48,10 +52,18 @@ def enable_base_agent_logging(): |
48 | 52 |
|
49 | 53 | class Agent(ABC): |
50 | 54 |
|
51 | | - def __init__(self, model=None, api_key: str = None, anthropic_api_key: str = None, model_id: str = None): |
| 55 | + def __init__( |
| 56 | + self, |
| 57 | + model=None, |
| 58 | + api_key: str = None, |
| 59 | + anthropic_api_key: str = None, |
| 60 | + model_id: str = None, |
| 61 | + logging_level=None, |
| 62 | + ): |
52 | 63 | self.model_id = model_id or (getattr(model, "model_id", None) if model else None) or "deepseek" |
53 | 64 | self.api_key = api_key or (getattr(model, "api_key", None) if model else None) |
54 | 65 | self.anthropic_api_key = anthropic_api_key |
| 66 | + self.logging_level = DEFAULT_AGENT_LOGGING_LEVEL if logging_level is None else logging_level |
55 | 67 | self.agents = {} |
56 | 68 | self.setup_agents() |
57 | 69 |
|
@@ -99,7 +111,7 @@ def run_task(self, task_description: str, context: dict = None) -> dict: |
99 | 111 | return results |
100 | 112 |
|
101 | 113 | prompt = self._create_task_prompt(task_description, context) |
102 | | - base = getattr(manager_agent, "_base_prompt", None) |
| 114 | + base = getattr(manager_agent, "_base_prompt", None) or getattr(manager_agent, "prompt", None) |
103 | 115 | full = (base + "\n\n" + prompt) if base else prompt |
104 | 116 |
|
105 | 117 | logger.info(f"Using {manager_agent.name} for task: {task_description}") |
|
0 commit comments