Function search_py_elements_inside_project

src/hyperagent/code_search.py:74–149

Get all matched identifiers from a repo Args: names (list): The list of identifiers to be searched backend (object): The backend object that provides search functionality num_result (int, optional): The maximum number of matches for each identifier to return. Default

(names, backend, num_result=3, verbose=False)

        return "\n".join(results)

def search_py_elements_inside_project(names, backend, num_result=3, verbose=False):
    """Get all matched identifiers from a repo
    
    Args:
        names (list): The list of identifiers to be searched
        backend (object): The backend object that provides search functionality
        num_result (int, optional): The maximum number of matches for each identifier to return. Defaults to 2.
        verbose (bool, optional): If true, additional information about the search results is provided. Defaults to False.
    
    Returns:
        list: A list of matched identifiers
    
    Raises:
        None
    
    Examples:
        >>> search_py_elements_inside_project(['foo', 'bar'], backend)
        {'foo': [{'name': 'FooClass', 'full_name': 'my_module.FooClass', 'documentation': 'This is the documentation for FooClass', 'implementation': 'class FooClass:\n    def __init__(self):\n        pass\n\n    def foo_method(self):\n        pass\n'}],
         'bar': [{'name': 'bar_function', 'full_name': 'my_module.bar_function', 'documentation': 'This is the documentation for bar_function', 'implementation': 'def bar_function():\n    pass\n'}]}
    """
    output_dict = {name: [] for name in names}
    for name in names:
        if not name.endswith(".py"):
            class_definitions = backend.search(f"class {name}", all_scopes=True)
            function_definitions = backend.search(f"def {name}", all_scopes=True)
            variable_definitions = backend.search(name, all_scopes=True)
            idx = 0
            for definition in class_definitions:
                if definition.is_definition():
                    extracted_definition = {
                        "path": definition.module_path,
                        "full_name": definition.full_name,
                        "implementation": get_code_jedi(definition, verbose)
                    }
                    output_dict[name].append(extracted_definition)
                    idx += 1
                    if idx == num_result:
                        break
            
            idx = 0
            for definition in function_definitions:
                if definition.is_definition():
                    extracted_definition = {
                        "path": definition.module_path,
                        "full_name": definition.full_name,
                        "implementation": get_code_jedi(definition, verbose),
                    }
                    output_dict[name].append(extracted_definition)
                    idx += 1
                    if idx == num_result:
                        break
            
            idx = 0
            for definition in variable_definitions:
                extracted_definition = {
                    "path": definition.module_path,
                    "full_name": definition.full_name,
                    "implementation": definition.description,