Method rerank

autoagent/memory/code_memory.py:90–174

(self, query_text: str, query_results: List[Dict])

        
        return unique_lines
    def rerank(self, query_text: str, query_results: List[Dict]) -> List[Dict]:
        system_prompt = \
        """
        You are a helpful assistant that reranks the given code files (containing the path of files and Overview of the content of files) based on the query.
        You should rerank the code files based on the query, and the most relevant code files should be ranked on the top. 
        You should select the top 5 code files to answer the query, by giving the file path of the code files.

        Example: 
        [Query]: "The definition of 'BaseAgent'"
        [Code files]: 
        File: /Users/tangjiabin/Documents/reasoning/SelfAgent/sa/agents/__init__.py
        Content: from .ABCAgent import ABCAgent
        from .BaseAgent import BaseAgent
        from .ManagerAgent import ManagerAge...
        ---
        File: /Users/tangjiabin/Documents/reasoning/SelfAgent/sa/agents/__init__.py
        Content: from .ABCAgent import ABCAgent
        from .BaseAgent import BaseAgent
        from .ManagerAgent import ManagerAge...
        ---
        File: /Users/tangjiabin/Documents/reasoning/SelfAgent/sa/agents/__init__.py
        Content: from .ABCAgent import ABCAgent
        from .BaseAgent import BaseAgent
        from .ManagerAgent import ManagerAge...
        ---
        File: /Users/tangjiabin/Documents/reasoning/SelfAgent/sa/agents/__init__.py
        Content: from .ABCAgent import ABCAgent
        from .BaseAgent import BaseAgent
        from .ManagerAgent import ManagerAge...
        ---
        File: /Users/tangjiabin/Documents/reasoning/SelfAgent/sa/agent_prompts/__init__.py
        Content: from .BasePrompt import BasePromptGen, ManagerPromptGen, PromptGen
        ...
        ---
        File: /Users/tangjiabin/Documents/reasoning/SelfAgent/sa/agent_prompts/__init__.py
        Content: from .BasePrompt import BasePromptGen, ManagerPromptGen, PromptGen
        ...
        ---
        File: /Users/tangjiabin/Documents/reasoning/SelfAgent/sa/agent_prompts/__init__.py
        Content: from .BasePrompt import BasePromptGen, ManagerPromptGen, PromptGen
        ...
        ---
        File: /Users/tangjiabin/Documents/reasoning/SelfAgent/sa/agent_prompts/__init__.py
        Content: from .BasePrompt import BasePromptGen, ManagerPromptGen, PromptGen
        ...
        ---
        File: /Users/tangjiabin/Documents/reasoning/SelfAgent/sa/agents/BaseAgent.py
        Content: from typing import List

        from sa.actions import BaseAction, FinishAct, ThinkAct, PlanAct
        from sa.age...
        ---
        File: /Users/tangjiabin/Documents/reasoning/SelfAgent/sa/agents/BaseAgent.py
        Content: from typing import List

        from sa.actions import BaseAction, FinishAct, ThinkAct, PlanAct
        from sa.age...
        ---