Files
OpenHands/tests/unit
Kaushik Deka 5bb931e4d6 Add prompt caching (Sonnet, Haiku only) (#3411)
* Add prompt caching

* remove anthropic-version from extra_headers

* change supports_prompt_caching method to attribute

* change caching strat and log cache statistics

* add reminder as a new message to fix caching

* fix unit test

* append reminder to the end of the last message content

* move token logs to post completion function

* fix unit test failure

* fix reminder and prompt caching

* unit tests for prompt caching

* add test

* clean up tests

* separate reminder, use latest two messages

* fix tests

---------

Co-authored-by: tobitege <10787084+tobitege@users.noreply.github.com>
Co-authored-by: Xingyao Wang <xingyao6@illinois.edu>
Co-authored-by: Engel Nyst <enyst@users.noreply.github.com>
2024-08-26 20:46:44 -04:00
..
2024-08-26 01:31:37 -06:00

Introduction

This folder contains unit tests that could be run locally.

Run all test:

poetry run pytest ./tests/unit

Run specific test file:

poetry run pytest ./tests/unit/test_micro_agents.py

Run specific unit test

poetry run pytest ./tests/unit/test_micro_agents.py::test_coder_agent_with_summary

For a more verbose output, to above calls the -v flag can be used (even more verbose: -vv and -vvv):

poetry run pytest -v ./tests/unit/test_micro_agents.py

More details see pytest doc