adjusted existing unit tests to work with new contextualization logic

Author: ds-jakub-cierocki

tests/unit/test_iql_format.py

@@ -23,6 +23,10 @@ async def test_iql_prompt_format_default() -> None:
             "You MUST use only these methods:\n"
             "\n\n"
             "It is VERY IMPORTANT not to use methods other than those listed above."
+            "Finally, if a called function argument value is not directly specified in the query but instead requires knowledge of some additional context, than substitute that argument value by: BaseCallerContext()."
+            'The typical input phrase referencing some additional context contains the word "my" or similar phrasing, e.g. "my position name", "my company valuation".'
+            "In that case, the part of the output will look like this:"
+            "filter4(BaseCallerContext())"
             """If you DON'T KNOW HOW TO ANSWER DON'T SAY \"\", SAY: `UNSUPPORTED QUERY` INSTEAD! """
             "This is CRUCIAL, otherwise the system will crash. ",
             "is_example": False,
@@ -53,6 +57,10 @@ async def test_iql_prompt_format_few_shots_injected() -> None:
             "You MUST use only these methods:\n"
             "\n\n"
             "It is VERY IMPORTANT not to use methods other than those listed above."
+            "Finally, if a called function argument value is not directly specified in the query but instead requires knowledge of some additional context, than substitute that argument value by: BaseCallerContext()."
+            'The typical input phrase referencing some additional context contains the word "my" or similar phrasing, e.g. "my position name", "my company valuation".'
+            "In that case, the part of the output will look like this:"
+            "filter4(BaseCallerContext())"
             """If you DON'T KNOW HOW TO ANSWER DON'T SAY \"\", SAY: `UNSUPPORTED QUERY` INSTEAD! """
             "This is CRUCIAL, otherwise the system will crash. ",
             "is_example": False,

tests/unit/test_iql_generator.py

@@ -80,6 +80,7 @@ async def test_iql_generation(iql_generator: IQLGenerator, event_tracker: EventT
             source="filter_by_id(1)",
             allowed_functions=filters,
             event_tracker=event_tracker,
+            contexts=None
         )