fix: Fixing codebase agent

[vishalshenoy]

Flushing out some details in agent implementation

[codecov]

❌ 1 Tests Failed:

Tests completed	Failed	Passed	Skipped
29	1	28	0

View the top 1 failed test(s) by shortest run time

::tests.unit.codegen.sdk.core.test_directory

Stack Traces | 0s run time

ImportError while importing test module '.../sdk/core/test_directory.py'.
Hint: make sure your test modules/packages have valid Python names.
Traceback:
.../sdk/core/test_directory.py:8: in <module>
    from codegen.sdk.codebase.codebase_graph import CodebaseGraph
E   ModuleNotFoundError: No module named 'codegen.sdk.codebase.codebase_graph'

To view more test analytics, go to the Test Analytics Dashboard
_{📋 Got 3 mins? Take this short survey to help us improve Test Analytics.}