How to ask questions about a git repo using chatgbt's api

ask_questions_on_codebase.py

# call this script with the command: python3 ask_questions_on_codebase.py

from langchain.document_loaders import GitLoader
from langchain.indexes import VectorstoreIndexCreator
import os

# if this fails, just use the bash command export OPENAI_API_KEY=YOUR_KEY_HERE
os.environ['OPENAI_API_KEY'] = 'YOUR_KEY_HERE'
os.environ['OPENAI_MODEL'] = 'text-embedding-ada-002' # specify your model here

directory_path = './example_data/test_repo2/'
if os.path.exists(directory_path):
    loader = GitLoader(repo_path="./example_data/test_repo2/", branch="master")
else:
    loader = GitLoader(
        clone_url="https://github.com/DavidZirinsky/brush",
        repo_path="./example_data/test_repo2/",
        branch="master",
    )

index = VectorstoreIndexCreator().from_loaders([loader])
question = "can you show me examples where AlarmManager is used?"
answer = index.query(question)
print(f"question: {question}")
print(f"answer: {answer}")
breakpoint() # breakpoint is here to avoid running the script each time you have a question