ArkTS-CodeSearch: A Open-Source ArkTS Dataset for Code Retrieval

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: He, Yulong, Ermakov, Artem, Kovalchuk, Sergey, Aliev, Artem, Shalymov, Dmitry
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915783868678144
author He, Yulong
Ermakov, Artem
Kovalchuk, Sergey
Aliev, Artem
Shalymov, Dmitry
author_facet He, Yulong
Ermakov, Artem
Kovalchuk, Sergey
Aliev, Artem
Shalymov, Dmitry
contents ArkTS is a core programming language in the OpenHarmony ecosystem, yet research on ArkTS code intelligence is hindered by the lack of public datasets and evaluation benchmarks. This paper presents a large-scale ArkTS dataset constructed from open-source repositories, targeting code retrieval and code evaluation tasks. We design a single-search task, where natural language comments are used to retrieve corresponding ArkTS functions. ArkTS repositories are crawled from GitHub and Gitee, and comment-function pairs are extracted using tree-sitter-arkts, followed by cross-platform deduplication and statistical analysis of ArkTS function types. We further evaluate existing open-source code embedding models on the single-search task and perform fine-tuning using both ArkTS and TypeScript training datasets, resulting in a high-performing model for ArkTS code understanding. This work establishes the first systematic benchmark for ArkTS code retrieval. Both the dataset and our fine-tuned model are available at https://huggingface.co/hreyulog/embedinggemma_arkts and https://huggingface.co/datasets/hreyulog/arkts-code-docstring .
format Preprint
id arxiv_https___arxiv_org_abs_2602_05550
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle ArkTS-CodeSearch: A Open-Source ArkTS Dataset for Code Retrieval
He, Yulong
Ermakov, Artem
Kovalchuk, Sergey
Aliev, Artem
Shalymov, Dmitry
Software Engineering
Computation and Language
ArkTS is a core programming language in the OpenHarmony ecosystem, yet research on ArkTS code intelligence is hindered by the lack of public datasets and evaluation benchmarks. This paper presents a large-scale ArkTS dataset constructed from open-source repositories, targeting code retrieval and code evaluation tasks. We design a single-search task, where natural language comments are used to retrieve corresponding ArkTS functions. ArkTS repositories are crawled from GitHub and Gitee, and comment-function pairs are extracted using tree-sitter-arkts, followed by cross-platform deduplication and statistical analysis of ArkTS function types. We further evaluate existing open-source code embedding models on the single-search task and perform fine-tuning using both ArkTS and TypeScript training datasets, resulting in a high-performing model for ArkTS code understanding. This work establishes the first systematic benchmark for ArkTS code retrieval. Both the dataset and our fine-tuned model are available at https://huggingface.co/hreyulog/embedinggemma_arkts and https://huggingface.co/datasets/hreyulog/arkts-code-docstring .
title ArkTS-CodeSearch: A Open-Source ArkTS Dataset for Code Retrieval
topic Software Engineering
Computation and Language
url https://arxiv.org/abs/2602.05550