mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-21 04:44:59 +00:00
78 lines
1.7 KiB
Python
Executable file
78 lines
1.7 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
import os
|
|
import sys
|
|
import copy
|
|
import random
|
|
import json
|
|
|
|
from pathlib import Path
|
|
from collections import defaultdict
|
|
|
|
import os
|
|
import openai
|
|
|
|
from dump import dump
|
|
|
|
openai.api_key = os.getenv("OPENAI_API_KEY")
|
|
|
|
prompt_webdev = '''
|
|
I want you to act as a web development expert.
|
|
I want you to answer only with code.
|
|
Make the requested change to the provided code and output the changed code.
|
|
MAKE NO OTHER CHANGES!
|
|
Do not provide explanations!
|
|
'''
|
|
|
|
class Chat:
|
|
fnames = []
|
|
def system(self, prompt):
|
|
self.system_prompt = prompt
|
|
def file(self, fname):
|
|
self.fnames.append(fname)
|
|
def request(self, prompt):
|
|
self.request_prompt = prompt
|
|
|
|
def run(self):
|
|
prompt = self.request_prompt + '\n###\n'
|
|
|
|
for fname in self.fnames:
|
|
prompt += '\n'
|
|
prompt += fname.name
|
|
prompt += '\n```\n'
|
|
prompt += fname.read_text()
|
|
prompt += '\n```\n'
|
|
|
|
messages = [
|
|
dict(role = 'system', content = self.system_prompt),
|
|
dict(role = 'user', content = prompt),
|
|
]
|
|
|
|
completion = openai.ChatCompletion.create(
|
|
model="gpt-3.5-turbo",
|
|
messages=messages,
|
|
stream = True
|
|
)
|
|
for chunk in completion:
|
|
try:
|
|
text = chunk.choices[0].delta.content
|
|
except AttributeError:
|
|
continue
|
|
sys.stdout.write(text)
|
|
sys.stdout.flush()
|
|
|
|
|
|
chat = Chat()
|
|
|
|
chat.system(prompt_webdev)
|
|
|
|
chat.request('''
|
|
Replace *ALL* the speaker icons with a speech bubble icon.
|
|
''')
|
|
|
|
dname = Path('../easy-chat')
|
|
chat.file(dname / 'index.html')
|
|
chat.file(dname / 'chat.js')
|
|
chat.file(dname / 'chat.css')
|
|
|
|
chat.run()
|