├── .gitignore ├── LICENSE ├── README.md ├── requirements.txt └── typegen ├── ast_operation.py ├── config.py ├── cot.py ├── demo.py ├── evaluate.py ├── gen.py ├── incoder.py ├── mask.py ├── preprocess.py ├── slice.py ├── typegen.py └── unixcoder.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | hityper 2 | fastbm25 3 | jieba 4 | torch 5 | timeout_decorator 6 | openai 7 | six -------------------------------------------------------------------------------- /typegen/ast_operation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/ast_operation.py -------------------------------------------------------------------------------- /typegen/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/config.py -------------------------------------------------------------------------------- /typegen/cot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/cot.py -------------------------------------------------------------------------------- /typegen/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/demo.py -------------------------------------------------------------------------------- /typegen/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/evaluate.py -------------------------------------------------------------------------------- /typegen/gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/gen.py -------------------------------------------------------------------------------- /typegen/incoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/incoder.py -------------------------------------------------------------------------------- /typegen/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/mask.py -------------------------------------------------------------------------------- /typegen/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/preprocess.py -------------------------------------------------------------------------------- /typegen/slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/slice.py -------------------------------------------------------------------------------- /typegen/typegen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/typegen.py -------------------------------------------------------------------------------- /typegen/unixcoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JohnnyPeng18/TypeGen/HEAD/typegen/unixcoder.py --------------------------------------------------------------------------------