gpt-migrate

gpt 不同语言代码迁移，比如java项目转python项目

Develop

python main.py --sourcedir /path/to/my-python-app --sourceentry app.py --targetdir /path/to/my-nodejs-app --targetlang nodejs

--sourcedir 指定原项目目录，--targetlang指定目标语言，如nodejs

原理分析

1、检测 sourcedir 是什么语言的代码：

    detected_language = detect_language(sourcedir) if not sourcelang else sourcelang

在 config.py 中，定义了常见的编程语言，detect_language 函数遍历 sourcedir 目录，找出后缀最多的文件的后缀，然后再确定是哪种编程语言。

2、创建环境

        create_environment(globals)

    docker_prompt_template = prompt_constructor(HIERARCHY, GUIDELINES, WRITE_CODE, CREATE_DOCKER, SINGLEFILE)

其实就是拼接 prompt , 替换 targetlang，targetdir 等几个变量：

    llm_write_file(prompt,
                    target_path="Dockerfile",
                    waiting_message="Creating your environment...",
                    success_message=f"Created Docker environment for {globals.targetlang} project in directory '{globals.targetdir}'.",
                    globals=globals)

prompt 模板在 gpt_migrate/prompts 目录中。注意，llm_write_file 函数写Dockerfile，怎么写呢？上面只是拼接了模板的 prompt而已，那这些 prompt 其实是调用openai来生成的：


        file_name,language,file_content = globals.ai.write_code(prompt)[0]
        
        
            
    def write_code(self, prompt):
        message=[{"role": "user", "content": str(prompt)}] 
        response = openai.ChatCompletion.create(
            messages=message,
            stream=False,
            model=self.model_name,
            max_tokens=self.max_tokens,
            temperature=self.temperature
        )

3、代码迁移，如java代码迁移到python语言：

        def migrate(sourcefile, globals, parent_file=None):
            # recursively work through each of the files in the source directory, starting with the entrypoint.
            internal_deps_list, external_deps_list = get_dependencies(sourcefile=sourcefile,globals=globals)
            for dependency in internal_deps_list:
                migrate(dependency, globals, parent_file=sourcefile)
            file_name = write_migration(sourcefile, external_deps_list, target_deps_per_file.get(sourcefile), globals)
            target_deps_per_file[parent_file].append(file_name)

        migrate(sourceentry, globals)

仔细看发现 migrate 采用了递归，

    external_deps_prompt_template = prompt_constructor(HIERARCHY, GUIDELINES, GET_EXTERNAL_DEPS)

这里 prompt 发生了变化，GET_EXTERNAL_DEPS

4、综上：

拼接 prompt 调用openai，创建一个dockerfile测试环境。
环拼接 prompt ，遍历所有依赖；
再拼接 prompt，转换代码。
启动docker测试环境，测试代码

README.md 2.9 KB History Raw

gpt-migrate

Develop

原理分析

README.md 2.9 KB

History Raw