Code retrieval method and apparatus based on large language model

An electronic device, comprising:

• at least one processor;
• and
• a memory communicatively connected with the at least one processor;
• wherein the memory stores instructions executable by the at least one processor, and the instructions are executed by the at least one processor to enable the at least one processor to perform a method for code retrieval based on a large language model, wherein the method for code retrieval based on a large language model comprises:
• acquiring a code retrieval query to obtain a retrieval vector of the code retrieval query;
• acquiring a first index of a target code library, the first index comprising a plurality of code blocks and a plurality of code block vectors;
• acquiring a target code block according to the retrieval vector and the first index;
• acquiring a second index of the target code library, the second index being a code architecture knowledge graph, nodes in the code architecture knowledge graph representing code files, and edges in the code architecture knowledge graph representing dependency relationships between the code files;
• acquiring a target code file corresponding to the target code block according to a source code file corresponding to the target code block and the second index;
• and
• acquiring a retrieval result according to the target code block and the target code file.

A non-transitory computer readable storage medium with computer instructions stored thereon, wherein the computer instructions are used for causing a method for code retrieval based on a large language model, wherein the method for code retrieval based on a large language model comprises:

• acquiring a code retrieval query to obtain a retrieval vector of the code retrieval query;
• acquiring a first index of a target code library, the first index comprising a plurality of code blocks and a plurality of code block vectors;
• acquiring a target code block according to the retrieval vector and the first index;
• acquiring a second index of the target code library, the second index being a code architecture knowledge graph, nodes in the code architecture knowledge graph representing code files, and edges in the code architecture knowledge graph representing dependency relationships between the code files;
• acquiring a target code file corresponding to the target code block according to a source code file corresponding to the target code block and the second index;
• and
• acquiring a retrieval result according to the target code block and the target code file.