116 lines
3.1 KiB
Python
116 lines
3.1 KiB
Python
import os, logging
|
|
DEBUG = os.environ.get('DEBUG')
|
|
logging.basicConfig(
|
|
format='[%(asctime)s] %(levelname)s %(module)s/%(funcName)s - %(message)s',
|
|
level=logging.DEBUG if DEBUG else logging.INFO)
|
|
|
|
import requests
|
|
import subprocess
|
|
|
|
LLAMA_URL = 'http://10.55.0.105:11434/api/generate'
|
|
NOTES_DIR = '/home/tanner/notes-git/notes'
|
|
#NOTES_DIR = '/home/tanner/'
|
|
|
|
|
|
def controller_message(message):
|
|
payload = dict(home=message)
|
|
logging.info('Controller message: %s', message)
|
|
r = requests.post('https://tbot.tannercollin.com/message', data=payload, timeout=10)
|
|
if r.status_code == 200:
|
|
return True
|
|
else:
|
|
logging.exception('Unable to communicate with controller! Message: ' + message)
|
|
return False
|
|
|
|
def llama(prompt):
|
|
data = dict(model='llama3.1', prompt=prompt, stream=False)
|
|
try:
|
|
r = requests.post(LLAMA_URL, json=data, timeout=20)
|
|
r.raise_for_status()
|
|
r = r.json()
|
|
return r['response']
|
|
except BaseException as e:
|
|
logging.error('Problem with llama: {} - {}'.format(e.__class__.__name__, str(e)))
|
|
return False
|
|
|
|
|
|
def git_diff():
|
|
result = subprocess.run(['git', 'diff', '--cached', '-U0'], stdout=subprocess.PIPE, cwd=NOTES_DIR)
|
|
diff = result.stdout.decode()
|
|
|
|
if diff:
|
|
logging.info('Analyzing diff:\n%s', diff)
|
|
|
|
return diff
|
|
|
|
def git_add():
|
|
result = subprocess.run(['git', 'add', '-A'], stdout=subprocess.PIPE, cwd=NOTES_DIR)
|
|
|
|
def git_commit(message):
|
|
result = subprocess.run(['git', 'commit', '-m', message], stdout=subprocess.PIPE, cwd=NOTES_DIR)
|
|
return result.stdout.decode()
|
|
|
|
|
|
def generate_message(diff):
|
|
prompt = '''
|
|
Write a ONE LINE git commit message for the following diff.
|
|
Respond with only the commit message.
|
|
Do not include quotes.
|
|
Do not include file extensions.
|
|
DO NOT just say "Update notes", "Add new notes", "Refactor notes", "Make documentation updates".
|
|
Tersely summarize how each note changed.
|
|
Write in imperitive tense.
|
|
\n\n==============\n\n'''
|
|
|
|
prompt += diff
|
|
|
|
message = llama(prompt)
|
|
|
|
if not message:
|
|
logging.info(' Failed to generate')
|
|
return False
|
|
|
|
logging.info('Generated message: %s', message)
|
|
|
|
if '.md' in message.lower():
|
|
logging.info(' Message contains .md extension, aborting')
|
|
return False
|
|
elif 'notes/' in message.lower():
|
|
logging.info(' Message contains notes directory, aborting')
|
|
return False
|
|
elif '\n' in message:
|
|
logging.info(' Message contains new lines, aborting')
|
|
return False
|
|
|
|
return message
|
|
|
|
|
|
|
|
def main():
|
|
git_add() # add so we can diff new files
|
|
|
|
diff = git_diff()
|
|
|
|
if not diff:
|
|
logging.info('No diff, exiting.')
|
|
exit(0)
|
|
|
|
for _ in range(5):
|
|
message = generate_message(diff)
|
|
if message:
|
|
break
|
|
else: # for loop never broke
|
|
msg = 'Unable to generate acceptable diff message.'
|
|
controller_message(msg)
|
|
logging.info(msg)
|
|
message = 'Auto-commit, can\'t generate message'
|
|
|
|
if DEBUG:
|
|
logging.info('Running in debug, not actually committing.')
|
|
else:
|
|
git_commit(message)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|