Skip to content

Commit 606a926

Browse files
quick fix for agent default behaviour
1 parent fddae60 commit 606a926

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

src/everyrow/ops.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@
5151
InputData = UUID | list[dict[str, Any]] | dict[str, Any]
5252

5353

54+
DEFAULT_EFFORT_LEVEL = EffortLevel.MEDIUM
5455
class DefaultAgentResponse(BaseModel):
5556
answer: str
5657

@@ -137,7 +138,7 @@ async def single_agent[T: BaseModel](
137138
task: str,
138139
session: Session | None = None,
139140
input: BaseModel | UUID | Result | None = None,
140-
effort_level: EffortLevel | None = EffortLevel.LOW,
141+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
141142
llm: LLM | None = None,
142143
iteration_budget: int | None = None,
143144
include_research: bool | None = None,
@@ -151,7 +152,7 @@ async def single_agent(
151152
task: str,
152153
session: Session | None = None,
153154
input: BaseModel | UUID | Result | None = None,
154-
effort_level: EffortLevel | None = EffortLevel.LOW,
155+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
155156
llm: LLM | None = None,
156157
iteration_budget: int | None = None,
157158
include_research: bool | None = None,
@@ -164,7 +165,7 @@ async def single_agent[T: BaseModel](
164165
task: str,
165166
session: Session | None = None,
166167
input: BaseModel | DataFrame | UUID | Result | None = None,
167-
effort_level: EffortLevel | None = EffortLevel.LOW,
168+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
168169
llm: LLM | None = None,
169170
iteration_budget: int | None = None,
170171
include_research: bool | None = None,
@@ -220,7 +221,7 @@ async def single_agent_async[T: BaseModel](
220221
task: str,
221222
session: Session,
222223
input: BaseModel | DataFrame | UUID | Result | None = None,
223-
effort_level: EffortLevel | None = EffortLevel.LOW,
224+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
224225
llm: LLM | None = None,
225226
iteration_budget: int | None = None,
226227
include_research: bool | None = None,
@@ -258,7 +259,7 @@ async def agent_map(
258259
task: str,
259260
session: Session | None = None,
260261
input: DataFrame | UUID | TableResult | None = None,
261-
effort_level: EffortLevel | None = EffortLevel.LOW,
262+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
262263
llm: LLM | None = None,
263264
iteration_budget: int | None = None,
264265
include_research: bool | None = None,
@@ -321,7 +322,7 @@ async def agent_map_async(
321322
task: str,
322323
session: Session,
323324
input: DataFrame | UUID | TableResult,
324-
effort_level: EffortLevel | None = EffortLevel.LOW,
325+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
325326
llm: LLM | None = None,
326327
iteration_budget: int | None = None,
327328
include_research: bool | None = None,

0 commit comments

Comments
 (0)