Skip to main content

Extract and structure all the data from a Git repository to make them usable in RAG.

Project description

Apophenia

Build PyPI PyPI - Python Version PyPI - Status Downloads Downloads

Apophenia give meaning to any existing Git repository.

Apophenia extract and structure all the data from a Git repository to make them usable in RAG or within AI agents.

Apophenia impose a meaningful interpretation on a nebulous stimulus (a Git repo).

Install

$ pip install apophenia

Usage

Extract data from a given repository:

$ apophenia https://github.com/4383/niet \
  --faiss_path /tmp/results.faiss \
  --metadata_path /tmp/results.json

And use generated data in a RAG (python snippet example):

import faiss
import json
import numpy as np
from sentence_transformers import SentenceTransformer
from transformers import AutoModelForCausalLM, AutoTokenizer

# Load the FAISS index and the JSON metadata previously generated
def load_index_and_metadata(faiss_path, metadata_path):
    index = faiss.read_index(faiss_path)
    with open(metadata_path, 'r', encoding='utf-8') as f:
        metadata = json.load(f)
    return index, metadata

# Embedding of the user request
def embed_query(query, model):
    return model.encode(query, convert_to_tensor=True).cpu().numpy()

# Seach in the FAISS index
def search_in_faiss(index, query_embedding, metadata, k=5):
    distances, indices = index.search(np.array([query_embedding]), k)
    results = []
    for i, idx in enumerate(indices[0]):
        result = metadata[idx]
        result['distance'] = distances[0][i]
        results.append(result)
    return results

# Build a prompt for a generative model
def build_prompt(query, retrieved_info):
    prompt = f"Answer the following question based on the retrieved information:\n\n"
    prompt += f"Question: {query}\n\n"
    prompt += "Retrieved Information:\n"
    for info in retrieved_info:
        content_type = info.get("type", "unknown")
        content_preview = info.get("content_preview", "No preview available")
        prompt += f"- {content_type.upper()}: {content_preview}\n"
    prompt += "\nYour Answer:"
    return prompt

# Generate a response with a generative model
def generate_response(prompt, model_name="EleutherAI/gpt-neo-125M", max_length=200):
    tokenizer = AutoTokenizer.from_pretrained(model_name)
    model = AutoModelForCausalLM.from_pretrained(model_name)
    input_ids = tokenizer(prompt, return_tensors="pt").input_ids
    output = model.generate(input_ids, max_length=max_length)
    return tokenizer.decode(output[0], skip_special_tokens=True)


def run_rag_system(query, faiss_path, metadata_path, embedding_model_name, generative_model_name):
    # Load data (FAISS index and metadata, and embedding)
    index, metadata = load_index_and_metadata(faiss_path, metadata_path)
    embedding_model = SentenceTransformer(embedding_model_name)

    query_embedding = embed_query(query, embedding_model)

    # Search in FAISS
    retrieved_info = search_in_faiss(index, query_embedding, metadata)

    prompt = build_prompt(query, retrieved_info)

    response = generate_response(prompt, model_name=generative_model_name)

    return response, retrieved_info

if __name__ == "__main__":
    # Configuration
    FAISS_PATH = "results.faiss"
    METADATA_PATH = "results.json"
    EMBEDDING_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
    GENERATIVE_MODEL_NAME = "EleutherAI/gpt-neo-125M"

    query = "How does the authentication system work in this repository?"

    response, retrieved_info = run_rag_system(
        query=query,
        faiss_path=FAISS_PATH,
        metadata_path=METADATA_PATH,
        embedding_model_name=EMBEDDING_MODEL_NAME,
        generative_model_name=GENERATIVE_MODEL_NAME
    )

    print("Generated Response:")
    print(response)
    print("\nRetrieved Information:")
    for info in retrieved_info:
        print(info)

For more details:

$ apophenia -h

Applications

The results (FAISS vectors and JSON metadata) generated by Apophenia can be used within a RAG (Retrieval-Augmented Generation) system. Here’s a list of potential applications for using apophenia:

  • Generate enriched answers by combining documentation, commit messages, and code. Example questions include: "How do I use the authenticate_user function?" or "What is the structure of this project?"
  • Quickly search for specific parts of the code or documentation. Identify relevant functions or files based on queries like "Where is the authentication logic implemented?" or "Which module handles network connections?"
  • Retrieve historical information to understand bugs or errors. Analyze recent changes with queries like "What are the latest modifications in this file?" or "Which commits mention this bug?"
  • Automatically generate a changelog based on commit messages and diffs for a new release.
  • Identify outdated dependencies or technologies and plan migrations. For instance, answer queries like "Which files are using Eventlet?" or "Which commits introduced asyncio?"
  • Search for changes related to vulnerabilities or critical dependencies. Example questions include "Which files use OpenSSL?" or "Which commits fixed vulnerabilities?"
  • Generate technical guides or manuals from existing code and documentation fragments. For example, create an installation guide from README files and configuration scripts.
  • Understand individual contributions or file evolution by asking questions like "Who wrote this function?" or "What are John Doe's contributions?"
  • Search for specific concepts within the project, such as "Where is the caching logic handled?" or "Which files mention secure connections?"
  • Simplify onboarding for new developers by providing guided answers like "The main features of this project are documented in README.md." or "auth.py handles authentication logic."
  • Identify which files or functions are impacted by a specific commit with questions like "Which files were modified by this commit?" or "Which tests are affected by this change?"
  • Extract code examples from existing fragments in files or commits. For instance, generate a snippet to illustrate how to use a specific function or module.
  • Quickly find useful information to solve a technical issue, such as "Which file is responsible for this exception?" or "Which commit introduced this error?"
  • Identify the libraries used and their versions. Example questions include "Which version of Django is being used?" or "Which commits mention outdated dependencies?"
  • Search for changes related to performance optimization with questions like "Which commits optimized this file?" or "Which functions were refactored for better performance?"
  • Identify team members who are most active in certain areas of the project by asking "Who contributes the most to the networking module?" or "What are the primary files in this project?"
  • Create customized reports on the state or evolution of a project. For example, generate a report on the 10 most significant recent commits or list the main modules and the most modified files.
  • Integrate extracted data into CI/CD pipelines. For instance, identify critical files for a specific build task.
  • Compare versions of files or branches using diffs and commits.
  • Identify areas of the code that need documentation or refactoring by asking "Which files lack associated documentation?" or "Which commits mention suboptimal code?"

If you recognize yourself in one of these examples then Apophenia is for you:

$ pip install apophenia

Going Further with FAISS

You can use generated output FAISS with langchain or with any modern libraries like llamaindex

Where apophenia stands for?

Apophenia (/æpoʊˈfiːniə/) is the tendency to perceive meaningful connections between unrelated things.

Apophenia has also come to describe a human propensity to unreasonably seek definite patterns in random information, such as can occur in gambling.

https://en.wikipedia.org/wiki/Apophenia

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

apophenia-0.1.5.tar.gz (21.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

apophenia-0.1.5-py3-none-any.whl (19.1 kB view details)

Uploaded Python 3

File details

Details for the file apophenia-0.1.5.tar.gz.

File metadata

  • Download URL: apophenia-0.1.5.tar.gz
  • Upload date:
  • Size: 21.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/5.1.1 CPython/3.12.7

File hashes

Hashes for apophenia-0.1.5.tar.gz
Algorithm Hash digest
SHA256 936252a04487ca296977cd18b4143db7fff3e854fce8a90fb66b037b4ac93ba3
MD5 69e055cbbde6e4ff457b7c9435e620af
BLAKE2b-256 115fe74357be209cf4a769b220c84e36e03e83dff9d32e9797c40efd575e1cdc

See more details on using hashes here.

Provenance

The following attestation bundles were made for apophenia-0.1.5.tar.gz:

Publisher: main.yml on 4383/apophenia

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file apophenia-0.1.5-py3-none-any.whl.

File metadata

  • Download URL: apophenia-0.1.5-py3-none-any.whl
  • Upload date:
  • Size: 19.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/5.1.1 CPython/3.12.7

File hashes

Hashes for apophenia-0.1.5-py3-none-any.whl
Algorithm Hash digest
SHA256 922c4170469ace57453f2a48b87362ef3409b31c9f37afcb6663c72bcf6ea6bc
MD5 10722be6b8fcddf8cce6186da027ab4f
BLAKE2b-256 e77452e6eefecdb4f5273f658b7bd565961cd9a2bdee643b3962481e2b43c60a

See more details on using hashes here.

Provenance

The following attestation bundles were made for apophenia-0.1.5-py3-none-any.whl:

Publisher: main.yml on 4383/apophenia

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page