OMGeeky/gpt-pilot
1
0
Fork 0
mirror of https://github.com/OMGeeky/gpt-pilot.git synced 2026-02-23 15:49:50 +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