OMGeeky/gpt-pilot
1
0
Fork 0
mirror of https://github.com/OMGeeky/gpt-pilot.git synced 2026-01-08 20:07:02 +01:00
Code Issues Packages Projects Releases Wiki Activity

Finished implementation for saving and loading development steps

Browse Source
This commit is contained in:
Zvonimir Sabljic
2023-08-02 11:52:19 +02:00
parent 96633f2b57
commit 6c84d15b44
2 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
euclid/database/database.py
View File

@@ -106,20 +106,21 @@ def get_progress_steps(app_id, step=None):
return steps
def save_development_step(app_id, messages):
def save_development_step(app_id, messages, response):
app = get_app(app_id)
hash_id = hash_data(messages)
try:
dev_step = DevelopmentSteps.create(app=app, hash_id=hash_id, messages=messages)
dev_step = DevelopmentSteps.create(app=app, hash_id=hash_id, messages=messages, llm_response=response)
except IntegrityError:
print(f"A Development Step with hash_id {hash_id} already exists.")
return None
return dev_step
def get_development_step_by_hash_id(hash_id):
def get_development_step_from_messages(app_id, messages):
hash_id = hash_data(messages)
try:
dev_step = DevelopmentSteps.get(DevelopmentSteps.hash_id == hash_id)
dev_step = DevelopmentSteps.get((DevelopmentSteps.hash_id == hash_id) & (DevelopmentSteps.app == app_id))
except DoesNotExist:
print(f"No Development Step found with hash_id {hash_id}")
return None

13
euclid/helpers/AgentConvo.py
View File

@@ -1,4 +1,5 @@
import subprocess
from database.database import get_development_step_from_messages, save_development_step
from utils.utils import array_of_objects_to_string
from utils.llm_connection import get_prompt, create_gpt_chat_completion
from utils.utils import get_sys_message, find_role_from_step, capitalize_first_word_with_underscores
@@ -21,7 +22,17 @@ class AgentConvo:
prompt = get_prompt(prompt_path, prompt_data)
self.messages.append({"role": "user", "content": prompt})
response = create_gpt_chat_completion(self.messages, self.high_level_step, function_calls=function_calls)
# check if we already have the LLM response saved
saved_checkpoint = get_development_step_from_messages(self.agent.project.args['app_id'], self.messages)
if saved_checkpoint is not None:
# if we do, use it
response = saved_checkpoint.llm_response
self.messages = saved_checkpoint.messages
else:
# if we don't, get the response from LLM
response = create_gpt_chat_completion(self.messages, self.high_level_step, function_calls=function_calls)
save_development_step(self.agent.project.args['app_id'], self.messages, response)
# TODO handle errors from OpenAI
if response == {}: