[BUGFIX] Enhance key access in Context class to support direct and ne…

…sted dotted notation

src/koheesio/context.py

@@ -337,18 +337,20 @@ def get(self, key: str, default: Any = None, safe: bool = True) -> Any:
         Returns `c`
         """
         try:
-            # try full dotted natation first
+            # in case key is directly available, or is written in dotted notation
             try:
                 return self.__dict__[key]
             except KeyError:
                 pass
-
-            # handle nested keys
-            nested_keys = key.split(".")
-            value = self  # parent object
-            for k in nested_keys:
-                value = value[k]  # iterate through nested values
-            return value
+            if "." in key:
+                # handle nested keys
+                nested_keys = key.split(".")
+                value = self  # parent object
+                for k in nested_keys:
+                    value = value[k]  # iterate through nested values
+                return value
+
+            raise KeyError
 
         except (AttributeError, KeyError, TypeError) as e:
             if not safe:

tests/core/test_context.py

@@ -234,6 +234,11 @@ def test_from_dict():
             Context(foo="bar", baz="qux"),
             {"foo": "bar", "baz": "qux"},
         ),
+        (
+            # Test with dotted keys
+            Context({"a.b.c": 1}),
+            {"a.b.c": 1},
+        )
     ],
 )
 def test_to_dict(context, expected):