julep-ai / julep

A new DSL and server for AI agents and multi-step tasks
https://julep.ai
Apache License 2.0
2.67k stars 904 forks source link

fix(agents-api): Add `mmr_strength` in search tools & increase timeout for `transition_step` #808

Closed HamadaSalhab closed 3 weeks ago

HamadaSalhab commented 3 weeks ago

[!IMPORTANT] Add mmr_strength to search requests and increase transition_step timeout based on environment in agents-api.

  • Behavior:
    • Add mmr_strength parameter to _create_search_request() in execute_system.py for HybridDocSearchRequest, TextOnlyDocSearchRequest, and VectorDocSearchRequest.
    • Increase schedule_to_close_timeout for transition_step in transition.py to 600 seconds unless in debug or testing mode.
  • Misc:
    • Import debug and testing from env in transition.py.

This description was created by Ellipsis for bd64aa27254a7a19a7aa8320746599e67337ed85. It will automatically update as commits are pushed.

sweep-ai[bot] commented 3 weeks ago

Hey @HamadaSalhab, here is an example of how you can ask me to improve this pull request:

@sweep Add unit tests for `_create_search_request` function to verify the new `mmr_strength` parameter is correctly handled for all three request types (HybridDocSearchRequest, TextOnlyDocSearchRequest, and VectorDocSearchRequest), including:
- Default value (0) when not provided
- Custom value when provided
- Proper removal of the parameter from arguments dictionary

:book: For more information on how to use Sweep, please read our documentation.