Abstract:This paper proposes a learning‐based approach for API recommendation in smart contracts. We propose a code graph named pruned and augmented AST (pa‐AST) with the API sequence to capture the semantic features surrounding recommendation points. Meanwhile, we utilize a GAT‐based model for code feature learning and API recommendation. Application programming interface (API) recommendation plays a crucial role in facilitating smart contract development by providing developers with a ranked list of candidate APIs for specific recommendation points. Deep learning‐based approaches have shown promising results in this field. However, existing approaches mainly rely on token sequences or abstract syntax trees (ASTs) for learning recommendation point‐related features, which may overlook the essential knowledge implied in the relations between or within statements and may include task‐irrelevant components during feature learning. To address these limitations, we propose a novel code graph called pruned and augmented AST (pa‐AST). Our approach enhances the AST by incorporating additional knowledge derived from the control and data flow relations between and within statements in the smart contract code. Through this augmentation, the pa‐AST can better represent the semantic features of the code. Furthermore, we conduct AST pruning to eliminate task‐irrelevant components based on the identified flow relations. This step helps mitigate the interference caused by these irrelevant parts during the model feature learning process. Additionally, we extract the API sequence surrounding the recommendation point to provide supplementary knowledge for the model learning. The experimental results demonstrate our proposed approach achieving an average mean reciprocal rank (MRR) of 68.02%, outperforming the baselines' performance. Furthermore, through ablation experiments, we explore the effectiveness of our proposed code representation approach. The results indicate that combining pa‐AST with the API sequence yields improved performance compared with using them individually. Moreover, our AST augmentation and pruning techniques significantly contribute to the overall results.

Holistic Combination of Structural and Textual Code Information for Context Based API Recommendation

API method recommendation without worrying about the task-API knowledge gap.

Accurate Library Recommendation Using Combining Collaborative Filtering and Topic Model for Mobile Development.

PTM-APIRec: Leveraging Pre-trained Models of Source Code in API Recommendation

Hybrid Collaborative Filtering-Based API Recommendation

Recommending Code Reviewers for Proprietary Software Projects: A Large Scale Study

Compositional API Recommendation for Library-Oriented Code Generation

Generative API Usage Code Recommendation with Parameter Concretization

Multi-Modal API Recommendation

Tag recommendation in software information sites

Code Reviewer Recommendation in Tencent: Practice, Challenge, and Direction*

Context-aware API Recommendation Using Tensor Factorization.

Graph Neural Network Based Collaborative Filtering for API Usage Recommendation

Collective Intelligence for Smarter API Recommendations in Python

Code Recommendation With Natural Language Tags And Other Heterogeneous Data

Application programming interface recommendation for smart contract using deep learning from augmented code representation

Sequence-Aware API Recommendation Based on Collaborative Filtering

APIGen: Generative API Method Recommendation

Exploiting Heterogeneous Information for Tag Recommendation in Api Management

A Fusion of Java Domain Knowledge Base and Siamese Network for Java API Recommendation

Code context-based reviewer recommendation