Skip to content

Commit 85cc566

Browse files
quick fix for agent default behaviour (#136)
* quick fix for agent default behaviour * adjust docstring and RESEARCH.md * Fix ruff formatting in ops.py Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> --------- Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 21b8452 commit 85cc566

File tree

2 files changed

+13
-10
lines changed

2 files changed

+13
-10
lines changed

docs/reference/RESEARCH.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ Each row gets its own agent that researches independently.
6969
| `task` | str | The agent task describing what to research |
7070
| `session` | Session | Optional, auto-created if omitted |
7171
| `input` | BaseModel \| DataFrame \| UUID | Optional input context |
72-
| `effort_level` | EffortLevel | LOW, MEDIUM, or HIGH (default: LOW) |
72+
| `effort_level` | EffortLevel | LOW, MEDIUM, or HIGH (default: MEDIUM) |
7373
| `llm` | LLM | Optional agent LLM override |
7474
| `response_model` | BaseModel | Optional schema for structured output |
7575
| `return_table` | bool | (`single_agent` only) If True, returns a table instead of a scalar result |
@@ -78,8 +78,8 @@ Each row gets its own agent that researches independently.
7878

7979
The effort level lets you control how thorough the research is.
8080

81-
- `LOW`: Quick lookups, basic web searches, fast and cheap (default)
82-
- `MEDIUM`: More thorough research, multiple sources consulted
81+
- `LOW`: Just a single LLM call, not a real agent, cheapest & fastest
82+
- `MEDIUM`: More thorough research, multiple sources consulted (default)
8383
- `HIGH`: Deep research, cross-referencing sources, higher accuracy
8484

8585
### Response model

src/everyrow/ops.py

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

5353

54+
DEFAULT_EFFORT_LEVEL = EffortLevel.MEDIUM
55+
56+
5457
class DefaultAgentResponse(BaseModel):
5558
answer: str
5659

@@ -141,7 +144,7 @@ async def single_agent[T: BaseModel](
141144
task: str,
142145
session: Session | None = None,
143146
input: BaseModel | UUID | Result | None = None,
144-
effort_level: EffortLevel | None = EffortLevel.LOW,
147+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
145148
llm: LLM | None = None,
146149
iteration_budget: int | None = None,
147150
include_research: bool | None = None,
@@ -155,7 +158,7 @@ async def single_agent(
155158
task: str,
156159
session: Session | None = None,
157160
input: BaseModel | UUID | Result | None = None,
158-
effort_level: EffortLevel | None = EffortLevel.LOW,
161+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
159162
llm: LLM | None = None,
160163
iteration_budget: int | None = None,
161164
include_research: bool | None = None,
@@ -168,7 +171,7 @@ async def single_agent[T: BaseModel](
168171
task: str,
169172
session: Session | None = None,
170173
input: BaseModel | DataFrame | UUID | Result | None = None,
171-
effort_level: EffortLevel | None = EffortLevel.LOW,
174+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
172175
llm: LLM | None = None,
173176
iteration_budget: int | None = None,
174177
include_research: bool | None = None,
@@ -182,7 +185,7 @@ async def single_agent[T: BaseModel](
182185
session: Optional session. If not provided, one will be created automatically.
183186
input: Input data (BaseModel, DataFrame, UUID, or Result).
184187
effort_level: Effort level preset (low/medium/high). Mutually exclusive with
185-
custom params (llm, iteration_budget, include_research). Default: low.
188+
custom params (llm, iteration_budget, include_research). Default: medium.
186189
llm: LLM to use. Required when effort_level is None.
187190
iteration_budget: Number of agent iterations (0-20). Required when effort_level is None.
188191
include_research: Include research notes. Required when effort_level is None.
@@ -224,7 +227,7 @@ async def single_agent_async[T: BaseModel](
224227
task: str,
225228
session: Session,
226229
input: BaseModel | DataFrame | UUID | Result | None = None,
227-
effort_level: EffortLevel | None = EffortLevel.LOW,
230+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
228231
llm: LLM | None = None,
229232
iteration_budget: int | None = None,
230233
include_research: bool | None = None,
@@ -272,7 +275,7 @@ async def agent_map(
272275
task: str,
273276
session: Session | None = None,
274277
input: DataFrame | UUID | TableResult | None = None,
275-
effort_level: EffortLevel | None = EffortLevel.LOW,
278+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
276279
llm: LLM | None = None,
277280
iteration_budget: int | None = None,
278281
include_research: bool | None = None,
@@ -335,7 +338,7 @@ async def agent_map_async(
335338
task: str,
336339
session: Session,
337340
input: DataFrame | UUID | TableResult,
338-
effort_level: EffortLevel | None = EffortLevel.LOW,
341+
effort_level: EffortLevel | None = DEFAULT_EFFORT_LEVEL,
339342
llm: LLM | None = None,
340343
iteration_budget: int | None = None,
341344
include_research: bool | None = None,

0 commit comments

Comments
 (0)