Skip to content

Commit c322214

Browse files
authored
Fix/improve typing for get_codebase_session using default programming_language (#275)
# Motivation <!-- Why is this change necessary? --> # Content <!-- Please include a summary of the change --> # Testing <!-- How was the change tested? --> # Please check the following before marking your PR as ready for review - [ ] I have added tests for my changes - [ ] I have updated the documentation or added new documentation as needed
1 parent 7aa45c8 commit c322214

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

src/codegen/sdk/codebase/factory/get_session.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,22 @@
1515
from codegen.sdk.tree_sitter_parser import print_errors
1616

1717

18+
@overload
19+
def get_codebase_session(
20+
tmpdir: str | os.PathLike[str],
21+
programming_language: None = None,
22+
files: dict[str, str] = {},
23+
commit: bool = True,
24+
sync_graph: bool = True,
25+
verify_input: bool = True,
26+
verify_output: bool = True,
27+
repo_config: BaseRepoConfig | None = None,
28+
feature_flags: GSFeatureFlags = TestFlags,
29+
session_options: SessionOptions = SessionOptions(),
30+
secrets: Secrets = Secrets(),
31+
) -> AbstractContextManager[PyCodebaseType]: ...
32+
33+
1834
@overload
1935
def get_codebase_session(
2036
tmpdir: str | os.PathLike[str],

0 commit comments

Comments
 (0)